首页|期刊导航|南京信息工程大学学报|基于通道注意力与图注意力网络的脑部影像分类研究

基于通道注意力与图注意力网络的脑部影像分类研究OA

Brain image classification based on channel attention and graph attention network

中文摘要英文摘要

针对阿尔茨海默病(AD)与认知正常(CN)患者的脑部影像分类问题,本文提出一种基于通道注意力与图注意力网络的脑部影像分类模型.首先对原始DTI影像进行标准化预处理与张量拟合,获得三维FA图,随后沿轴向逐层切片,并应用双线性插值法将所有切片统一重采样到相同的大小,在此基础上,设计了通道注意力卷积模块,通过并行卷积提取空间细节特征,同时应用自适应核长的通道间注意力机制动态重标定通道权重,有效学习并压缩局部纹理特征,经全连接层映射得到特征向量并作为后续图中的节点嵌入.然后,依据切片的空间连通性构建邻接矩阵,将切片视为图中的节点,并将节点特征向量与邻接矩阵组成图数据,输入三层堆叠的图注意力卷积层,通过可学习的注意力系数逐层聚合相邻及远邻切片信息,实现跨切片全局结构的融合.最终,经全局平均池化与全连接层输出分类结果.实验结果表明,本文提出的方法不仅在准确度和稳定性上优于传统影像分类模型,还验证了局部-全局注意力架构的优势,提供了一种从切片到图信号转换的新思路.

To address brain image classification for distinguishing patients with Alzheimer's Disease(AD)from Cognitively Normal(CN)individuals,this study proposes a model based on channel attention and graph attention networks.First,the original Diffusion Tensor Imaging(DTI)data were preprocessed and tensor fitted to obtain a three-dimensional Fractional Anisotropy(FA)map.Then,it was sliced layer by layer along the axial direction,and all slices were uniformly resampled to the same size using bilinear interpolation.A channel attention convolution module was designed to extract spatial detail features through parallel convolutions.At the same time,the channel weights were dynamically recalibrated using the inter-channel attention mechanism with adaptive kernel length,ef-fectively learning and compressing local texture features.The feature vector was obtained through mapping in the fully connected layer and embedded as a node in the subsequent graph.Next,the adjacency matrix was constructed according to the spatial connectivity of the slices.The slices were regarded as nodes in the graph.The node feature vectors and the adjacency matrix were combined into graph data,which were input into three stacked graph attention convolution layers.The information of adjacent and distant slices was aggregated layer by layer through the learnable attention coefficient to achieve the fusion of the global structure across slices.Finally,the classification results were output through global average pooling and fully connected layers.Experimental results show that the proposed meth-od not only outperforms traditional image classification models in accuracy and stability,but also verifies the advan-tages of the local-global attention architecture.Furthermore,it offers a novel approach for converting slice-based data into graph signals.

林文轩;徐军

南京信息工程大学 自动化学院,南京,210044||南京信息工程大学 智能医学图像计算江苏省高校重点实验室,南京,210044南京信息工程大学 智能医学图像计算江苏省高校重点实验室,南京,210044||南京信息工程大学 未来技术学院,南京,210044

医药卫生

医学图像脑部影像分类深度学习注意力机制图卷积神经网络

medical imagebrain image classificationdeep learningattention mechanismgraph convolutional neural network(GCN)

《南京信息工程大学学报》 2026 (2)

211-220,10

国家自然科学基金(62171230,62101365,92159301)

10.13878/j.cnki.jnuist.20250327002

评论