关键词:
深度学习
图像配准
UNet网络
交叉注意力
摘要:
针对三维图像配准存在易丢失空间信息、拓扑结构保持困难及耗时长等问题,提出一种混合的Transformer-ConvNet模型。在经典VoxelMorph模型基础上,引入了交叉注意力模块,进行有效的远程建模和高维数据处理;针对运算量大的问题,在卷积层引入了inception模块,采用并行连接的方式将多个不同尺寸和不同类型的卷积层相互融合,学习细粒度特征,以提高配准精度。在脑部MR数据集上,进行了消融实验,结果表明,引入交叉注意力机制和inception模块网络较VoxelMorph,Dice系数提高了11.5%。将所提算法与4种经典算法进行了对比,结果表明,所提模型配准性能有了明显提升,较VoxelMorph、CycleMorph、ViT-V-Net、TransMorph分别提高9.6%、8.7%、9.1%、6.3%,且参数量较少。