arXiv reaDer
NVUM:堅牢な医用画像分類のための非揮発性バイアスのないメモリ
NVUM: Non-Volatile Unbiased Memory for Robust Medical Image Classification
実世界の大規模医療画像分析(MIA)データセットには、3つの課題があります。1)トレーニングの収束と一般化に影響を与えるノイズの多いラベル付きサンプルが含まれている、2)通常、クラスごとのサンプルの分布が不均衡である、3)通常サンプルが複数の診断を持つ可能性があるマルチラベル問題を構成します。現在のアプローチは、これらの問題のサブセットを解決するように一般的に訓練されていますが、3つの問題に同時に対処する方法を私たちは知りません。この論文では、Non-Volatile Unbiased Memory(NVUM)と呼ばれる新しいトレーニングモジュールを提案します。これは、ノイズの多いマルチラベル問題での新しい正則化損失に対して、モデルロジットの移動平均を非揮発性で保存します。さらに、不均衡な学習問題に対するNVUMupdateの分類予測の偏りをなくします。トレーニングは、Chest-Xray14とCheXpertによって形成されたノイズの多いマルチラベル不均衡胸部X線(CXR)トレーニングセットで実行され、このペーパーで提案された新しいベンチマークでNVUMを評価するために広範な実験を実行します。クリーンなマルチラベルCXRデータセットOpenIおよびPadChestで実行されます。私たちの方法は、すべての評価でノイズの多いラベルを処理できる以前の最先端のCXR分類器および以前の方法よりも優れています。
Real-world large-scale medical image analysis (MIA) datasets havethree challenges: 1) they contain noisy-labelled samples that affect training con-vergence and generalisation, 2) they usually have an imbalanced distribution ofsamples per class, and 3) they normally comprise a multi-label problem, wheresamples can have multiple diagnoses. Current approaches are commonly trainedto solve a subset of those problems, but we are unaware of methods that ad-dress the three problems simultaneously. In this paper, we propose a new trainingmodule called Non-Volatile Unbiased Memory (NVUM), which non-volatilitystores running average of model logits for a new regularization loss on noisymulti-label problem. We further unbias the classification prediction in NVUMupdate for imbalanced learning problem. We run extensive experiments to eval-uate NVUM on new benchmarks proposed by this paper, where training is per-formed on noisy multi-label imbalanced chest X-ray (CXR) training sets, formedby Chest-Xray14 and CheXpert, and the testing is performed on the clean multi-label CXR datasets OpenI and PadChest. Our method outperforms previous state-of-the-art CXR classifiers and previous methods that can deal with noisy labelson all evaluations.
updated: Thu Mar 03 2022 07:11:22 GMT+0000 (UTC)
published: Sat Mar 06 2021 07:42:36 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アソシエイト