深い畳み込みニューラルネットワークは、医療画像セグメンテーションタスクで卓越したパフォーマンスを示しています。教師あり深層学習方法をトレーニングするときの通常の問題は、取得に時間とコストがかかるラベル付きデータの欠如です。この論文では、限られたラベル付きサンプルと多数のラベルなし画像を使用してセグメンテーションネットワークをトレーニングするための学生と教師のアプローチに基づく新しい不確実性に基づく半教師あり学習を提案します。最初に、ベイズの深層学習を使用して、ラベル付けされたサンプルから教師のセグメンテーションモデルがトレーニングされます。トレーニングされたモデルは、ラベルのないセットのソフトセグメンテーションラベルと不確実性マップを生成するために使用されます。次に、生徒モデルは、ソフトにセグメント化されたサンプルと、新しく設計された損失関数を使用して教師モデルの不確実性から推定されたセグメント化品質の対応するピクセル単位の信頼度を使用して更新されます。網膜層セグメンテーションタスクに関する実験結果は、提案された方法が完全に監視されたアプローチと比較してセグメンテーション性能を改善し、専門家のアノテーターと同等であることを示しています。提案された半教師ありセグメンテーションフレームワークは重要な貢献であり、注釈付きの医用画像へのアクセスが困難なさまざまな画像モダリティにわたる生物医学的画像セグメンテーションに適用できます。
Deep convolutional neural networks have shown outstanding performance in medical image segmentation tasks. The usual problem when training supervised deep learning methods is the lack of labeled data which is time-consuming and costly to obtain. In this paper, we propose a novel uncertainty-guided semi-supervised learning based on a student-teacher approach for training the segmentation network using limited labeled samples and a large number of unlabeled images. First, a teacher segmentation model is trained from the labeled samples using Bayesian deep learning. The trained model is used to generate soft segmentation labels and uncertainty maps for the unlabeled set. The student model is then updated using the softly segmented samples and the corresponding pixel-wise confidence of the segmentation quality estimated from the uncertainty of the teacher model using a newly designed loss function. Experimental results on a retinal layer segmentation task show that the proposed method improves the segmentation performance in comparison to the fully supervised approach and is on par with the expert annotator. The proposed semi-supervised segmentation framework is a key contribution and applicable for biomedical image segmentation across various imaging modalities where access to annotated medical images is challenging