arXiv reaDer
二重不確実性推定を使用したラベルノイズによる医用画像分類の改善
Improving Medical Image Classification with Label Noise Using Dual-uncertainty Estimation
ディープニューラルネットワークはデータ駆動型であることが知られており、ラベルノイズはモデルのパフォーマンスに著しい影響を与える可能性があります。最近の研究では、ノイズの多いレートでも、従来の画像認識に対して優れた堅牢性が示されています。医療アプリケーションでは、医用画像データセットは非対称(クラス依存)ノイズを持ち、観察者のばらつきが大きい傾向があるため、ラベルノイズのあるデータセットからの学習はより困難です。この論文では、医用画像のラベルノイズの2つの一般的なタイプ、つまり専門家の意見の不一致によるラベルノイズの不一致と、誤った診断記録による単一ターゲットのラベルノイズについて体系的に説明および定義します。次に、医用画像分類タスクの中でこれら2つのラベルノイズを処理するための不確実性推定ベースのフレームワークを提案します。直接不確実性予測とモンテカルロドロップアウトを介して、不一致ラベルノイズと単一ターゲットラベルノイズを測定するための二重不確実性推定アプローチを設計します。ブースティングベースのカリキュラムトレーニング手順は、後で堅牢な学習のために導入されます。皮膚病変、前立腺癌、網膜疾患の3つの異なる疾患について広範な実験を行うことにより、この方法の有効性を示します。また、評価とベンチマークのための偏りのないゴールデンスタンダードデータセットを備えた10人以上の眼科医からの注釈で構成される大規模な再設計されたデータベースをリリースします。
Deep neural networks are known to be data-driven and label noise can have a marked impact on model performance. Recent studies have shown great robustness to classic image recognition even under a high noisy rate. In medical applications, learning from datasets with label noise is more challenging since medical imaging datasets tend to have asymmetric (class-dependent) noise and suffer from high observer variability. In this paper, we systematically discuss and define the two common types of label noise in medical images - disagreement label noise from inconsistency expert opinions and single-target label noise from wrong diagnosis record. We then propose an uncertainty estimation-based framework to handle these two label noise amid the medical image classification task. We design a dual-uncertainty estimation approach to measure the disagreement label noise and single-target label noise via Direct Uncertainty Prediction and Monte-Carlo-Dropout. A boosting-based curriculum training procedure is later introduced for robust learning. We demonstrate the effectiveness of our method by conducting extensive experiments on three different diseases: skin lesions, prostate cancer, and retinal diseases. We also release a large re-engineered database that consists of annotations from more than ten ophthalmologists with an unbiased golden standard dataset for evaluation and benchmarking.
updated: Sun Feb 28 2021 14:56:45 GMT+0000 (UTC)
published: Sun Feb 28 2021 14:56:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト