arXiv reaDer
半教師付き病変セグメンテーションのためのカットアンドペースト一貫性学習
Cut-Paste Consistency Learning for Semi-Supervised Lesion Segmentation
半教師あり学習は、データを大量に消費するディープ ニューラル ネットワークのトレーニングのデータ効率を改善する可能性を秘めています。これは、ラベル付けされたデータが不足している医療画像分析タスクにとって特に重要です。この作業では、カットペーストの増強と一貫性の正則化のアイデアに基づいて、病変セグメンテーション タスクの単純な半教師あり学習方法を提示します。ラベル付けされたデータで利用可能なマスク情報を活用することにより、ラベル付けされていない画像から部分的にラベル付けされたサンプルを合成して、通常の教師あり学習目標 (バイナリ クロス エントロピーなど) を適用できるようにします。さらに、合成画像のラベル付けされていない背景領域でのトレーニングを正規化するために、背景の一貫性用語を導入します。眼底写真データセットと脳 CT スキャン データセットを含む 2 つの公開病変セグメンテーション データセットに対する提案された方法の有効性を経験的に検証します。実験結果は、洗練されたネットワークコンポーネントを導入することなく、私たちの方法が他の自己訓練および一貫性ベースの方法よりも一貫した優れたパフォーマンスを達成することを示しています。
Semi-supervised learning has the potential to improve the data-efficiency of training data-hungry deep neural networks, which is especially important for medical image analysis tasks where labeled data is scarce. In this work, we present a simple semi-supervised learning method for lesion segmentation tasks based on the ideas of cut-paste augmentation and consistency regularization. By exploiting the mask information available in the labeled data, we synthesize partially labeled samples from the unlabeled images so that the usual supervised learning objective (e.g., binary cross entropy) can be applied. Additionally, we introduce a background consistency term to regularize the training on the unlabeled background regions of the synthetic images. We empirically verify the effectiveness of the proposed method on two public lesion segmentation datasets, including an eye fundus photograph dataset and a brain CT scan dataset. The experiment results indicate that our method achieves consistent and superior performance over other self-training and consistency-based methods without introducing sophisticated network components.
updated: Sat Oct 01 2022 04:43:54 GMT+0000 (UTC)
published: Sat Oct 01 2022 04:43:54 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アソシエイト