用图神经网络融合多模态脑影像,提升自闭症分类准确率
Enhanced Graph Convolutional Network with Chebyshev Spectral Graph and Graph Attention for Autism Spectrum Disorder Classification
- 结合切比雪夫谱卷积与注意力机制,高效建模脑区连接关系
- 在ABIDE I数据集上达到74.82%准确率和0.82的AUC
- 适合做脑科学与医疗人工智能交叉研究的参考
自闭症谱系障碍(ASD)是一种复杂的神经发育障碍,临床表现和神经基础差异大,导致早期客观诊断极为困难。本文提出一种融合切比雪夫谱图卷积与图注意力网络(GAT)的图卷积网络(GCN)模型,利用多模态神经影像与表型数据提升ASD分类精度。基于包含870名患者静息态功能磁共振(rs-fMRI)、结构磁共振(sMRI)及表型变量的ABIDE I数据集,模型采用多分支架构分别处理各模态数据,再通过拼接融合。基于站点相似性构建人群图结构,捕捉个体间关系。切比雪夫多项式滤波实现低复杂度局部谱学习,而GAT层通过注意力加权聚合邻域信息增强节点表征。模型在总输入维度为5,206的条件下,采用分层五折交叉验证训练。实验表明,该模型在全数据集上测试准确率达74.82%,AUC为0.82,优于多种先进基线方法,包括传统GCN、基于自编码器的深度神经网络及多模态CNN。
原文摘要 · Abstract (English)
ASD is a complicated neurodevelopmental disorder marked by variation in symptom presentation and neurological underpinnings, making early and objective diagnosis extremely problematic. This paper presents a Graph Convolutional Network (GCN) model, incorporating Chebyshev Spectral Graph Convolution and Graph Attention Networks (GAT), to increase the classification accuracy of ASD utilizing multimodal neuroimaging and phenotypic data. Leveraging the ABIDE I dataset, which contains resting-state functional MRI (rs-fMRI), structural MRI (sMRI), and phenotypic variables from 870 patients, the model leverages a multi-branch architecture that processes each modality individually before merging them via concatenation. Graph structure is encoded using site-based similarity to generate a population graph, which helps in understanding relationship connections across individuals. Chebyshev polynomial filters provide localized spectral learning with lower computational complexity, whereas GAT layers increase node representations by attention-weighted aggregation of surrounding information. The proposed model is trained using stratified five-fold cross-validation with a total input dimension of 5,206 features per individual. Extensive trials demonstrate the enhanced model's superiority, achieving a test accuracy of 74.82\% and an AUC of 0.82 on the entire dataset, surpassing multiple state-of-the-art baselines, including conventional GCNs, autoencoder-based deep neural networks, and multimodal CNNs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。