arXiv reaDer
MisMatch:一貫性ベースの半教師あり医療画像セグメンテーションに注意を払って予測信頼度を変更する方法を学ぶ
MisMatch: Learning to Change Predictive Confidences with Attention for Consistency-Based, Semi-Supervised Medical Image Segmentation
ラベルの欠如は、特に医用画像などのアプリケーションにおいて、画像の分類とセグメンテーションのための深層学習ベースの方法における基本的な制約の1つです。半教師あり学習(SSL)は、ラベルのカーシティの課題に対処するための有望な方法です。最先端のSSLメソッドは、一貫性の正則化を利用して、予測の信頼性の摂動に対して不変であるラベルのない予測を学習します。ただし、このようなSSLアプローチは、最適ではない可能性のある手作りの拡張手法に依存しています。この論文では、新しい整合性ベースの半教師ありセグメンテーション手法であるMisMatchを提案します。 MisMatchは、信頼度を上げたり下げたりして、ペアの予測を生成することを自動的に学習します。 MisMatchは、エンコーダーと2つのデコーダーで構成されます。 1つのデコーダーは、ラベルのないデータの関心領域(RoI)に対する積極的な注意を学習し、それによってRoIのより高い信頼性の予測を生成します。もう一方のデコーダーは、同じラベルのないデータに対するRoIの否定的な注意を学習するため、信頼性の低い予測が生成されます。次に、デコーダーのペアの予測間に整合性の正則化を適用します。評価のために、最初にCTベースの肺血管セグメンテーションタスクで広範な交差検証を実行し、ラベル全体の6.25%のみが使用されている場合、MisMatchが最先端の半教師あり手法よりも統計的に優れていることを示します。さらに、ラベル全体の6.25%を使用したMisMatchのパフォーマンスは、利用可能なすべてのラベルを使用する最先端の方法に匹敵します。 2番目の実験では、MisMatchは、MRIベースの脳腫瘍セグメンテーションタスクで最先端の方法よりも優れています。
The lack of labels is one of the fundamental constraints in deep learning based methods for image classification and segmentation, especially in applications such as medical imaging. Semi-supervised learning (SSL) is a promising method to address the challenge of labels carcity. The state-of-the-art SSL methods utilise consistency regularisation to learn unlabelled predictions which are invariant to perturbations on the prediction confidence. However, such SSL approaches rely on hand-crafted augmentation techniques which could be sub-optimal. In this paper, we propose MisMatch, a novel consistency based semi-supervised segmentation method. MisMatch automatically learns to produce paired predictions with increasedand decreased confidences. MisMatch consists of an encoder and two decoders. One decoder learns positive attention for regions of interest (RoI) on unlabelled data thereby generating higher confidence predictions of RoI. The other decoder learns negative attention for RoI on the same unlabelled data thereby generating lower confidence predictions. We then apply a consistency regularisation between the paired predictions of the decoders. For evaluation, we first 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. Furthermore MisMatch performance using 6.25% ofthe total labels is comparable to state-of-the-art methodsthat utilise all available labels. In a second experiment, MisMatch outperforms state-of-the-art methods on an MRI-based brain tumour segmentation task.
updated: Sat Oct 23 2021 09:22:41 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アソシエイト