MG-UNet:无监督3D医学图像的多粒度配准框架OA
MG-UNet:Multi-granularity UNet Framework for Unsupervised 3D Medical Image Registration
医学图像配准在临床诊断和治疗中至关重要,但现有方法在处理复杂解剖结构和变形场的稳定性方面仍存在挑战,且难以兼顾多尺度信息建模与高效计算.本文提出一种新的无监督3D医学图像配准模型——MG-UNet(Multi-Granularity UNet),旨在提高配准精度并优化变形场的合理性与光滑性.MG-UNet采用多尺度多粒度特征建模方案,通过多方向卷积初始层增强三维空间中的细粒度信息表达,利用多粒度空洞卷积模块(Multi-Granularity Dilated Convolu-tion,MGDC)提取不同感受野下的局部与全局特征,并设计多注意力融合模块(Multi-Attention Fusion,MAF)强化关键区域的特征响应.实验结果表明,MG-UNet在OASIS和IXI数据集上均取得了最优结果,在OASIS数据集上,Dice系数提高了2.3%,HD95下降12.2%,ASSD下降16.3%;在IXI数据集上,Dice系数达到了0.769,相比当前最佳方法进一步提升,同时MG-UNet-tiny的HD95和雅可比行列式负值百分比达到最低,并且其参数量仅为1.597 M,与基于Transformer的模型相比显著降低了计算量.这表明了MG-UNet模型的有效性,能够稳定捕获医学图像的解剖结构变形关系,为高效医学图像配准提供了新的技术路径.
Medical image registration is crucial for clinical diagnosis and treatment.However,existing methods face challenges in handling complex anatomical structures and ensuring stability in the deformation field,while also struggling to balance multi-scale information modeling with computational efficiency.This paper proposes a novel unsupervised 3D medical image registra-tion model—MG-UNet(Multi-Granularity UNet),aiming to enhance registration accuracy and optimize the rationality and smoothness of the deformation field.MG-UNet adopts a multi-scale,multi-granularity feature modeling approach:it enhances fine-grained information representation in 3D space through a multi-directional convolutional initial layer,extracts local and global features under different receptive fields via a Multi-Granularity Dilated Convolution(MGDC)module,and incorporates a Multi-attention Fusion(MAF)module to reinforce feature responses in key regions.Experimental results demonstrate that MG-UNet achieves state-of-the-art performance on both the OASIS and IXI datasets.On the OASIS dataset,the Dice coefficient im-proved by 2.3%,HD95 decreased by 12.2%,and ASSD decreased by 16.3%;on the IXI dataset,the Dice coefficient reached 0.769,further surpassing the current best methods.Additionally,the MG-UNet-tiny variant achieved the lowest HD95 and the lowest percentage of negative Jacobian determinants,with a parameter count of only 1.597 M,significantly reducing computa-tional load compared to Transformer-based models.These results demonstrate the effectiveness of the MG-UNet model in reli-ably capturing the deformation relationships of anatomical structures in medical images,offering a new technical pathway for effi-cient medical image registration.
金燕灵;赵玉龙;胡真
河海大学数学学院,江苏 南京 211100香港应用科技研究院,香港河海大学数学学院,江苏 南京 211100
信息技术与安全科学
医学图像配准无监督学习多粒度空洞卷积多注意力融合机制
medical image registrationunsupervised learningmulti-granularity dilated convolutionmulti-attention fusion mechanism
《计算机与现代化》 2026 (5)
66-77,12
国家重点研发计划项目(2024YFE0206600)
评论