生物医学画像セグメンテーションにおけるディープラーニングの広範な成功により、2つのドメイン間のギャップが異種機能を備えた目に見えないデータに展開されるとモデルのパフォーマンスに深刻な影響を与える可能性があるため、ドメインシフトは重大で困難な問題になります。この問題を軽減するために、クロスモダリティ生物医学画像セグメンテーションのために、ラベル付きソースドメインから学習したモデルをラベルなしターゲットドメインに一般化するための、新しい教師なしドメイン適応ネットワークを提示します。具体的には、私たちのアプローチは、条件付きドメインディスクリミネーター〜(CDD)とカテゴリー中心のプロトタイプアライナー〜(CCPA)の2つの主要モジュールで構成されています。分類器タスクの条件付きドメイン敵対的ネットワークから拡張されたCDDは、複雑なクロスモダリティ生物医学画像の処理に効果的かつ堅牢です。クロスドメインオブジェクト検出におけるグラフ誘導プロトタイプアライメントメカニズムから改善されたCCPAは、精巧なプロトタイプ表現を通じて正確なインスタンスレベルの機能を活用できます。さらに、エントロピーベースの損失を介してクラスの不均衡の悪影響に対処できます。心臓下部構造セグメンテーションタスクの公開ベンチマークに関する広範な実験は、私たちの方法がターゲットドメインでのパフォーマンスを大幅に改善することを示しています。
With the widespread success of deep learning in biomedical image segmentation, domain shift becomes a critical and challenging problem, as the gap between two domains can severely affect model performance when deployed to unseen data with heterogeneous features. To alleviate this problem, we present a novel unsupervised domain adaptation network, for generalizing models learned from the labeled source domain to the unlabeled target domain for cross-modality biomedical image segmentation. Specifically, our approach consists of two key modules, a conditional domain discriminator~(CDD) and a category-centric prototype aligner~(CCPA). The CDD, extended from conditional domain adversarial networks in classifier tasks, is effective and robust in handling complex cross-modality biomedical images. The CCPA, improved from the graph-induced prototype alignment mechanism in cross-domain object detection, can exploit precise instance-level features through an elaborate prototype representation. In addition, it can address the negative effect of class imbalance via entropy-based loss. Extensive experiments on a public benchmark for the cardiac substructure segmentation task demonstrate that our method significantly improves performance on the target domain.