分布マッチング損失の最小化は、画像分類のコンテキストでのドメイン適応への原則的なアプローチです。ただし、現在敵対的なモデルによって支配されているセグメンテーションネットワークの適応ではほとんど見過ごされています。ラベルのない入力から計算されたいくつかの幾何学的変換まで、ネットワーク出力空間での直接カーネル密度マッチングを促進する損失関数のクラスを提案します。中間ドメイン弁別器を使用するのではなく、私たちの直接的なアプローチは、単一の損失で分布のマッチングとセグメンテーションを統合します。したがって、トレーニングの品質、安定性、効率の両方を向上させながら、余分な敵対的なステップを回避することにより、セグメンテーションの適応を簡素化します。ネットワーク出力空間での敵対的トレーニングを介した最先端のセグメンテーション適応へのアプローチを並置します。異なる磁気共鳴画像(MRI)モダリティ間で脳のセグメンテーションを適応させるという挑戦的なタスクでは、私たちのアプローチは、精度と安定性の両方の点で大幅に優れた結果を達成します。
Minimization of distribution matching losses is a principled approach to domain adaptation in the context of image classification. However, it is largely overlooked in adapting segmentation networks, which is currently dominated by adversarial models. We propose a class of loss functions, which encourage direct kernel density matching in the network-output space, up to some geometric transformations computed from unlabeled inputs. Rather than using an intermediate domain discriminator, our direct approach unifies distribution matching and segmentation in a single loss. Therefore, it simplifies segmentation adaptation by avoiding extra adversarial steps, while improving both the quality, stability and efficiency of training. We juxtapose our approach to state-of-the-art segmentation adaptation via adversarial training in the network-output space. In the challenging task of adapting brain segmentation across different magnetic resonance images (MRI) modalities, our approach achieves significantly better results both in terms of accuracy and stability.