arXiv reaDer
不確実性に基づく半教師ありセグメンテーションにおけるラベリング表現の活用
Leveraging Labeling Representations in Uncertainty-based Semi-supervised Segmentation
半教師ありセグメンテーションは、ラベル付けされていないデータと少量のラベル付けされたデータを活用することにより、注釈の不足に対処します。ラベルのないデータを利用する主な方法は、教師が生徒のセグメンテーションをガイドする教師と生徒のネットワークを一般的に使用する一貫性トレーニングによるものです。ラベルのないデータの予測は信頼できないため、意味のある信頼できる予測から徐々に学習するために、不確実性を意識した方法が提案されています。ただし、不確実性の推定は、トレーニングステップごとに計算する必要があるモデル予測からの複数の推論に依存しており、計算コストが高くなります。この作品は、セグメンテーションマスクのラベリング表現を活用することにより、ピクセルレベルの不確実性を推定するための新しい方法を提案します。一方では、ラベリング表現は、利用可能なセグメンテーションマスクを表すために学習されます。学習したラベリング表現は、セグメンテーションの予測を一連のもっともらしいマスクにマッピングするために使用されます。このような再構築されたセグメンテーションマスクは、セグメンテーションネットワークを導くピクセルレベルの不確実性を推定するのに役立ちます。提案された方法は、ラベリング表現からの単一の推論で不確実性を推定し、それによって全体の計算を減らします。 MRIで左心房の3Dセグメンテーションに関する方法を評価し、ラベリング表現からの不確実性の推定により、最先端の方法よりもセグメンテーションの精度が向上することを示します。
Semi-supervised segmentation tackles the scarcity of annotations by leveraging unlabeled data with a small amount of labeled data. A prominent way to utilize the unlabeled data is by consistency training which commonly uses a teacher-student network, where a teacher guides a student segmentation. The predictions of unlabeled data are not reliable, therefore, uncertainty-aware methods have been proposed to gradually learn from meaningful and reliable predictions. Uncertainty estimation, however, relies on multiple inferences from model predictions that need to be computed for each training step, which is computationally expensive. This work proposes a novel method to estimate the pixel-level uncertainty by leveraging the labeling representation of segmentation masks. On the one hand, a labeling representation is learnt to represent the available segmentation masks. The learnt labeling representation is used to map the prediction of the segmentation into a set of plausible masks. Such a reconstructed segmentation mask aids in estimating the pixel-level uncertainty guiding the segmentation network. The proposed method estimates the uncertainty with a single inference from the labeling representation, thereby reducing the total computation. We evaluate our method on the 3D segmentation of left atrium in MRI, and we show that our uncertainty estimates from our labeling representation improve the segmentation accuracy over state-of-the-art methods.
updated: Thu Mar 10 2022 23:49:43 GMT+0000 (UTC)
published: Thu Mar 10 2022 23:49:43 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アソシエイト