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 performs comparable or better than the current state-of-the-art supervised and semi-supervised methods. Our code and models will be publicly available.
updated: Mon Jun 06 2022 01:30:03 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アソシエイト