arXiv reaDer
MisMatch: 限られたラベルを使用した微分形態学的特徴摂動の一貫性によるキャリブレーションされたセグメンテーション
MisMatch: Calibrated Segmentation via Consistency on Differential Morphological Feature Perturbations with Limited Labels
半教師あり学習 (SSL) は、医用画像におけるラベル不足の問題に対処するための有望な機械学習パラダイムです。 SSL メソッドは、もともと画像分類で開発されました。画像分類における最先端の SSL メソッドは、一貫性の正則化を利用して、入力レベルの摂動に対して不変であるラベルのない予測を学習します。ただし、イメージ レベルの摂動は、セグメンテーションの設定におけるクラスターの仮定に違反します。さらに、既存の画像レベルの摂動は手作りであり、最適ではない可能性があります。したがって、既存の SSL 画像分類方法をそのままセグメンテーションに適用することは簡単なことではありません。この論文では、2 つの異なる学習形態学的特徴の摂動から派生したペアの予測間の整合性に基づく半教師付きセグメンテーション フレームワークである MisMatch を提案します。 MisMatch は、エンコーダーと 2 つのデコーダーで構成されます。 1 つのデコーダーは、ラベル付けされていないデータの前景に対する肯定的な注意を学習し、それによって前景の拡張された特徴を生成します。もう一方のデコーダは、同じラベルなしデータで前景に対する否定的な注意を学習し、それによって前景の侵食された特徴を生成します。最初に2D U-netベースのMisMatchフレームワークを開発し、CTベースの肺血管セグメンテーションタスクで広範な相互検証を実行し、MisMatchが統計的に最先端の半教師あり方法よりも優れていることを示します。ラベルが使用されます。 2 番目の実験では、U-net ベースの MisMatch が、MRI ベースの脳腫瘍セグメンテーション タスクで最先端の方法よりも優れていることを示します。 3 番目の実験では、左心房セグメンテーション タスクで、3D MisMatch が入力レベル拡張を使用する以前の方法よりも優れていることを示します。最後に、ベースラインを超える MisMatch のパフォーマンスの向上は、より優れたキャリブレーションに起因する可能性があることがわかりました。
Semi-supervised learning (SSL) is a promising machine learning paradigm to address the issue of label scarcity in medical imaging. SSL methods were originally developed in image classification. The state-of-the-art SSL methods in image classification utilise consistency regularisation to learn unlabelled predictions which are invariant to input level perturbations. However, image level perturbations violate the cluster assumption in the setting of segmentation. Moreover, existing image level perturbations are hand-crafted which could be sub-optimal. Therefore, it is a not trivial to straightforwardly adapt existing SSL image classification methods in segmentation. In this paper, we propose MisMatch, a semi-supervised segmentation framework based on the consistency between paired predictions which are derived from two differently learnt morphological feature perturbations. MisMatch consists of an encoder and two decoders. One decoder learns positive attention for foreground on unlabelled data thereby generating dilated features of foreground. The other decoder learns negative attention for foreground on the same unlabelled data thereby generating eroded features of foreground. We first develop a 2D U-net based MisMatch framework and perform extensive cross-validation on a CT-based pulmonary vessel segmentation task and show that MisMatch statistically outperforms state-of-the-art semi-supervised methods when only 6.25% of the total labels are used. In a second experiment, we show that U-net based MisMatch outperforms state-of-the-art methods on an MRI-based brain tumour segmentation task. In a third experiment, we show that a 3D MisMatch outperforms a previous method using input level augmentations, on a left atrium segmentation task. Lastly, we find that the performance improvement of MisMatch over the baseline might originate from its better calibration.
updated: Sat Oct 29 2022 19:27:48 GMT+0000 (UTC)
published: Sat Oct 23 2021 09:22:41 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アソシエイト