arXiv reaDer
医療画像セグメンテーションのためのドメイン適応に向けた自己ペースの対照学習のマージン保存
Margin Preserving Self-paced Contrastive Learning Towards Domain Adaptation for Medical Image Segmentation
教師なしドメイン適応(UDA)でソースドメインとターゲットドメインの間のギャップを埋めるために、最も一般的な戦略は、敵対的学習を通じて特徴空間の周辺分布を一致させることに焦点を当てています。ただし、このようなカテゴリに依存しないグローバルアラインメントでは、クラスレベルの同時分布を活用できないため、アラインされた分布の識別性が低下します。この問題に対処するために、この論文では、クロスモーダル医療画像セグメンテーションのための新しいマージン保存自己ペース対照学習(MPSCL)モデルを提案します。対照学習における対照ペアの従来の構築とは異なり、ドメイン適応カテゴリのプロトタイプは、正と負のサンプルペアを構成するために利用されます。漸進的に洗練されたセマンティックプロトタイプのガイダンスにより、対照的な損失を維持する新しいマージンが、埋め込まれた表現空間の識別可能性を高めるために提案されます。対照的な学習の監視を強化するために、より有益な疑似ラベルが自分のペースでターゲットドメインに生成されるため、UDAのカテゴリを意識した分布の調整に役立ちます。さらに、ドメイン不変表現は、2つのドメイン間の共同対照学習を通じて学習されます。クロスモーダル心臓セグメンテーションタスクに関する広範な実験は、MPSCLがセマンティックセグメンテーションのパフォーマンスを大幅に改善し、さまざまな最先端の方法を大幅に上回っていることを示しています。
To bridge the gap between the source and target domains in unsupervised domain adaptation (UDA), the most common strategy puts focus on matching the marginal distributions in the feature space through adversarial learning. However, such category-agnostic global alignment lacks of exploiting the class-level joint distributions, causing the aligned distribution less discriminative. To address this issue, we propose in this paper a novel margin preserving self-paced contrastive Learning (MPSCL) model for cross-modal medical image segmentation. Unlike the conventional construction of contrastive pairs in contrastive learning, the domain-adaptive category prototypes are utilized to constitute the positive and negative sample pairs. With the guidance of progressively refined semantic prototypes, a novel margin preserving contrastive loss is proposed to boost the discriminability of embedded representation space. To enhance the supervision for contrastive learning, more informative pseudo-labels are generated in target domain in a self-paced way, thus benefiting the category-aware distribution alignment for UDA. Furthermore, the domain-invariant representations are learned through joint contrastive learning between the two domains. Extensive experiments on cross-modal cardiac segmentation tasks demonstrate that MPSCL significantly improves semantic segmentation performance, and outperforms a wide variety of state-of-the-art methods by a large margin.
updated: Sun Jan 16 2022 13:16:33 GMT+0000 (UTC)
published: Mon Mar 15 2021 15:23:10 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アソシエイト