通过记忆解剖变异实现可解释的医学图像分割域适应
RemInD: Remembering Anatomical Variations for Interpretable Domain Adaptive Medical Image Segmentation
- 用锚点记忆解剖结构变化,映射图像为加权平均
- 在心脏与腹部数据集上达最优性能,仅用单一对齐策略
- 预测依赖可解释权重向量与空间形变,适合临床可信分析
本文提出一种新的贝叶斯框架,用于无监督域适应下的医学图像分割。现有方法虽尝试通过域对齐提升性能,但缺乏显式且可解释的机制来确保目标域特征捕捉有意义的解剖结构信息,且易受维度诅咒影响,导致可解释性差、计算效率低。为此,我们提出 RemInD,其灵感源于人类认知:学习一个与域无关的潜在流形,以多个锚点记忆解剖变异。通过将图像映射为加权锚点平均,保证预测的真实性和可靠性。该设计模拟人类先构建代表性成分,再从记忆中检索组合指导分割的过程。模型预测由两个可解释因素决定:低维锚点权重向量和空间形变,从而实现高效且符合几何特性的域适应。通过在两个公开数据集(心脏与腹部影像)上的实验验证,RemInD 仅用单一对齐策略即达到当前最优表现,优于依赖多重复杂对齐策略的现有方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。