arXiv reaDer
ACPL:半監視医用画像分類のためのアンチカリキュラム疑似ラベリング
ACPL: Anti-curriculum Pseudo-labelling forSemi-supervised Medical Image Classification
医療画像分析(MIA)における効果的な半教師あり学習(SSL)は、2つの課題に対処する必要があります。1)マルチクラス(病変分類など)とマルチラベル(複数疾患診断など)の両方の問題に効果的に取り組む。 2)不均衡な学習を処理する(疾患の有病率のばらつきが大きいため)。 SSL MIAを調査する1つの戦略は、疑似ラベリング戦略に基づいていますが、いくつかの欠点があります。疑似ラベリングは、一般に一貫性学習よりも精度が低く、特にマルチクラスとマルチラベルの両方の問題に対応するように設計されておらず、不均衡な学習によって問題が発生する可能性があります。この論文では、しきい値によって信頼できる疑似ラベルを選択する従来の方法とは異なり、アンチカリキュラム疑似ラベル(ACPL)と呼ばれる新しいSSLアルゴリズムを提案します。これは、有益なラベルなしサンプルを選択する新しい手法を導入し、トレーニングバランスを改善し、モデルを可能にします。マルチラベル問題とマルチクラス問題の両方で機能し、分類器の正確なアンサンブルによって疑似ラベルを推定します(疑似ラベルの精度を向上させます)。胸部疾患のマルチラベル分類用の胸部X-Ray14と皮膚病変のマルチクラス分類用のISIC2018の2つの公的医療画像分類ベンチマークでACPLを評価するために広範な実験を実行します。私たちの方法は、両方のデータセットで以前のSOTASSL方法よりも優れています。
Effective semi-supervised learning (SSL) in medical im-age analysis (MIA) must address two challenges: 1) workeffectively on both multi-class (e.g., lesion classification)and multi-label (e.g., multiple-disease diagnosis) problems,and 2) handle imbalanced learning (because of the highvariance in disease prevalence). One strategy to explorein SSL MIA is based on the pseudo labelling strategy, butit has a few shortcomings. Pseudo-labelling has in generallower accuracy than consistency learning, it is not specifi-cally design for both multi-class and multi-label problems,and it can be challenged by imbalanced learning. In this paper, unlike traditional methods that select confident pseudo label by threshold, we propose a new SSL algorithm, called anti-curriculum pseudo-labelling (ACPL), which introduces novel techniques to select informative unlabelled samples, improving training balance and allowing the model to work for both multi-label and multi-class problems, and to estimate pseudo labels by an accurate ensemble of classifiers(improving pseudo label accuracy). We run extensive experiments to evaluate ACPL on two public medical image classification benchmarks: Chest X-Ray14 for thorax disease multi-label classification and ISIC2018 for skin lesion multi-class classification. Our method outperforms previous SOTA SSL methods on both datasets.
updated: Thu Nov 25 2021 05:31:52 GMT+0000 (UTC)
published: Thu Nov 25 2021 05:31:52 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アソシエイト