RemInD: Remembering Anatomical Variations for Interpretable Domain Adaptive Medical Image Segmentation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Xin, Guo, Yin, Zhang, Kaiyu, Balu, Niranjan, Mossa-Basha, Mahmud, Shapiro, Linda, Yuan, Chun
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917924500930560
author Wang, Xin
Guo, Yin
Zhang, Kaiyu
Balu, Niranjan
Mossa-Basha, Mahmud
Shapiro, Linda
Yuan, Chun
author_facet Wang, Xin
Guo, Yin
Zhang, Kaiyu
Balu, Niranjan
Mossa-Basha, Mahmud
Shapiro, Linda
Yuan, Chun
contents This work presents a novel Bayesian framework for unsupervised domain adaptation (UDA) in medical image segmentation. While prior works have explored this clinically significant task using various strategies of domain alignment, they often lack an explicit and explainable mechanism to ensure that target image features capture meaningful structural information. Besides, these methods are prone to the curse of dimensionality, inevitably leading to challenges in interpretability and computational efficiency. To address these limitations, we propose RemInD, a framework inspired by human adaptation. RemInD learns a domain-agnostic latent manifold, characterized by several anchors, to memorize anatomical variations. By mapping images onto this manifold as weighted anchor averages, our approach ensures realistic and reliable predictions. This design mirrors how humans develop representative components to understand images and then retrieve component combinations from memory to guide segmentation. Notably, model prediction is determined by two explainable factors: a low-dimensional anchor weight vector, and a spatial deformation. This design facilitates computationally efficient and geometry-adherent adaptation by aligning weight vectors between domains on a probability simplex. Experiments on two public datasets, encompassing cardiac and abdominal imaging, demonstrate the superiority of RemInD, which achieves state-of-the-art performance using a single alignment approach, outperforming existing methods that often rely on multiple complex alignment strategies.
format Preprint
id arxiv_https___arxiv_org_abs_2502_10887
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RemInD: Remembering Anatomical Variations for Interpretable Domain Adaptive Medical Image Segmentation
Wang, Xin
Guo, Yin
Zhang, Kaiyu
Balu, Niranjan
Mossa-Basha, Mahmud
Shapiro, Linda
Yuan, Chun
Computer Vision and Pattern Recognition
This work presents a novel Bayesian framework for unsupervised domain adaptation (UDA) in medical image segmentation. While prior works have explored this clinically significant task using various strategies of domain alignment, they often lack an explicit and explainable mechanism to ensure that target image features capture meaningful structural information. Besides, these methods are prone to the curse of dimensionality, inevitably leading to challenges in interpretability and computational efficiency. To address these limitations, we propose RemInD, a framework inspired by human adaptation. RemInD learns a domain-agnostic latent manifold, characterized by several anchors, to memorize anatomical variations. By mapping images onto this manifold as weighted anchor averages, our approach ensures realistic and reliable predictions. This design mirrors how humans develop representative components to understand images and then retrieve component combinations from memory to guide segmentation. Notably, model prediction is determined by two explainable factors: a low-dimensional anchor weight vector, and a spatial deformation. This design facilitates computationally efficient and geometry-adherent adaptation by aligning weight vectors between domains on a probability simplex. Experiments on two public datasets, encompassing cardiac and abdominal imaging, demonstrate the superiority of RemInD, which achieves state-of-the-art performance using a single alignment approach, outperforming existing methods that often rely on multiple complex alignment strategies.
title RemInD: Remembering Anatomical Variations for Interpretable Domain Adaptive Medical Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.10887