基于Transformer和图卷积神经网络的自闭症谱系障碍分类
Classification of Autism Spectrum Disorder Based on Transformer and Graph Convolutional Neural Networks
阿日茜 1彭博 2戴亚康 2庞春颖1
作者信息
- 1. 长春理工大学 生命科学技术学院,长春 130022
- 2. 中国科学院苏州生物医学工程技术研究所,苏州 215163
- 折叠
摘要
自闭症谱系障碍(ASD)是一种神经发育障碍,其特征包括社交、沟通和行为方面的严重缺陷.早期准确的诊断和干预对改善患者预后至关重要,然而由于其症状的多样性和复杂性,早期诊断常具有挑战性.近年来,结合功能性磁共振成像(fMRI)等神经影像技术和深度学习算法,为ASD的诊断提供了更准确的手段.脑功能连接网络数据中存在着复杂的局部和全局连接模式,传统的图卷积网络(GCN)更擅长捕捉局部连接模式,而Transformer模型则更适合捕捉全局依赖关系.因此,将这两种模型有效地融合起来以综合利用局部和全局信息对于提高ASD分类的性能至关重要.通过功能性磁共振成像等技术建立脑网络,结合GCN和Transformer模型,并利用数据增强技术,提高了对ASD分类的建模效果.使用联合模型和数据增强技术,在测试集上取得了 87.10 的AUC,显著高于单独使用GCN和Transformer模型的性能.这一研究不仅为ASD的诊断提供了一种高效手段,也有望为深入理解其潜在的神经机制提供重要启示.
Abstract
Autism Spectrum Disorder(ASD)is a neurodevelopmental disorder characterized by severe deficits in social in-teraction,communication,and behavior.Early accurate diagnosis and intervention are crucial for improving patient outcomes,yet early diagnosis is often challenging due to the diversity and complexity of symptoms.In recent years,the combination of neuroimaging techniques such as functional magnetic resonance imaging(fMRI)and deep learning algorithms has provided more accurate means for diagnosing ASD.Functional connectivity network data exhibits complex patterns of both local and global connections,with traditional Graph Convolutional Networks(GCNs)being better at capturing local connectivity patterns,while Transformer models are more suitable for capturing global dependencies.Therefore,effectively integrating these two models to leverage both local and global information is essential for improving the performance of ASD classifica-tion.By establishing brain networks using techniques like functional MRI,combined with GCN and Transformer models,and utilizing data augmentation techniques,we have enhanced the modeling effectiveness for ASD classification.Using the joint model and data augmentation techniques,we achieved an AUC of 87.10 on the test set,significantly higher than the performance of using GCN and Transformer models alone.This study not only provides an efficient approach for the diagnosis of ASD but also holds promise for offering important insights into its underlying neural mechanisms.
关键词
自闭症谱系障碍/功能性磁共振成像/脑功能连接网络/深度学习Key words
autism spectrum disorder/functional magnetic resonance imaging/brain functional connectivity networks/deep learning引用本文复制引用
出版年
2024