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

通过联合优化网络架构与目标函数实现可变形医学图像配准的自动化学习

Automated Learning for Deformable Medical Image Registration by Jointly Optimizing Network Architectures and Objective Functions

Dalian University of Technology(大连理工大学);Guilin University of Electronic Technology(桂林电子科技大学);University of Pennsylvania(宾夕法尼亚大学)

Xin Fan, Zi Li, Ziyang Li, Xiaolin Wang, Risheng Liu, Zhongxuan Luo, Hao Huang

AI总结:

本文提出 AutoReg 自动化配准算法,通过三层框架联合搜索并优化网络架构与训练目标,使临床用户可在多中心医学体数据上获得最先进且更高效的配准模型。

摘要:

可变形图像配准在多种医学图像分析任务中发挥着关键作用。无论是源自传统能量优化还是深度网络,一种成功的配准算法都需要计算机专家付出大量努力,为特定类型的医学数据精心设计配准能量或仔细调节网络架构。为解决上述问题,本文提出了一种自动化学习配准算法(AutoReg),该算法协同优化架构及其相应的训练目标,使非计算机专家(例如医学/临床用户)能够方便地为各种场景找到现成可用的配准算法。具体而言,我们建立了一个三层框架,通过自动搜索机制和协同优化来推导配准网络架构与目标。我们在多中心体数据数据集和多种配准任务上开展了图像配准实验。大量结果表明,我们的 AutoReg 能够针对给定体数据自动学习最优的深度配准网络,并取得最先进的性能,同时相比主流 UNet 架构显著提升了计算效率(在相同配置下,处理一对三维图像的时间从0.558秒缩短至0.270秒)。

英文摘要:

Deformable image registration plays a critical role in various tasks of medical image analysis. A successful registration algorithm, either derived from conventional energy optimization or deep networks requires tremendous efforts from computer experts to well design registration energy or to carefully tune network architectures for the specific type of medical data. To tackle the aforementioned problems, this paper proposes an automated learning registration algorithm (AutoReg) that cooperatively optimizes both architectures and their corresponding training objectives, enabling non-computer experts, e.g., medical/clinical users, to conveniently find off-the-shelf registration algorithms for diverse scenarios. Specifically, we establish a triple-level framework to deduce registration network architectures and objectives with an auto-searching mechanism and cooperating optimization. We conduct image registration experiments on multi-site volume datasets and various registration tasks. Extensive results demonstrate that our AutoReg may automatically learn an optimal deep registration network for given volumes and achieve state-of-the-art performance, also significantly improving computation efficiency than the mainstream UNet architectures (from 0.558 to 0.270 seconds for a 3D image pair on the same configuration).

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

↑