arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03585cs.CVeess.IV
收藏

仅解码器图像配准

Decoder-Only Image Registration

School of Computer Science, University of Birmingham(伯明翰大学计算机学院);Department of Computing and Mathematics, Manchester Metropolitan University(曼彻斯特城市大学计算与数学系)

Xi Jia, Wenqi Lu, Xinxing Cheng, Jinming Duan

AI总结:

针对无监督医学图像配准中编码器-解码器架构的局限性,提出仅含可学习解码器的LessNet,以手工特征替代编码器,在脑MRI数据集上实现与VoxelMorph等相当的配准精度,同时大幅降低计算开销。

摘要:

在无监督医学图像配准中,主流方法采用编码器-解码器网络架构,能够从给定的图像对中精确预测稠密、全分辨率位移场。尽管该架构在文献中广泛使用,我们认为有必要使编码器和解码器均可学习。为此,我们提出一种新颖的网络架构,本文中称为LessNet,该架构仅包含可学习的解码器,完全省略了可学习编码器的使用。LessNet用简单的手工特征替代可学习编码器,完全消除了在编码器中学习(优化)网络参数的需求。因此,这产生了一种紧凑、高效且仅解码器的3D医学图像配准架构。在两个公开的脑部MRI数据集上的评估表明,我们的仅解码器LessNet能够有效且高效地学习3D中的稠密位移场和微分同胚形变场。此外,我们的仅解码器LessNet能够达到与最先进方法(如VoxelMorph和TransMorph)相当的配准性能,同时所需计算资源显著减少。我们的代码和预训练模型可在https://github.com/xi-jia/LessNet获取。

英文摘要:

In unsupervised medical image registration, the predominant approaches involve the utilization of a encoder-decoder network architecture, allowing for precise prediction of dense, full-resolution displacement fields from given paired images. Despite its widespread use in the literature, we argue for the necessity of making both the encoder and decoder learnable in such an architecture. For this, we propose a novel network architecture, termed LessNet in this paper, which contains only a learnable decoder, while entirely omitting the utilization of a learnable encoder. LessNet substitutes the learnable encoder with simple, handcrafted features, eliminating the need to learn (optimize) network parameters in the encoder altogether. Consequently, this leads to a compact, efficient, and decoder-only architecture for 3D medical image registration. Evaluated on two publicly available brain MRI datasets, we demonstrate that our decoder-only LessNet can effectively and efficiently learn both dense displacement and diffeomorphic deformation fields in 3D. Furthermore, our decoder-only LessNet can achieve comparable registration performance to state-of-the-art methods such as VoxelMorph and TransMorph, while requiring significantly fewer computational resources. Our code and pre-trained models are available at https://github.com/xi-jia/LessNet.

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

↑