基于空间与频率融合及图神经网络的帕金森病MRI数据分类

胡国庆 ,  施振佺 ,  黄嘉爽 ,  丁卫平

南京大学学报(自然科学) ›› 2026, Vol. 62 ›› Issue (01) : 48 -58.

PDF (1407KB)
南京大学学报(自然科学) ›› 2026, Vol. 62 ›› Issue (01) : 48 -58. DOI: 10.13232/j.cnki.jnju.2026.01.005

基于空间与频率融合及图神经网络的帕金森病MRI数据分类

作者信息 +

Parkinson's disease MRI data classification based on spatial⁃frequency fusion and graph neural network

Author information +
文章历史 +
PDF (1440K)

摘要

帕金森病(Parkinson's Disease,PD)是一种常见的神经系统退行性疾病,其早期诊断对于延缓疾病进展至关重要.磁共振成像(Magnetic Resonance Imaging,MRI)凭借无创性和高分辨率的特点,在对PD的诊断中得到了广泛应用.然而,现有方法往往仅依赖单一域信息,存在信息建模不足的问题,同时,PD的病理变化不是孤立的,现有方法对图像块之间的区域关联性往往考虑不多,忽略了脑区之间的相互影响.因此,提出一种融合空间与频率信息的双分支深度学习框架,空间分支采用视觉Transformer来捕获MRI图像中的全局空间关系,频域分支则利用全局滤波网络来提取频域特征,通过高斯加权欧氏距离来构建邻接矩阵,并引入图卷积网络来建模图像块间的拓扑关系.在模型训练中,选取轴向2D切片并结合ImageNet的预训练权重进行迁移学习,通过多数投票策略比较单个受试者的多切片结果,最终生成受试者级别的分类结果.在包含患者与健康对照的PD数据集上进行评估,实验结果显示,所提方法的准确率、特异性与F1⁃score等指标均优于多个现有方法,证实其在临床应用中的有效性.

Abstract

Parkinson's disease (PD) is a common neurodegenerative disorder,and early diagnosis is crucial for slowing disease progression. Magnetic resonance imaging (MRI) has been widely used in the diagnosis of PD due to its non⁃invasive nature and high⁃resolution capabilities. However,existing methods often rely on information from a single domain,resulting in insufficient information modeling. Furthermore,the pathological changes in PD are not isolated,existing approaches frequently fail to account for regional correlations between image patches,thereby neglecting the functional interactions among brain regions. To address these limitations,we propose a two⁃branch deep learning framework that integrates spatial and frequency⁃domain information. The spatial branch employs a Vision Transformer to capture global spatial relationships in MRI images,while the frequency branch utilizes GFNet (Global Filter Network) to extract frequency⁃domain features. An adjacency matrix is constructed using Gaussian⁃weighted Euclidean distance,and a graph convolutional network (GCN) is introduced to model the topological relationships between image patches. During model training,axial 2D slices are selected and fine⁃tuned using pre⁃trained weights from ImageNet through transfer learning. A majority voting strategy is then applied to aggregate predictions from multiple slices of a single subject to produce a subject⁃level classification result. The proposed method was evaluated on a PD dataset comprising both patients and healthy controls. Experimental results demonstrate that our approach outperforms several state⁃of⁃the⁃art methods in terms of key metrics including accuracy,specificity,and F1⁃score,thereby confirming its potential for effective clinical application.

Graphical abstract

关键词

帕金森病 / 磁共振成像 / 深度学习 / 图卷积网络 / Transformer

Key words

Parkinson's disease / MRI / deep learning / graph convolutional network / Transformer

引用本文

引用格式 ▾
胡国庆,施振佺,黄嘉爽,丁卫平. 基于空间与频率融合及图神经网络的帕金森病MRI数据分类[J]. 南京大学学报(自然科学), 2026, 62(01): 48-58 DOI:10.13232/j.cnki.jnju.2026.01.005

登录浏览全文

4963

注册一个新账户 忘记密码

帕金森病(Parkinson's Disease,PD)是一种渐进性神经退行性疾病,主要影响中枢神经系统1,主要是由于黑质区多巴胺神经元的退化,从而影响运动功能.由于缺乏可靠的生物标志物,且疾病前期脑部结构的改变较为细微,PD的早期精准诊断仍是临床面临的重大挑战2.近年来,结构磁共振成像(Structural Magnetic Resonance Imaging,sMRI)已成为一种极具价值的无创成像手段,能够揭示PD患者与健康对照者之间的神经解剖学差异,为计算机辅助诊断及疾病监测提供了可能3.
早期基于sMRI的PD分类方法,通常先通过感兴趣区域分析或基于体素的形态学分析(Voxel⁃Based Morphometry,VBM)提取手工设计的形态学特征(如皮质下体积和皮质厚度),并结合支持向量机、随机森林等传统机器学习分类器4-5来进行分类.这类方法虽然能达到中等的诊断准确率,但有明显的局限,一方面依赖人工设计的特征表示,另一方面对高维特征的处理能力有限,过多无关特征会干扰分类决策,导致“维度灾难”.
随着深度学习的兴起,卷积神经网络(Convolutional Neural Network,CNN)成为从MRI中自动提取特征的主流框架,并在神经系统疾病辅助诊断领域取得了显著进展6.例如,Sivaranjini and Sujatha7采用AlexNet,通过迁移学习对PPMI数据库中的MRI图像进行PD检测.Zhang et al8开发了基于ResNet的框架,以灰质sMRI切片为输入,将特定区域特征提取与注意力机制相结合,增强灰质特征的表征能力.然而,由于固有的局部感受野限制,CNN在建模大脑区域之间的全局关系方面存在不足9.此外,CNN通常仅在空间域进行建模,往往忽略磁共振成像采集过程中固有的频率信息,可能遗漏具有互补性的判别线索.
近年来,深度学习领域涌现出众多基于Transformer架构的方法,如视觉Transformer (Vision Transformer,ViT)10.ViT通过自注意力机制可有效捕捉远距离特征间的联系,提升全脑范围内的表示能力.Swin Transformer11通过窗口滑动机制引入局部感受野,并构建层次化结构以实现局部到全局的特征聚合.DeiT12通过引入知识蒸馏、简化位置嵌入、优化训练策略的方式,使其在小规模数据集上仍能取得良好的性能.然而,纯Transformer模型在局部细节特征提取方面比CNN弱,且对大规模数据依赖较高,训练稳定性较差.同时,PD的病理变化不是孤立的,现有方法对图像块之间的区域关联性往往考虑不多,忽略了脑区之间的相互影响.因此,在识别早期PD患者时仍面临以下关键挑战:一是从复杂结构中有效提取局部细微病变特征;二是在全脑范围内建模多区域之间的空间拓扑关系;三是充分融合空间和频率信息,以提升对病理区域的敏感性和诊断的准确性.
针对以上问题,本文提出了一种融合空间与频域特征的双分支模型PDFormer,结合空间与频率双域特征、局部与全局联合建模以及图神经网络融合机制,显著提升了模型对复杂脑部结构异常的识别能力.
本文的具体贡献如下.
(1)联合建模空间域与频率域特征:采用视觉Transformer提取MRI图像中的全局空间特征,并结合全局滤波网络捕捉频率域中的局部高频细节,通过特征拼接获得融合表示,有效整合多尺度信息.
(2)基于欧氏距离构建拓扑邻接图:提出一种高斯加权的patch中心距离图建模方法,用于刻画MRI图像中patch之间的空间关系,增强结构信息表达能力.
(3)引入图卷积神经网络(Graph Convolution Network,GCN)进行高阶特征融合:利用GCN对patch间拓扑结构进行建模,提升模型对复杂结构变化的感知能力,并通过切片级多数投票策略提升诊断稳定性.

1 相关工作

1.1 基于传统机器学习的PD分类方法

传统的基于机器学习的方法多是分两步进行13,即先手动或者自动地从影像数据中提取与疾病相关的特征或者感兴趣区域,然后将提取到的内容送入机器学习模型进行分类.例如,Abós et al14基于Brainnetome脑图谱从静息态fMRI (Functional Magnetic Resonance Imaging)重建功能连接组,利用逻辑回归进行特征选择,结合支持向量机进行训练和分类.Anita15提出一种基于三维SPECT (Single⁃Photon Emission Computed Tomography)的早期PD诊断方法,从三组不同的体绘制图像(Volume Rendering Image,VRI)切片中提取形状特征和表面拟合特征,使用遗传算法从提取的特征中选择最优特征集,并使用径向基函数ELM (Extreme Learning Machine)分类器和径向基函数支持向量机分类器进行分类.Babu et al16提出一种带有递归特征消除的元认知径向基函数网络(McRBFN),使用基于体素的形态测量特征并采用基于投影的学习算法.然而,由于PD相关脑区的结构复杂,手动定义最优特征集往往十分困难.再加上医学影像数据的高维特性,限制了传统机器学习模型的有效性.

1.2 基于深度学习的空间域与频域建模

深度学习利用受人脑启发的神经网络结构来实现特征学习与数据分析17,各种深度学习网络框架也已经被开发出来,并广泛应用于计算机视觉、语音识别与数据挖掘等领域.

大多数基于MRI的PD预测研究使用CNN进行预测.例如,Mary and Suganthi18用MFEA (Multifactorial Evolutionary Algorithm)提取多种特征,再利用主成分分析进行特征降维,将处理后的特征输入带有YOLO应用风格的DNetCNN模型进行PD的分类.Chakraborty et al19构建了一个3D CNN,用于从整个脑部MRI图像中学习复杂的三维结构模式,以区分PD病例与健康对照.Zhu20提出一种混合深度学习的模型,结合来自MRI脑部扫描和患者特征(症状数据)的多模态信息将患者划分为五个阶段,实现更细致的严重程度评估.

与此同时,基于Transformer的架构也作为一种新的范式出现,利用自注意力机制捕捉全局上下文关系.最初,该类方法应用于自然语言处理任务,在BERT21与GPT⁃322等模型中凭借强大的特征表征能力取得了显著成功.受此启发,研究者将Transformer编码器拓展至计算机视觉领域,提出如ViT等具有影响力的模型用于图像分类.与空间建模互补的是,频域建模近年来也成为一种有效方法,用于挖掘MRI中固有的频谱信息.例如,GFNet23通过快速傅里叶变换构建全局滤波器,从而增强频域表征能力.

1.3 基于图神经网络的拓扑特征学习

近年来,图神经网络(Graph Neural Networks,GNNs)因其能够建模非欧几里得结构并捕捉区域或图像块之间的拓扑关系,而逐渐受到关注.随着图卷积网络的发展,研究者提出多种复杂的消息传递机制.例如,Atwood and Towsley24提出的扩散卷积神经网络(DCNN)通过在图上定义基于随机游走的扩散过程来实现卷积操作,从而在图结构数据中有效捕捉局部与多步邻域信息.Tran et al25引入基于最短路径距离的多阶卷积方法,将不同距离的邻居信息按不同权重聚合,让模型在训练中学习“远邻居”的贡献.Gilmer et al26提出消息传递神经网络(Message Passing Neural Network,MPNN),通过迭代的消息传递与节点更新机制来学习图结构表示.这些方法在分子建模、社交网络等任务中已展现出优越的性能.

在结构性MRI分析中,基于图的方法的独特优势在于其能够显式编码空间关系,而这类关系在CNN与Transformer架构中通常是隐式捕捉的.同时,虽然GCN在拓扑特征学习方面发展迅速,但与图像分类任务的结合研究仍较为有限,这也为本研究提供了新的切入点.

2 提出的方法

2.1 PDFormer框架概述

考虑到PD相关的脑部变化在早期阶段较微弱且具有异质性,因此,仅依赖单一空间域或频域特征的建模方法往往无法充分刻画病变区域的信息.将PD图像分类视为多层次特征融合与拓扑关系建模问题,提出PDFormer模型.图1为提出的模型的框架示意图.

具体地,所提模型包含四个关键步骤.首先,选择关键轴向切片以构建输入数据.其次,设计一个双域特征提取框架,其中一个分支用于处理空间域表征,另一个分支则用于提取频域特征.随后,构建图拓扑结构,其中节点表示MRI图像块,边则基于欧式距离编码空间邻接关系,利用GCN实现高阶特征融合与增强特征表示.最后,模型对每个轴向切片生成初步的分类结果,并通过多数投票策略整合多个切片的预测结果,以获得受试者级别的PD诊断决策.

2.2 切片选择

在进行经验分析和多次实验(详见4.1)后,在Z轴方向选取轴向平面索引位于76~90序列的连续15张切片来训练网络,最终获得2190张PD患者的轴向切片和2325张健康对照(Healthy Control,HC)个体的轴向切片,每张切片均为182×218的灰度图像.对切片后的数据进行预处理,使用数据增强技术,包括随机旋转(-20°~20°)和随机翻转,以此来增加数据多样性.将图像尺寸调整为224×224,并对像素值归一化到0,1.

2.3 空间表征学习

本研究中ViT是左侧主干网络,其结构类似标准ViT模型,由多层Transformer编码器组成,每层由多头自注意力和多层感知机交替堆叠而成.在每个Transformer层之前都会进行层归一化,并在每个模块后加入残差连接以优化梯度流动.

对输入的2D MRI切片xRH×W×C(空间分辨率为H×W且包含C个通道)进行分块,将其划分为多个P×P的patch,xpRN×P2CP×P表示每个patch的大小,N=HW/P2表示总的patch个数,也是Transformer的输入序列长度.每个patch被展平后通过可训练的线性投影矩阵WpRP2C×D映射到固定维度D,形成Transformer的输入序列,具体计算如下:

z0=xpWp+Epos

其中,EposRN×D为可学习的位置编码,用于弥补Transformer结构中缺乏空间位置信息的不足.ViT的Transformer编码器计算如下:

zl'=MSALNzl-1+zl-1l=1,,L
zl=MLPLNzl'+zl'l=1,,L

其中,L为Transformer的层数.本研究使用12层Transformer编码器.

2.4 频域感知特征编码

在MRI过程中,数据首先在频域中采集,然后经过傅里叶逆变换转换到空间域供临床观察,因此,利用频域信息来辅助分类任务具有重要的医学意义.然而,直接在高维频域数据上进行深度学习计算代价较高,特别是在处理高分辨率影像时,为此,采用GFNet进行高效的频域特征提取.GFNet的核心思想是用快速傅里叶变换替代自注意力机制,以降低计算复杂度.

在PDFormer右侧分支中对输入的2D MRI切片进行非重叠patch划分.与ViT的处理方式一致,将每个patch展平为L=H×W个维度为D的tokens序列,每个空间域tokenxRH×W×D由2D FFT转换为频域中的复杂张量XRN×D

X=2D FFTx

其次,引入可训练的全局滤波器K来对FFT变换后的特征进行调制,以增强对高频和低频信息的建模能力:

X˜=XK

其中,⊙表示逐元素乘法;K是一个全局滤波器,其维度与X相同,并通过网络训练进行优化.

最后,经过频域调制后的特征仍然位于频域,因此需要通过二维傅里叶逆变换(IFFT)将其转换回空间域,以便进行后续特征融合和分类:

x'=IFFTX˜

其中,x'表示逆变换后的特征,它既包含原始MRI影像的空间信息,也结合了频域增强后的特征表示.GFNet采用全局平均池化代替ViT结构中的分类token,以减少参数数量,同时保留重要特征信息.本研究使用GFNet⁃B变体,其包含19层网络结构,嵌入维度为512.

2.5 基于图的拓扑建模

对于每个图像块pi

i1,2,,N,空间特征fivRDv由ViT编码器提取,频域特征figRDg通过GFNet编码器在执行二维傅里叶变换后获得.最终的图像块级表征fiRDv+Dg通过拼接空间特征和频域特征得到:

fi=Concatfiv,fig

为了建模图像块之间的拓扑结构,构建一个无向图G=V,A,其中,每个节点viV对应一个图像块,邻接矩阵ARN×N根据图像块中心之间的欧几里得距离计算得到,并通过高斯核进行调制.具体地,首先计算每个图像块的几何中心坐标.给定一个H×W的图像和P×P的图像块,图像被划分为h=H/P行和w=W/P列,总计N=h×w个图像块.位于第i行第j列的图像块的中心坐标ch,cw计算如下:

ch=iP+P2    i0,1,,h-1cw=jP+P2    j0,1,,w-1

对于任意两个图像块pm=chm,cwmpn=chn,cwn,其中心点之间的欧几里得距离dmn为:

dmn=chm-chn2+cwm-cwn2

图像块mn之间的连接强度通过高斯核定义为:

wmn=exp-dmn22σ2

这里的高斯核是基于欧几里得距离的高斯权重函数,用来衡量两个图像块中心之间的连接强度,距离越近权重越大,距离越远权重越小,σ是控制衰减速率的超参数.遍历所有块之间的中心点组合,将权重存入邻接矩阵 A .

邻接矩阵 A 的定义为:

Am,n=A[n,m]=wmn,    mn1,         m=n

将拼接后的图像块表征X=f1,f2,,fN

RN×Dv+Dg与邻接矩阵 A 一同输入GCN网络,以执行基于拓扑的特征聚合.GCN的运算定义为:

Hl+1=σA˜HlWl

其中,Hl是第l层的输入特征矩阵(H0=X为初始输入特征矩阵),Wl是第l层的可训练权重矩阵,A˜为归一化后的邻接矩阵,σ为激活函数.输出Hl+1为经过信息传递后更新的图像块级表征.

2.6 分类与多数投票策略

在通过空间⁃频域融合与基于图的拓扑建模获得判别性表征后,每张2D MRI切片将被单独分类.具体地,对图卷积模块输出的特征进行全局平均池化,以生成每张切片的切片级特征向量.随后,该向量通过一个全连接层,并接入Softmax激活函数,以预测该切片在两个类别上的概率分布.

对于每位受试者,模型独立处理其多张轴位切片,生成切片级的分类预测.为了获得受试者级的诊断结果,采用多数投票策略,即将所有切片预测结果中出现频率最高的类别作为最终的预测:

yfinal=modey1,y2,,yS

其中,yi0,1表示第i张切片的预测标签,S为该受试者的切片总数.mode函数返回在所有切片预测中出现次数最多的类别标签.

3 实验结果

3.1 数据集与预处理

采用的数据集涵盖了患者组和健康对照组,共301例受试者,其中,PD组146例,HC组155例.其中,PD受试者:(1)符合PD的诊断标准;(2)接受MRI扫描;(3)未接受过脑部手术;(4)没有颅内占位性病变、头部外伤史或任何其他影响大脑重要结构的疾病.排除标准:(1)MRI检查禁忌证;(2)排除运动伪影和金属伪影.HC组无任何已知的神经系统疾病或认知障碍.

所有影像数据均使用配备24个通道头线圈的3⁃T磁共振成像扫描仪(GE Signa)采集,T1加权成像的扫描参数如表1所示.扫描图像经过筛选后分别归入相应研究组.

由于原始扫描图像之间存在空间差异,需要对图像进行预处理,以便将所有扫描图像统一到同一空间.使用蒙特利尔神经研究所(Montreal Neurological Institute)创建的MNI152_T1_1mm模板,借助FSL工具(https://fsl.fmrib.ox.ac.uk)完成线性配准.线性配准后,不同扫描对象的头颅大小和空间朝向得以统一.图2展示了配准前后的MRI扫描结果.

本研究中,为了标准化模型输入并降低计算复杂度,从每例配准后的三维MRI中,沿轴状面方向提取15张连续切片(切片索引76至90),每张图像的分辨率为182×218像素.这些二维切片被用于后续模型的训练与分类,所有数据均以NIfTI格式(.nii)存储.

3.2 实验设置

采用十折交叉验证来对数据集进行训练与测试,以确保实验结果的稳定性和可靠性,并获得更具泛化能力的评估指标.此外,为了进一步确保实验结果的可靠性,通过计算十折的平均性能来报告最终的分类结果.采用迁移学习策略,具体地,ViT与GFNet的参数初始化均来自在ImageNet数据集上预训练的权重.

所有实验均在单个NVIDIA RTX 4090 GPU上进行,使用的深度学习框架为PyTorch 2.0.1.采用SGD优化器对所有网络进行训练,并设置动量值为0.9,以加速收敛并减少梯度更新过程中的振荡.初始学习率设为0.0001,每批次数据的大小为16,总训练epoch数为150.以上参数配置均适用于两种Transformer网络.表2简要总结了提出的框架的网络参数细节.

3.3 评价指标

为了全面评估所提方法的分类性能,采用准确率、灵敏度、特异性和F1⁃score来进行评估.这些指标通过四个值进行计算:真正例(TP)、真负例(TN)、假正例(FP)和假负例(FN).

准确率(Accuracy,ACC)表示分类模型正确识别的样本占总样本数的比例:

ACC=TP+TNTP+TN+FP+FN

灵敏度(Sensitivity,SEN),也称召回率(Recall),表示模型正确识别患病患者的能力:

SEN=TPTP+FN

特异性(Specificity,SPE),也称为真负率(True Negative Rate),是衡量模型正确识别健康个体的能力,可表示为:

SPE=TNTN+FP

F1⁃score是精确度和召回率的调和平均数,综合考虑了精确度与召回率的平衡:

F1score=2TP2TP+FP+FN

3.4 对比实验

为了验证所提模型的有效性,将其与五个常见的深度学习模型进行对比,包括DenseNet (2D)27,MobileNetV2 (2D)28,ResNet (2D)29,ResNet (3D)和SFCN (3D).对于2D CNN模型,步骤包括输入切片、数据增强、采用多数投票策略等.对于3D网络架构,输入为完整的3D大脑MRI数据,以进行分类结果预测.

图3展示了不同方法在PD数据集上分类的性能对比,图中的误差棒表示每个指标的标准差,反映了模型性能的波动范围.PDFormer的误差棒较短,表明其性能更稳定.相比之下,某些方法的误差棒较长,表明其性能波动较大,可能受到数据分布或模型复杂度的影响.

为了提供更详细的对比分析,表3列出了不同模型在各个性能指标上的平均值和标准差,表中黑体字表示最优的性能.

3.5 消融实验

为了探究所提模型各核心组件对最终性能的具体贡献,进行了五组消融实验.在每个实验中去除模型的一个关键部分并评估其对性能的影响.表4展示了在PD数据集第四折上的消融实验结果,作为十折交叉验证实验中的代表性结果,表中黑体字表示最优的性能.

由表可见,仅使用ViT提取空间域特征时,模型在负样本识别SPE=0.700上存在一定的改进空间.仅使用GFNet提取频域特征时,模型能有效识别健康对照,但对PD患者的识别效果较差.去除多数投票机制后的模型仍然使用ViT,GFNet和GCN层进行特征提取和融合,此时模型的ACC=80%,SEN=93.3%,说明去除多数投票机制后,模型对单一切片的分类性能有所提升,但整体分类效果(SPEF1⁃score)有所下降.去除GCN层后,模型虽然仍使用ViT和GFNet进行特征提取,但不再对提取的特征进行高阶关系建模,此时模型的ACC=76.67%,和PDFormer相比,性能有所下降.

综合来看,完整的PDFormer表现最优.

4 讨论

4.1 切片选择的影响

完成配准及尺寸归一化等标准预处理操作后,将所有MRI图像体积统一调整为固定大小,从处理后的体积图像中提取224×224的轴向2D切片.考虑到MRI体积的首尾部分通常包含的结构信息较少,注意力集中在体积图像的中部区域,以获得更具判别力的特征.

首先以体积中间位置为中心,提取连续15张切片进行训练,分析其分类性能.随后,逐步扩大切片跨度,分别构建30张和45张连续切片的输入组合,并对三种切片数量对应的分类性能进行比较,结果如图4所示.

实验结果表明,在使用连续15张切片时(图4a),模型的ACCF1⁃score已经表现良好,并能较稳定地区分样本差异.但进一步将跨度扩大至30或45张时,模型的性能没有得到显著的提升.如图4b和图4c所示,虽然部分切片组合的ACCF1⁃score略有浮动,但整体趋势趋于平稳,甚至在某些范围内还略有下降.

此外,切片数量的增加也显著增加了模型训练的计算开销.在当前的GPU环境下,15张切片的训练时长约为4 h,30张切片为7 h,45张切片则超过10 h.因此,为了在保证模型性能的同时兼顾训练效率,最终选择切片范围为第76~90张的15张切片组合作为模型输入.在该设置下,模型取得了最佳的分类表现,实现了性能与计算成本之间的良好权衡.

4.2 基于Grad⁃CAM的敏感区域可视化

为了更深入地理解所提模型在区分PD与HC时的决策过程,在测试集上进行了基于Grad⁃CAM的可解释性分析.测试集包含30例MRI扫描(PD患者15例,HC 15例).对于每个3D数据,选取索引为76~90的轴向切片,共获得450张用于可视化的2D图像.

基于训练好的PDFormer模型,将Grad⁃CAM应用于ViT和GFNet分支所提取的深层特征图,为每一张切片生成注意力图.这些注意力图能突出地显示模型在判别过程中所依赖的关键区域,图5展示了来自PD (图5a)与HC (图5b)两类受试者的代表性样例,每幅图被划分为两组(每类各包含两个样本).在每个可视化样例中,上排显示的是ViT分支的Grad⁃CAM结果,下排对应GFNet分支的结果,可以直接比较两个子网络的空间关注差异.图中红色区域表示模型关注度高的敏感区域.由图可见,ViT分支的Grad⁃CAM热力图较分散,模型在多个区域均给予一定的权重.GFNet分支的Grad⁃CAM热力图更集中.两个分支在同一样本上的关注区域存在一定互补性,证明双分支结构在全局⁃局部特征建模上的合理性.这种差异性与互补性也为后续的特征融合提供了理论支持,有助于提升整体分类性能.

5 结论

本文提出的PDFormer双分支网络模型结合了ViT与GFNet,联合建模MRI图像中的空间与频率特征,同时引入GCN建模图像块间的空间拓扑结构,在PD分类任务中取得了优于现有方法的性能表现,提高了其鲁棒性和预测能力.

未来将尝试引入多模态影像信息,如R2*,FLAIR,DTI等多模态影像,以获得更全面的特征表达,进一步提高分类性能.

参考文献

[1]

Bidesi N S RVang Andersen IWindhorst A Det al. The role of neuroimaging in Parkinson's disease. Journal of Neurochemistry2021159(4):660-689.

[2]

Qu YLi JQin Qet al. A systematic review and meta⁃analysis of inflammatory biomarkers in Parkinson's disease. npj Parkinson's Disease20239(1):18.

[3]

Khanna KGambhir SGambhir M. A novel technique for classifying Parkinson's disease using structural MRI scans. Multimedia Tools and Applications202382(29):46011-46036.

[4]

Haq A ULi J PMemon M Het al. Feature selection based on L1⁃norm support vector machine and effective recognition system for Parkinson's disease using voice recordings. IEEE Access2019(7):37718-37734.

[5]

Xu SPan Z. A novel ensemble of random forest for assisting diagnosis of Parkinson's disease on small handwritten dynamics dataset. International Journal of Medical Informatics2020,144:104283.

[6]

Litjens GKooi TBejnordi B Eet al. A survey on deep learning in medical image analysis. Medical Image Analysis2017,42:60-88.

[7]

Sivaranjini SSujatha C M. Deep learning based diagnosis of Parkinson's disease using convolutional neural network. Multimedia Tools and Applications202079(21):15467-15479.

[8]

Zhang YTeng QLiu Yet al. Diagnosis of Alzheimer's disease based on regional attention with sMRI gray matter slices. Journal of Neuroscience Methods2022,365:109376.

[9]

Luo WLi YUrtasun Ret al. Understanding the effective receptive field in deep convolutional neural networks. Advances in Neural Information Processing Systems2016:29.

[10]

Dosovitskiy ABeyer LKolesnikov Aet al. An image is worth 16×16 words:Transformers for image recognition at scale. 2020,arXiv:2010,11929.

[11]

Liu ZLin YCao Yet al. Swin transformer:Hierarchical vision transformer using shifted windows∥Proceedings of the IEEE/CVF Inter⁃national Conference on Computer Vision. Montreal,Canada:IEEE,2021:10012-10022.

[12]

Touvron HCord MDouze Met al. Training data⁃efficient image transformers & distillation through attention∥International Conference on Machine Learning. Online:PMLR,2021:10347-10357.

[13]

李西,姜孟. 机器学习在帕金森病诊断中的应用研究. 电子科技大学学报,2024,53(2):315-320.

[14]

Abós ABaggio H CSegura Bet al. Discriminating cognitive status in Parkinson's disease through functional connectomics and machine learning. Scientific Reports20177(1):45347.

[15]

Anita S. Improved classification accuracy for diagnosing the early stage of Parkinson's disease using alpha stable distribution. IETE Journal of Research202369(1):92-103.

[16]

Babu G SSuresh SMahanand B S. A novel PBL⁃McRBFN⁃RFE approach for identification of critical brain regions responsible for Parkinson's disease. Expert Systems with Applications201441(2):478-488.

[17]

孙志军,薛磊,许阳明,. 深度学习研究综述. 计算机应用研究,2012,29(8):2806-2810.

[18]

Mary GSuganthi N. Detection of Parkinson's disease with multiple feature extraction models and darknet CNN classification. Computer Systems Science & Engineering202243(1):333-345.

[19]

Chakraborty SAich SKim H C. Detection of Parkinson's disease from 3T T1 weighted MRI scans using 3D convolutional neural network. Diagnostics202010(6):402.

[20]

Zhu S. Early diagnosis of Parkinson's disease by analyzing magnetic resonance imaging brain scans and patient characteristic∥2022 10th International Conference on Bioinformatics and Computational Biology. Hangzhou,China:IEEE,2022:116-123.

[21]

Devlin JChang M WLee Ket al. Bert:Pre⁃training of deep bidirectional transformers for language understanding. 2018,arXiv:

[22]

Brown T BMann BRyder Net al. Language models are few⁃shot learners. 2020,arXiv:2005. 14165.

[23]

Rao YZhao WZhu Zet al. Global filter networks for image classification. Advances in Neural Information Processing Systems2021,34:980-993.

[24]

Atwood JTowsley D. Diffusion⁃convolutional neural networks. 2016,arXiv:

[25]

Tran D VNavarin NSperduti A. On filter size in graph convolutional networks. 2018,arXiv:1811. 10435

[26]

Gilmer JSchoenholz S SRiley P Fet al. Neural message passing for quantum chemistry∥Proceedings of the 34th International Conference on Machine Learning. Sydney,Australia:2017:1263-1272.

[27]

Huang GLiu ZVan Der Maaten Let al. Densely connected convolutional networksnetworks∥2017 IEEEConference on Computer Vision and Pattern Recognition. Honolulu,HA,USA:IEEE,2017:2261-2269.

[28]

Sandler MHoward AZhu Met al. Mobilenetv2:Inverted residuals and linear bottlenecks∥2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City,UT,USA:IEEE,2018:4510-4520.

[29]

He KZhang XRen Set al. Deep residual learning for image recognition∥2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas,CA,USA:IEEE,2016:770-778.

基金资助

国家自然科学基金(62176131)

AI Summary AI Mindmap
PDF (1407KB)

0

访问

0

被引

详细

导航
相关文章

AI思维导图

/