arXiv reaDer
半教師あり医療画像セグメンテーションのための相互一貫性学習
Mutual Consistency Learning for Semi-supervised Medical Image Segmentation
この論文では、半教師あり医療画像セグメンテーションのためにラベルなしデータを効果的に活用するための新しい相互整合性ネットワーク(MC-Net +)を提案します。 MC-Net +モデルは、限定された注釈でトレーニングされた深いモデルが、医療画像のセグメンテーションのためにあいまいな領域(たとえば、接着エッジや細い枝)で非常に不確実で誤分類されやすい予測を出力する傾向があるという観察に動機付けられています。これらの地域レベルの挑戦的なサンプルを活用することで、半教師ありセグメンテーションモデルのトレーニングをより効果的にすることができます。したがって、提案するMC-Net +モデルは、2つの新しい設計で構成されています。まず、モデルには1つの共有エンコーダーと複数のわずかに異なるデコーダーが含まれています(つまり、異なるアップサンプリング戦略を使用しています)。複数のデコーダーの出力の統計的不一致は、モデルの不確実性を示すために計算されます。これは、ラベルのないハード領域を示します。次に、1つのデコーダーの確率出力と他のデコーダーのソフト疑似ラベルの間に新しい相互整合性制約を適用します。このようにして、トレーニング中の複数の出力の不一致(つまり、モデルの不確実性)を最小限に抑え、モデルにそのような困難な領域で不変の結果を生成させ、より有用な機能をキャプチャすることを目指します。 MC-Net +のセグメンテーション結果を、3つの公的医療データセットに対する5つの最先端の半教師ありアプローチと比較しました。 2つの一般的な半教師あり設定を使用した拡張実験は、他の既存の方法よりも優れたモデルのパフォーマンスを示しています。これにより、半教師あり医療画像セグメンテーションの新しい最先端が設定されます。
In this paper, we propose a novel mutual consistency network (MC-Net+) to effectively exploit the unlabeled data for semi-supervised medical image segmentation. The MC-Net+ model is motivated by the observation that deep models trained with limited annotations are prone to output highly uncertain and easily mis-classified predictions in ambiguous regions (e.g., adhesive edges or thin branches) for medical image segmentation. Leveraging these region-level challenging samples can make the semi-supervised segmentation model training more effective. Therefore, our proposed MC-Net+ model consists of two new designs. First, the model contains one shared encoder and multiple slightly different decoders (i.e., using different up-sampling strategies). The statistical discrepancy of multiple decoders' outputs is computed to denote the model's uncertainty, which indicates the unlabeled hard regions. Second, we apply a novel mutual consistency constraint between one decoder's probability output and other decoders' soft pseudo labels. In this way, we minimize the discrepancy of multiple outputs (i.e., the model uncertainty) during training and force the model to generate invariant results in such challenging regions, aiming at capturing more useful features. We compared the segmentation results of our MC-Net+ with five state-of-the-art semi-supervised approaches on three public medical datasets. Extension experiments with two common semi-supervised settings demonstrate the superior performance of our model over other existing methods, which sets a new state of the art for semi-supervised medical image segmentation.
updated: Wed Apr 20 2022 00:48:55 GMT+0000 (UTC)
published: Tue Sep 21 2021 04:47:42 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アソシエイト