arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.12868cs.CV

UniReg: 一种用于可控CT图像配准的通用模型

UniReg: Conditional Unified Model for Medical Image Registration

The University of Hong Kong(香港大学);DAMO Academy, Alibaba Group(阿里巴巴集团达摩院);The First Affiliated Hospital of Zhejiang University(浙江大学第一附属医院)

Zi Li, Jianpeng Zhang, Tai Ma, Tony C. W. Mok, Yan-Jie Zhou, Zeli Chen, Xianghua Ye, Le Lu, Cheng Chen, Dakai Jin

AI总结:

提出UniReg,首个条件统一模型,通过整合解剖结构先验、配准类型约束和实例特征,实现多场景CT图像配准的精确性和泛化性。

摘要:

基于学习的医学图像配准在匹配传统方法精度的同时,提供了优越的计算效率。然而,现有方法在不同临床场景中泛化能力差,需要为特定配准任务(如个体间/个体内配准或解剖区域特定对齐)开发多个孤立的网络,导致开发流程繁琐。为克服这一局限,我们提出了UniReg,首个用于多场景CT图像配准的条件统一模型,它结合了任务特定学习方法的精度优势和传统优化方法的泛化能力。我们的关键创新是一个统一的配准框架,该框架根据以下条件自适应估计变形场:(1)解剖结构先验,(2)配准类型约束(个体间/个体内),以及(3)实例特定特征,从而在单个模型中实现跨异构场景的最优对齐。通过在多个CT/MR配准数据集上的全面实验,UniReg相比当前最先进的基于学习方法取得了更优的平均配准精度,同时展现出强大的跨场景泛化能力。此外,通过用一个紧凑的统一模型替代多个孤立的任务特定模型,UniReg显著降低了总体训练负担,包括总训练成本和模型冗余。

英文摘要:

Learning-based medical image registration has matched the accuracy of conventional methods while offering superior computational efficiency. However, existing approaches suffer from poor generalization across diverse clinical scenarios, requiring the laborious development of multiple isolated networks for specific registration tasks, \emph{e.g.}, inter-/intra-subject registration or anatomical region-specific alignment, leading to cumbersome development pipelines. To overcome this limitation, we propose \textbf{UniReg}, the first conditional unified model for multi-scenario medical image registration, which combines the precision advantages of task-specific learning methods with the generalization of traditional optimization methods. Our key innovation is a unified registration framework that adaptively estimates deformation fields conditioned on: (1) anatomical structure priors, (2) registration type constraints (inter/intra-subject), and (3) instance-specific features, enabling effective alignment across heterogeneous CT and MR registration scenarios within a single model. Through comprehensive experiments on multiple CT/MR registration datasets, UniReg achieves superior average registration accuracy compared with current state-of-the-art learning-based methods while exhibiting strong cross-scenario generalization. Moreover, by replacing multiple isolated task-specific models with a compact unified model, UniReg substantially reduces the overall training burden in terms of total training cost and model redundancy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑