医療画像セグメンテーションタスクでの深層学習手法の成功には、通常、大量のラベル付きデータが必要です。ただし、信頼できる注釈を取得するには、費用と時間がかかります。半教師あり学習は、取得がはるかに簡単なラベルなしデータを利用することにより、医療画像のセグメンテーションで大きな注目を集めています。本論文では、半教師あり医療画像セグメンテーションのための新しいデュアルタスク相互学習フレームワークを提案します。私たちのフレームワークは、領域ベースの形状制約の学習と境界ベースの表面不一致の学習という2つのタスクに基づく2つの個別のセグメンテーションネットワークの統合として定式化できます。教師と学生のネットワーク間の一方向の転送とは異なり、デュアルタスクの学生のアンサンブルは、トレーニングプロセス中に、互いに協力して暗黙的に有用な知識を探索することができます。ターゲットのセグメンテーション確率マップと符号付き距離マップを共同で学習することにより、フレームワークは幾何学的形状の制約を適用し、より信頼性の高い情報を学習できます。実験結果は、私たちの方法がラベルのないデータを活用することによってパフォーマンスの向上を達成し、最先端の半教師ありセグメンテーション方法よりも優れていることを示しています。
The success of deep learning methods in medical image segmentation tasks usually requires a large amount of labeled data. However, obtaining reliable annotations is expensive and time-consuming. Semi-supervised learning has attracted much attention in medical image segmentation by taking the advantage of unlabeled data which is much easier to acquire. In this paper, we propose a novel dual-task mutual learning framework for semi-supervised medical image segmentation. Our framework can be formulated as an integration of two individual segmentation networks based on two tasks: learning region-based shape constraint and learning boundary-based surface mismatch. Different from the one-way transfer between teacher and student networks, an ensemble of dual-task students can learn collaboratively and implicitly explore useful knowledge from each other during the training process. By jointly learning the segmentation probability maps and signed distance maps of targets, our framework can enforce the geometric shape constraint and learn more reliable information. Experimental results demonstrate that our method achieves performance gains by leveraging unlabeled data and outperforms the state-of-the-art semi-supervised segmentation methods.