arXiv reaDer
半教師あり医用画像セグメンテーションの再考: 分散削減の観点
Rethinking Semi-Supervised Medical Image Segmentation: A Variance-Reduction Perspective
医用画像のセグメンテーションでは、意味的に似ているサンプルと似ていないサンプルのペアを対比することで、視覚的表現の品質を向上させるための対比学習が支配的な方法です。これは、グラウンド トゥルース ラベルにアクセスせずに、真に異なる解剖学的特徴を持つ負の例をサンプリングすると、パフォーマンスが大幅に向上するという観測によって可能になります。ただし、実際には、これらのサンプルは同様の解剖学的特徴に由来する可能性があり、モデルは少数の尾部クラスのサンプルを区別するのに苦労する可能性があり、尾部クラスが誤分類されやすくなり、どちらも通常モデルの崩壊につながります。この論文では、医用画像セグメンテーションにおける層化グループサンプリング理論を備えた半教師付き対照学習(CL)フレームワークであるARCOを提案します。具体的には、最初に、分散削減推定の概念を通じて ARCO を構築することを提案し、特定の分散削減手法が、非常に限られたラベルを持つ医用画像セグメンテーション タスクで特に有益であることを示します。さらに、これらのサンプリング手法が分散削減において普遍的であることを理論的に証明します。最後に、ラベル設定が異なる 3 つのベンチマーク データセットでアプローチを実験的に検証し、方法は最先端の半教師付き方法より一貫して優れています。さらに、これらのサンプリング手法を使用して CL フレームワークを強化し、以前の方法よりも大幅に向上することを示します。私たちの仕事は、医療画像分析タスクを達成するための現在の自己管理目標の制限を定量化することにより、半教師付き医療画像セグメンテーションに向けた重要なステップであると考えています。
For medical image segmentation, contrastive learning is the dominant practice to improve the quality of visual representations by contrasting semantically similar and dissimilar pairs of samples. This is enabled by the observation that without accessing ground truth label, negative examples with truly dissimilar anatomical features, if sampled, can significantly improve the performance. In reality, however, these samples may come from similar anatomical features and the models may struggle to distinguish the minority tail-class samples, making the tail classes more prone to misclassification, both of which typically lead to model collapse. In this paper, we propose ARCO, a semi-supervised contrastive learning (CL) framework with stratified group sampling theory in medical image segmentation. In particular, we first propose building ARCO through the concept of variance-reduced estimation, and show that certain variance-reduction techniques are particularly beneficial in medical image segmentation tasks with extremely limited labels. Furthermore, we theoretically prove these sampling techniques are universal in variance reduction. Finally, we experimentally validate our approaches on three benchmark datasets with different label settings, and our methods consistently outperform state-of-the-art semi-supervised methods. Additionally, we augment the CL frameworks with these sampling techniques and demonstrate significant gains over previous methods. We believe our work is an important step towards semi-supervised medical image segmentation by quantifying the limitation of current self-supervision objectives for accomplishing medical image analysis tasks.
updated: Sat Mar 11 2023 04:19:13 GMT+0000 (UTC)
published: Fri Feb 03 2023 13:50:25 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アソシエイト