対照学習(CL)は、医療画像セグメンテーションのコンテキストで専門家の注釈に依存することなく、有用な表現を学習することを目的としています。既存のアプローチは、すべての入力特徴を同じ定数ベクトルにマッピングするだけで、主に単一の正のベクトル(つまり、同じ画像の拡張)をバッチの残りの部分全体の負のセットと対比します。印象的な経験的パフォーマンスにもかかわらず、これらの方法には次の欠点があります。(1)問題が簡単な解決策に崩壊するのを防ぐことは依然として手ごわい課題です。 (2)同じ画像に異なる解剖学的構造が存在するため、同じ画像内のすべてのボクセルが等しく陽性であるとは限らないと主張します。この作業では、特徴とバッチ次元の両方に沿って3D空間コンテキストと豊富な解剖学的情報をキャプチャすることにより、低レベルと高レベルの特徴を効果的に学習するための新しい対照ボクセルワイズ表現学習(CVRL)メソッドを紹介します。具体的には、最初に、3D表現の次元間で機能の多様性を促進するための新しいCL戦略を紹介します。 3D画像の2レベルの対照最適化(つまり、低レベルと高レベル)を通じてフレームワークをトレーニングします。 2つのベンチマークデータセットと異なるラベル付き設定での実験は、提案されたフレームワークの優位性を示しています。さらに重要なことは、私たちの方法が標準的なCLアプローチから硬度を意識した特性の利点を継承していることも証明しています。
Contrastive learning (CL) aims to learn useful representation without relying on expert annotations in the context of medical image segmentation. Existing approaches mainly contrast a single positive vector (i.e., an augmentation of the same image) against a set of negatives within the entire remainder of the batch by simply mapping all input features into the same constant vector. Despite the impressive empirical performance, those methods have the following shortcomings: (1) it remains a formidable challenge to prevent the collapsing problems to trivial solutions; and (2) we argue that not all voxels within the same image are equally positive since there exist the dissimilar anatomical structures with the same image. In this work, we present a novel Contrastive Voxel-wise Representation Learning (CVRL) method to effectively learn low-level and high-level features by capturing 3D spatial context and rich anatomical information along both the feature and the batch dimensions. Specifically, we first introduce a novel CL strategy to ensure feature diversity promotion among the 3D representation dimensions. We train the framework through bi-level contrastive optimization (i.e., low-level and high-level) on 3D images. Experiments on two benchmark datasets and different labeled settings demonstrate the superiority of our proposed framework. More importantly, we also prove that our method inherits the benefit of hardness-aware property from the standard CL approaches.