arXiv reaDer
半教師あり医用画像セグメンテーションのための特徴表現学習の探索
Exploring Feature Representation Learning for Semi-supervised Medical Image Segmentation
この論文では、半教師あり医用画像セグメンテーションのためのシンプルかつ効果的な 2 段階のフレームワークを紹介します。一貫性正則化や擬似ラベル付けなど、主に予測に直接基づく擬似監視に依存する従来の最先端の半教師ありセグメンテーション手法とは異なり、私たちの重要な洞察は、ラベル付きおよびラベルなし(つまり、擬似)を使用した特徴表現学習を探索することです。ラベル付き)画像を使用して、よりコンパクトで分離性の高い特徴空間を正規化します。これにより、低密度の決定境界学習への道が開かれ、セグメンテーションのパフォーマンスが向上します。第 1 段階でラベル付き画像を利用する境界を意識したコントラスト損失と、第 2 段階でラベル付き画像と擬似ラベル付き画像の両方を最適化するプロトタイプ認識型コントラスト損失を含む、段階適応型コントラスト学習方法が提案されています。 。プロトタイプを意識した対比学習において重要な役割を果たす、より正確なプロトタイプ推定を取得するために、より高品質の疑似ラベルを生成するための偶然の不確実性を認識する手法、すなわち AUA を提案します。 AUA は、画像の曖昧性を利用して、予測の一貫性を適応的に正規化します。画像の曖昧さは、その重要性にもかかわらず、既存の研究では十分に検討されていません。私たちの方法は、3 つの公的医療画像セグメンテーション ベンチマークで最良の結果を達成しました。
This paper presents a simple yet effective two-stage framework for semi-supervised medical image segmentation. Unlike prior state-of-the-art semi-supervised segmentation methods that predominantly rely on pseudo supervision directly on predictions, such as consistency regularization and pseudo labeling, our key insight is to explore the feature representation learning with labeled and unlabeled (i.e., pseudo labeled) images to regularize a more compact and better-separated feature space, which paves the way for low-density decision boundary learning and therefore enhances the segmentation performance. A stage-adaptive contrastive learning method is proposed, containing a boundary-aware contrastive loss that takes advantage of the labeled images in the first stage, as well as a prototype-aware contrastive loss to optimize both labeled and pseudo labeled images in the second stage. To obtain more accurate prototype estimation, which plays a critical role in prototype-aware contrastive learning, we present an aleatoric uncertainty-aware method, namely AUA, to generate higher-quality pseudo labels. AUA adaptively regularizes prediction consistency by taking advantage of image ambiguity, which, given its significance, is under-explored by existing works. Our method achieves the best results on three public medical image segmentation benchmarks.
updated: Sun Jul 30 2023 13:54:47 GMT+0000 (UTC)
published: Mon Nov 22 2021 05:06:12 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アソシエイト