医療画像分析における自動セグメンテーションは、手動でラベル付けされた大量のデータを必要とする困難な作業です。ただし、医療データに手動で注釈を付けるのは面倒なことが多く、既存の学習ベースのアプローチのほとんどは、効果的な幾何学的制約なしにオブジェクトの境界を正確に描くことができません。自己監視学習のサブエリアである対照学習は、最近、複数のアプリケーション分野で有望な方向性として注目されています。この作業では、限られた注釈でボリューム医療画像セグメンテーションのグローバルローカル視覚表現を学習するための幾何学的制約を持つ新しい対照ボクセルワイズ表現蒸留(CVRD)メソッドを提示します。私たちのフレームワークは、3D空間コンテキストと豊富な解剖学的情報をキャプチャすることにより、グローバルおよびローカルの機能を効果的に学習できます。具体的には、3D画像からグローバル情報を学習するためのボクセルからボリュームへの対比アルゴリズムを導入し、埋め込みスペースでローカルキューを明示的に利用するためにローカルボクセルからボクセルへの蒸留を実行することを提案します。さらに、弾性相互作用ベースのアクティブコンターモデルを幾何学的正則化項として統合し、エンドツーエンドの学習方法で高速で信頼性の高いオブジェクトの描写を可能にします。心房セグメンテーションチャレンジデータセットの結果は、特に注釈付きデータの数が非常に限られている設定で、提案されたスキームの優位性を示しています。コードはhttps://github.com/charlesyou999648/CVRDで入手できます。
Automated segmentation in medical image analysis is a challenging task that requires a large amount of manually labeled data. However, manually annotating medical data is often laborious, and most existing learning-based approaches fail to accurately delineate object boundaries without effective geometric constraints. Contrastive learning, a sub-area of self-supervised learning, has recently been noted as a promising direction in multiple application fields. In this work, we present a novel Contrastive Voxel-wise Representation Distillation (CVRD) method with geometric constraints to learn global-local visual representations for volumetric medical image segmentation with limited annotations. Our framework can effectively learn global and local features by capturing 3D spatial context and rich anatomical information. Specifically, we introduce a voxel-to-volume contrastive algorithm to learn global information from 3D images, and propose to perform local voxel-to-voxel distillation to explicitly make use of local cues in the embedding space. Moreover, we integrate an elastic interaction-based active contour model as a geometric regularization term to enable fast and reliable object delineations in an end-to-end learning manner. Results on the Atrial Segmentation Challenge dataset demonstrate superiority of our proposed scheme, especially in a setting with a very limited number of annotated data. The code will be available at https://github.com/charlesyou999648/CVRD.