arXiv:2412.17982cs.CV2024-12中稿 · Medical Image Anal…被引 2

让深度学习自动学会不同部位用不同变形强度,提升医学图像配准精度。

Unsupervised learning of spatially varying regularization for diffeomorphic image registration

  • 基于概率模型,从数据中端到端学习空间可变的正则化强度
  • 在多个公开数据集上显著提升配准性能,同时保持形变平滑
  • 无需手动调参,支持多种网络结构,适合医学影像研究者

空间可变正则化能适应解剖区域在可变形图像配准中的形变差异。传统优化方法已利用空间可变正则化处理解剖细节,但多数现代深度学习模型倾向使用空间不变正则化,即在整个图像上施加统一的正则化强度,可能忽略局部变化。本文提出一种分层概率模型,对形变正则化强度引入先验分布,实现从数据中端到端学习空间可变的形变正则化器。该方法易于实现,可与多种配准网络架构集成。通过贝叶斯优化实现超参数自动调优,高效确定任意配准任务的最佳超参数。在多个公开数据集上的全面评估表明,该方法显著提升配准性能,增强深度学习配准的可解释性,同时保持形变光滑。

原文摘要 · Abstract (English)

Spatially varying regularization accommodates the deformation variations that may be necessary for different anatomical regions during deformable image registration. Historically, optimization-based registration models have harnessed spatially varying regularization to address anatomical subtleties. However, most modern deep learning-based models tend to gravitate towards spatially invariant regularization, wherein a homogenous regularization strength is applied across the entire image, potentially disregarding localized variations. In this paper, we propose a hierarchical probabilistic model that integrates a prior distribution on the deformation regularization strength, enabling the end-to-end learning of a spatially varying deformation regularizer directly from the data. The proposed method is straightforward to implement and easily integrates with various registration network architectures. Additionally, automatic tuning of hyperparameters is achieved through Bayesian optimization, allowing efficient identification of optimal hyperparameters for any given registration task. Comprehensive evaluations on publicly available datasets demonstrate that the proposed method significantly improves registration performance and enhances the interpretability of deep learning-based registration, all while maintaining smooth deformations.

图像配准深度学习医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。