セグメンテーション領域に関するドメイン不変の事前知識を埋め込む制約付きの定式化でセグメンテーションネットワークを適応させることを提案します。そのような知識は、ソースサンプルから推定された、または先験的に知られている、単純な解剖学的情報、例えば、構造のサイズまたは形状の形をとることができる。私たちの方法は、ラベルのないターゲットサンプルのネットワーク出力にドメイン不変の不等式制約を課します。これは、事前知識の不確実性が許容されるターゲットドメインとソースドメイン間の予測統計を暗黙的に照合します。標準的な確率的勾配降下法に完全に適した微分可能なペナルティで制約された問題に対処し、デュアルプロジェクションを使用した計算コストの高いラグランジュ最適化の必要性を排除します。現在の2段階の敵対的トレーニングとは異なり、私たちの定式化は、単一のネットワークでの単一の損失に基づいています。これにより、トレーニングの収束と品質を向上させながら、余分な敵対的ステップを回避することで適応を簡素化します。私たちのアプローチを最先端の敵対的手法と比較すると、さまざまなMRIモダリティ間で脊椎のセグメンテーションを適応させるという困難なタスクで大幅に優れたパフォーマンスが得られます。私たちの結果は、サイズの事前の不正確さに対するロバスト性も示しており、ターゲットドメインで直接トレーニングされた完全に監視されたモデルの精度に近づいています。私たちの方法は、さまざまな制約やセグメンテーションの問題に簡単に使用できます。
We propose to adapt segmentation networks with a constrained formulation, which embeds domain-invariant prior knowledge about the segmentation regions. Such knowledge may take the form of simple anatomical information, e.g., structure size or shape, estimated from source samples or known a priori. Our method imposes domain-invariant inequality constraints on the network outputs of unlabeled target samples. It implicitly matches prediction statistics between target and source domains with permitted uncertainty of prior knowledge. We address our constrained problem with a differentiable penalty, fully suited for standard stochastic gradient descent approaches, removing the need for computationally expensive Lagrangian optimization with dual projections. Unlike current two-step adversarial training, our formulation is based on a single loss in a single network, which simplifies adaptation by avoiding extra adversarial steps, while improving convergence and quality of training. The comparison of our approach with state-of-the-art adversarial methods reveals substantially better performance on the challenging task of adapting spine segmentation across different MRI modalities. Our results also show a robustness to imprecision of size priors, approaching the accuracy of a fully supervised model trained directly in a target domain.Our method can be readily used for various constraints and segmentation problems.