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

一种用于无监督可变形图像配准的轻量级残差网络

A lightweight residual network for unsupervised deformable image registration

University of Innsbruck(因斯布鲁克大学);Medical University of Innsbruck(因斯布鲁克医科大学)

Ahsan Raza Siyal, Astrid Ellen Grams, Markus Haltmeier

AI总结:

本文提出一种嵌入并行空洞卷积块的残差U-Net,以极低参数量增强感受野,在有限训练数据上实现与Transformer方法相当甚至略优的无监督可变形图像配准性能。

摘要:

精确的体图像配准对于临床常规工作和计算机辅助医学诊断具有重要意义。近年来,研究者开始在基于学习的医学图像配准方法中使用Transformer,并取得了显著成功。由于强大的全局建模能力,Transformer被认为是比卷积神经网络(CNN)更适合配准的选择。然而,它们使用参数量庞大的笨重模型,这需要高计算边缘设备才能部署为便携式设备或在医院中使用。Transformer还需要大量训练数据才能产生显著结果,而收集合适的标注数据通常具有挑战性。尽管现有的基于CNN的图像配准可以提供丰富的局部信息,但它们的全局建模能力较差,难以处理长距离信息交互,从而限制了配准性能。在这项工作中,我们提出了一种基于CNN的配准方法,该方法具有增强的感受野、较少的参数量,并且在有限的训练数据集上取得了显著结果。为此,我们提出了一种残差U-Net,其中嵌入了并行空洞卷积块以增强感受野。所提出的方法在患者间数据集和基于图谱的数据集上进行了评估。我们表明,所提出方法的性能与基于Transformer的方法相当且略好,而仅使用了其参数量的1.5%。

英文摘要:

Accurate volumetric image registration is highly relevant for clinical routines and computer-aided medical diagnosis. Recently, researchers have begun to use transformers in learning-based methods for medical image registration, and have achieved remarkable success. Due to the strong global modeling capability, Transformers are considered a better option than convolutional neural networks (CNNs) for registration. However, they use bulky models with huge parameter sets, which require high computation edge devices for deployment as portable devices or in hospitals. Transformers also need a large amount of training data to produce significant results, and it is often challenging to collect suitable annotated data. Although existing CNN-based image registration can offer rich local information, their global modeling capability is poor for handling long-distance information interaction and limits registration performance. In this work, we propose a CNN-based registration method with an enhanced receptive field, a low number of parameters, and significant results on a limited training dataset. For this, we propose a residual U-Net with embedded parallel dilated-convolutional blocks to enhance the receptive field. The proposed method is evaluated on inter-patient and atlas-based datasets. We show that the performance of the proposed method is comparable and slightly better than transformer-based methods by using only $\SI{1.5}{\percent}$ of its number of parameters.

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

↑