LLM2D
改进医学图像配准基础模型的泛化能力
Improving Generalization of Medical Image Registration Foundation Model
作者: Jing Hu, Kaiwei Yu, Hongjiang Xian, Shu Hu, Xin Wang
发布日期: 5/13/2025
arXiv ID: oai:arXiv.org:2505.06527v1

摘要

arXiv:2505.06527v1 类型:跨领域 摘要:可变形配准是医学图像处理中的一个基本任务,旨在通过建立图像之间的非线性对应关系实现精确对齐。传统方法具有良好的适应性和可解释性,但受限于计算效率。尽管深度学习方法在提高配准速度和准确性方面取得了显著进步,但在不同数据集和任务上的灵活性和通用性方面仍然不足。近年来,基础模型作为一种有前途的方向出现,利用大型和多样化的数据集学习图像配准的通用特征和变换模式,从而展示了强大的跨任务迁移能力。然而,这些模型在面对新颖的解剖结构、变化的成像条件或未见的模态时,仍然存在泛化能力和鲁棒性方面的问题。为了解决这些局限性,本文将Sharpness-Aware Minimization (SAM)融入基础模型,以增强其在医学图像配准中的泛化能力和鲁棒性。通过优化损失景观的平坦性,SAM改进了模型在多样化数据分布下的稳定性,并增强了其处理复杂临床场景的能力。实验结果表明,集成SAM的基础模型在跨数据集配准性能上取得了显著改进,为医学图像配准技术的进步提供了新的见解。我们的代码可从https://github.com/Promise13/fm_sam{https://github.com/Promise13/fm_sam获取。