about
Constrained Domain Adaptation for Segmentation (arxiv.org)
2 points by sel1 on Aug 12, 2019 | hide | past | pdf | discuss on HN

In plain words: It trains a segmentation network on unlabeled scans by enforcing size or shape rules on its guesses, in one loss, not the usual adversarial training. On spine MRI across scan types, it beat adversarial methods and neared fully supervised accuracy despite rough size rules.

Abstract · Constrained domain adaptation for Image segmentation

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.

Mathilde Bateson, Jose Dolz, Hoel Kervadec, Hervé Lombaert, Ismail Ben Ayed
arXiv:1908.02996 · cs.CV · submitted Aug 8, 2019 · updated Apr 11, 2021
abstract · pdf · html · Published in IEEE Transactions on Medical Imaging. First version in MICCAI 2019

add comment on HN