arXiv reaDer
解剖学を意識した対照的蒸留による半教師付き医用画像セグメンテーションのブートストラップ
Bootstrapping Semi-supervised Medical Image Segmentation with Anatomical-aware Contrastive Distillation
対照的な学習は、医療画像セグメンテーションのコンテキストでの注釈不足の問題に対して大きな期待を示しています。既存のアプローチは、通常、ラベル付けされた医療画像とラベル付けされていない医療画像の両方に対してバランスの取れたクラス分布を前提としています。しかし、実際の医用画像データは一般的に不均衡 (つまり、マルチクラス ラベルの不均衡) であるため、自然に輪郭がぼやけ、まれなオブジェクトに誤ってラベルを付けることがよくあります。さらに、すべての陰性サンプルが等しく陰性であるかどうかは不明のままです。この作業では、半教師付き医用画像セグメンテーションのための、解剖学を意識した対照的蒸留フレームワークである ACTION を提示します。具体的には、最初に、ポジティブとネガティブのペア間のバイナリ監視ではなく、ネガをソフトにラベル付けすることにより、反復対照蒸留アルゴリズムを開発します。また、サンプリングされたデータの多様性を強化するために、ポジティブ セットと比較して、ランダムに選択されたネガティブ セットからより意味的に類似した特徴をキャプチャします。次に、より重要な問題を提起します。不均衡なサンプルを実際に処理して、パフォーマンスを向上させることができるでしょうか?したがって、ACTION の重要なイノベーションは、データセット全体にわたるグローバルな意味関係と、隣接するピクセル間の局所的な解剖学的特徴を最小限の追加メモリ フットプリントで学習することです。トレーニング中に、ハード ネガティブ ピクセルのまばらなセットを積極的にサンプリングすることで、解剖学的コントラストを導入します。これにより、セグメンテーション境界がより滑らかになり、より正確な予測が生成されます。 2 つのベンチマーク データセットとさまざまなラベルのない設定での広範な実験により、ACTION が現在の最先端の半教師付きメソッドよりも大幅に優れていることが示されています。
Contrastive learning has shown great promise over annotation scarcity problems in the context of medical image segmentation. Existing approaches typically assume a balanced class distribution for both labeled and unlabeled medical images. However, medical image data in reality is commonly imbalanced (i.e., multi-class label imbalance), which naturally yields blurry contours and usually incorrectly labels rare objects. Moreover, it remains unclear whether all negative samples are equally negative. In this work, we present ACTION, an Anatomical-aware ConTrastive dIstillatiON framework, for semi-supervised medical image segmentation. Specifically, we first develop an iterative contrastive distillation algorithm by softly labeling the negatives rather than binary supervision between positive and negative pairs. We also capture more semantically similar features from the randomly chosen negative set compared to the positives to enforce the diversity of the sampled data. Second, we raise a more important question: Can we really handle imbalanced samples to yield better performance? Hence, the key innovation in ACTION is to learn global semantic relationship across the entire dataset and local anatomical features among the neighbouring pixels with minimal additional memory footprint. During the training, we introduce anatomical contrast by actively sampling a sparse set of hard negative pixels, which can generate smoother segmentation boundaries and more accurate predictions. Extensive experiments across two benchmark datasets and different unlabeled settings show that ACTION significantly outperforms the current state-of-the-art semi-supervised methods.
updated: Tue Mar 07 2023 02:14:39 GMT+0000 (UTC)
published: Mon Jun 06 2022 01:30:03 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アソシエイト