arXiv reaDer
トランスダクティブ画像セグメンテーション:自己訓練と不確実性推定の効果
Transductive image segmentation: Self-training and effect of uncertainty estimation
半教師あり学習(SSL)は、トレーニング中にラベルのないデータを使用して、より良いモデルを学習します。医療画像セグメンテーションのためのSSLに関するこれまでの研究は、主に、見えないデータへのモデルの一般化を改善することに焦点を当てていました。ただし、一部のアプリケーションでは、私たちの主な関心は一般化ではなく、モデル開発中に完全に利用可能な特定のラベルなしデータベースで最適な予測を取得することです。例には、画像表現型を抽出するための集団研究が含まれます。この作業では、SSLの見落とされがちな側面である変換について調査します。これは、一般化を改善するのではなく、トレーニング中の最適化に含まれる場合に、対象のラベルなしデータに対して行われる予測の品質に焦点を当てています。私たちは自己訓練の枠組みに焦点を当て、その形質導入の可能性を探ります。情報獲得のレンズを通してそれを分析し、学習が較正されたモデルまたは自信のないモデルの使用から利益を得ることが明らかになります。外傷性脳損傷のマルチクラスセグメンテーションのための大規模なMRIデータベースでの私たちの広範な実験は、トランスダクティブ予測とインダクティブ予測を比較したときに有望な結果を示しています。この研究は、医療画像分析に適したパラダイムであるトランスダクティブ学習に関するさらなる研究を刺激すると信じています。
Semi-supervised learning (SSL) uses unlabeled data during training to learn better models. Previous studies on SSL for medical image segmentation focused mostly on improving model generalization to unseen data. In some applications, however, our primary interest is not generalization but to obtain optimal predictions on a specific unlabeled database that is fully available during model development. Examples include population studies for extracting imaging phenotypes. This work investigates an often overlooked aspect of SSL, transduction. It focuses on the quality of predictions made on the unlabeled data of interest when they are included for optimization during training, rather than improving generalization. We focus on the self-training framework and explore its potential for transduction. We analyze it through the lens of Information Gain and reveal that learning benefits from the use of calibrated or under-confident models. Our extensive experiments on a large MRI database for multi-class segmentation of traumatic brain lesions shows promising results when comparing transductive with inductive predictions. We believe this study will inspire further research on transductive learning, a well-suited paradigm for medical image analysis.
updated: Mon Jul 19 2021 15:26:07 GMT+0000 (UTC)
published: Mon Jul 19 2021 15:26:07 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アソシエイト