arXiv reaDer
半教師あり医療画像セグメンテーションのためのハード領域の相互一貫性の実施
Enforcing Mutual Consistency of Hard Regions for Semi-supervised Medical Image Segmentation
この論文では、半教師あり医療画像セグメンテーションのためにラベルのないハード領域を効果的に活用するための新しい相互整合性ネットワーク(MC-Net +)を提案しました。 MC-Net +モデルは、限定された注釈でトレーニングされた深いモデルが、画像セグメンテーションタスクのあいまいな領域(接着エッジや細い枝など)で非常に不確実で誤分類されやすい予測を出力する傾向があるという観察に動機付けられています。これらの地域レベルの挑戦的なサンプルを活用することで、半教師ありセグメンテーションモデルのトレーニングをより効果的にすることができます。したがって、提案するMC-Net +モデルは、2つの新しい設計で構成されています。まず、モデルには1つの共有エンコーダーと複数の視覚的に異なるデコーダーが含まれています(つまり、異なるアップサンプリング戦略を使用しています)。複数のデコーダーの出力の統計的不一致は、モデルの不確実性を示すために計算されます。これは、ラベルのないハード領域を示します。次に、1つのデコーダーの確率出力と他のデコーダーのソフト疑似ラベルの間に新しい相互整合性制約が適用されます。このようにして、トレーニング中のモデルの不確実性を最小限に抑え、一般化された特徴表現を学習するために、モデルにラベルなしデータのこのような困難な領域で不変で低エントロピーの結果を生成させます。 MC-Net +のセグメンテーション結果を、3つの公的医療データセットに対する5つの最先端の半教師ありアプローチと比較しました。 2つの一般的な半教師あり設定を使用した拡張実験は、他の既存の方法よりも優れたモデルのパフォーマンスを示しています。これにより、半教師あり医療画像セグメンテーションの新しい最先端が設定されます。
In this paper, we proposed a novel mutual consistency network (MC-Net+) to effectively exploit the unlabeled hard regions 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 the ambiguous regions (e.g. adhesive edges or thin branches) for the image segmentation task. 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 sightly 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, a new mutual consistency constraint is enforced between one decoder's probability output and other decoders' soft pseudo labels. In this way, we minimize the model's uncertainty during training and force the model to generate invariant and low-entropy results in such challenging areas of unlabeled data, in order to learn a generalized feature representation. We compared the segmentation results of the 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: Tue Sep 21 2021 04:47:42 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アソシエイト