LLM2D
基于图的跳跃连接与变换器结合的TransGUNet:医疗影像分割
TransGUNet: Transformer Meets Graph-based Skip Connection for Medical Image Segmentation
作者: Ju-Hyeon Nam, Nur Suriza Syazwany, Sang-Chul Lee
发布日期: 2/17/2025
arXiv ID: oai:arXiv.org:2502.09931v1

摘要

arXiv:2502.09931v1 类型:交叉 摘要:跳连接工程主要用来解决编码器与解码器之间的语义差距,同时整合全局依赖关系以理解医学图像分割中复杂解剖结构之间的关系。虽然已有模型提出了基于变换器的方法来在跳连接中引入全局依赖关系,但它们往往在高计算复杂性下难以捕捉详细的局部特征。相比之下,图神经网络(GNNs)利用图结构有效地捕捉局部和全局特征。利用这些特性,我们引入了一种注意力跨尺度图神经网络(ACS-GNN),它通过将跨尺度特征图转换为图结构,并通过节点注意力捕捉复杂解剖结构来增强跳接连接框架。此外,我们观察到深度学习模型往往会产生不具有信息量的特征图,这会降低空间注意力图的质量。为了解决这一问题,我们结合熵驱动的功能选择(EFS)与空间注意力,为每个通道计算熵分数,并过滤掉高熵特征图。我们的创新框架TransGUNet包括ACS-GNN和基于EFS的空间注意力,通过结合GNN以及可靠的三维空间注意力图,充分利用GNN的特性来提高在不同模态下的领域泛化能力,确保跳接连接中的更稳健特征。通过全面的实验和分析,TransGUNet在六个已见数据集和八个未见数据集上的分割性能均表现出优越性,并且相比以前的方法具有显著更高的效率。