arXiv reaDer
半教師ありドメインの一貫性による適応型医用画像セグメンテーション 正則化された解きほぐされた対照学習
Semi-supervised Domain Adaptive Medical Image Segmentation through Consistency Regularized Disentangled Contrastive Learning
教師なしドメイン適応 (UDA) はドメイン シフトを軽減する有望な方向性ですが、教師ありのそれには及ばません。この研究では、医療画像セグメンテーションのための比較的研究されていない半教師ありドメイン適応 (SSDA) を調査します。SSDA では、少数のラベル付きターゲット サンプルにアクセスすることで適応パフォーマンスを大幅に向上させることができます。具体的には、2 段階のトレーニング プロセスを提案します。まず、エンコーダーは、ピクセル レベルの特徴の一貫性制約とともに、新しいドメイン コンテンツ解絡対比学習 (CL) を使用して、自己学習パラダイムで事前トレーニングされます。提案された CL は、エンコーダにコンテンツ固有だがドメイン不変のセマンティクスをソース画像とターゲット画像からグローバル スケールで学習させるのに対し、一貫性正則化は空間感度を維持することでローカルなピクセル レベルの情報のマイニングを強制します。この事前トレーニングされたエンコーダーは、デコーダーとともに、半教師あり設定を使用して下流のタスク (ピクセルレベルのセグメンテーション) に合わせてさらに微調整されます。さらに、提案した方法が UDA 設定用に簡単に拡張できることを実験的に検証し、提案した戦略の優位性を高めます。 2 つのドメイン適応型画像セグメンテーション タスクを評価したところ、提案した手法は SSDA 設定と UDA 設定の両方で SoTA 手法よりも優れています。コードは https://github.com/hritam-98/GFDA-disentangled で入手できます。
Although unsupervised domain adaptation (UDA) is a promising direction to alleviate domain shift, they fall short of their supervised counterparts. In this work, we investigate relatively less explored semi-supervised domain adaptation (SSDA) for medical image segmentation, where access to a few labeled target samples can improve the adaptation performance substantially. Specifically, we propose a two-stage training process. First, an encoder is pre-trained in a self-learning paradigm using a novel domain-content disentangled contrastive learning (CL) along with a pixel-level feature consistency constraint. The proposed CL enforces the encoder to learn discriminative content-specific but domain-invariant semantics on a global scale from the source and target images, whereas consistency regularization enforces the mining of local pixel-level information by maintaining spatial sensitivity. This pre-trained encoder, along with a decoder, is further fine-tuned for the downstream task, (i.e. pixel-level segmentation) using a semi-supervised setting. Furthermore, we experimentally validate that our proposed method can easily be extended for UDA settings, adding to the superiority of the proposed strategy. Upon evaluation on two domain adaptive image segmentation tasks, our proposed method outperforms the SoTA methods, both in SSDA and UDA settings. Code is available at https://github.com/hritam-98/GFDA-disentangled
updated: Thu Jul 06 2023 06:13:22 GMT+0000 (UTC)
published: Thu Jul 06 2023 06:13:22 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アソシエイト