この論文は、半教師あり医療画像セグメンテーションのためのシンプルで効果的な2段階のフレームワークを提示します。私たちの重要な洞察は、セグメンテーションのパフォーマンスを向上させるために、ラベル付きおよびラベルなし(つまり、疑似ラベル付き)の画像を使用して学習する特徴表現を調査することです。最初の段階では、高品質の疑似ラベルを生成するためのセグメンテーションパフォーマンスを向上させるために、不確実性を意識したエアロティック手法、つまりAUAを紹介します。医用画像に固有のあいまいさを考慮して、AUAはあいまいさの少ない画像の一貫性を適応的に正規化します。表現学習を強化するために、最初の段階でラベル付き画像を正則化する境界認識対照損失と、2番目の段階でラベル付き画像と疑似ラベル画像の両方を最適化するプロトタイプ認識対照損失を含む段階適応対照学習法を提案します。ステージ。境界を意識したコントラスト損失は、計算コストを削減するために、セグメンテーション境界周辺のピクセルのみを最適化します。プロトタイプ対応のコントラスト損失は、各クラスの重心を構築してペアワイズ比較の計算コストを削減することにより、ラベル付き画像と疑似ラベル付き画像の両方を完全に活用します。私たちの方法は、2つの公的医療画像セグメンテーションベンチマークで最良の結果を達成します。特に、私たちの方法は、わずか5%のラベル付き画像に依存して、結腸腫瘍のセグメンテーションのDiceで以前の最先端技術を5.7%上回っています。
This paper presents a simple yet effective two-stage framework for semi-supervised medical image segmentation. Our key insight is to explore the feature representation learning with labeled and unlabeled (i.e., pseudo labeled) images to enhance the segmentation performance. In the first stage, we present an aleatoric uncertainty-aware method, namely AUA, to improve the segmentation performance for generating high-quality pseudo labels. Considering the inherent ambiguity of medical images, AUA adaptively regularizes the consistency on images with low ambiguity. To enhance the representation learning, we propose a stage-adaptive contrastive learning method, including a boundary-aware contrastive loss to regularize the labeled images in the first stage and a prototype-aware contrastive loss to optimize both labeled and pseudo labeled images in the second stage. The boundary-aware contrastive loss only optimizes pixels around the segmentation boundaries to reduce the computational cost. The prototype-aware contrastive loss fully leverages both labeled images and pseudo labeled images by building a centroid for each class to reduce computational cost for pair-wise comparison. Our method achieves the best results on two public medical image segmentation benchmarks. Notably, our method outperforms the prior state-of-the-art by 5.7% on Dice for colon tumor segmentation relying on just 5% labeled images.