arXiv reaDer
NoTeacherによる放射線画像の半教師あり分類:平均ではない教師
Semi-supervised classification of radiology images with NoTeacher: A Teacher that is not Mean
深層学習モデルは、放射線画像分類で強力なパフォーマンスを実現しますが、その実用的なアプリケーションは、大きなラベル付きトレーニングデータセットの必要性によってボトルネックになっています。半教師あり学習(SSL)アプローチは、大きなラベルのないデータセットと一緒に小さなラベルの付いたデータセットを活用し、ラベル付けのコストを削減する可能性を提供します。この作業では、確率的グラフィカルモデルを組み込んだ新しい整合性ベースのSSLフレームワークであるNoTeacherを紹介します。時間的アンサンブルを介して更新された教師ネットワークを維持するMeanTeacherとは異なり、NoTeacherは2つの独立したネットワークを採用しているため、教師ネットワークは不要です。 NoTeacherをカスタマイズして、放射線画像分類のさまざまな課題を処理する方法を示します。具体的には、2Dおよび3D入力、ユニラベルおよびマルチラベル分類、およびトレーニングデータのラベル付き部分とラベルなし部分の間のクラス分布の不一致を伴うシナリオへの適応について説明します。放射線医学の主力モダリティ(X線、CT、MRI)にまたがる3つの公開ベンチマークデータセットの現実的な経験的評価では、NoTeacherが5〜15%未満のラベリング予算で完全に監視されたAUROCの90〜95%以上を達成することを示しています。さらに、NoTeacherは、最小限のハイパーパラメータ調整で確立されたSSLメソッドよりも優れており、放射線アプリケーションでの半教師あり学習の原則的かつ実用的なオプションとしての意味合いがあります。
Deep learning models achieve strong performance for radiology image classification, but their practical application is bottlenecked by the need for large labeled training datasets. Semi-supervised learning (SSL) approaches leverage small labeled datasets alongside larger unlabeled datasets and offer potential for reducing labeling cost. In this work, we introduce NoTeacher, a novel consistency-based SSL framework which incorporates probabilistic graphical models. Unlike Mean Teacher which maintains a teacher network updated via a temporal ensemble, NoTeacher employs two independent networks, thereby eliminating the need for a teacher network. We demonstrate how NoTeacher can be customized to handle a range of challenges in radiology image classification. Specifically, we describe adaptations for scenarios with 2D and 3D inputs, uni and multi-label classification, and class distribution mismatch between labeled and unlabeled portions of the training data. In realistic empirical evaluations on three public benchmark datasets spanning the workhorse modalities of radiology (X-Ray, CT, MRI), we show that NoTeacher achieves over 90-95% of the fully supervised AUROC with less than 5-15% labeling budget. Further, NoTeacher outperforms established SSL methods with minimal hyperparameter tuning, and has implications as a principled and practical option for semisupervised learning in radiology applications.
updated: Tue Aug 10 2021 03:08:35 GMT+0000 (UTC)
published: Tue Aug 10 2021 03:08:35 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アソシエイト