Cheng Yu, Liu Lin, Gu Xiaoai, Lu Zhonghao, Xia Yujing, Chen Juan, Tang Lin
School of Information, Yunnan Normal University, Yunnan, China.
School of Information, Yunnan Normal University, Yunnan, China; Engineering Research Center of Computer Vision and Intelligent Control Technology, Department of Education of Yunnan Province.
J Biomed Inform. 2023 Oct;146:104484. doi: 10.1016/j.jbi.2023.104484. Epub 2023 Sep 1.
Autism spectrum disorder (ASD) is a pervasive developmental disorder, and the earlier detection and timely intervention for treatment positively affect the prognosis of patients. Deep learning algorithms based on graph structure have achieved good results in autism prediction in recent years. However, there are problems with standardized operations in extracting features and combining neighborhood node features with the structure of the graph dependent, which limits the generalization ability of the trained model to other graph structures. In this paper, we propose a graph fusion autism prediction model based on attentional mechanisms(AGF) to address the above problems. The AGF model represents the overall population (patients or healthy controls) as a sparse graph, where nodes are subjects, and non-imaging features are integrated as edge weights. Different weights can be defined for different nodes in the neighborhood through the attention mechanism without relying on prior knowledge of the graph structure. The model is also able to dynamically fuse multiple sparse graphs obtained from different non-imaging features by way of training weight assignment. Its performance is also compared with several other models (e.g., S-AGF, GCN, etc.), and the results show that it has superior prediction accuracy compared to the baseline model. The results show that this improvement of graph fusion works better on the ABIDE databases, and the classification accuracy can reach 73.9%. The datasets and source code are freely available at https://github.com/chengyu-github1012/Graph-Fusion.git. Strengths and limitations of this study: graph fusion; disease prediction; noise.
自闭症谱系障碍(ASD)是一种广泛性发育障碍,早期发现并及时进行干预治疗对患者的预后有积极影响。近年来,基于图结构的深度学习算法在自闭症预测方面取得了良好效果。然而,在提取特征以及将邻域节点特征与依赖的图结构相结合的过程中存在标准化操作问题,这限制了训练模型对其他图结构的泛化能力。在本文中,我们提出了一种基于注意力机制的图融合自闭症预测模型(AGF)来解决上述问题。AGF模型将总体人群(患者或健康对照)表示为一个稀疏图,其中节点是个体,非成像特征作为边权重进行整合。通过注意力机制,可以为邻域中的不同节点定义不同权重,而无需依赖图结构的先验知识。该模型还能够通过训练权重分配动态融合从不同非成像特征获得的多个稀疏图。其性能也与其他几种模型(如S - AGF、GCN等)进行了比较,结果表明它与基线模型相比具有更高的预测准确率。结果表明,这种图融合改进在ABIDE数据库上效果更好,分类准确率可达73.9%。数据集和源代码可在https://github.com/chengyu - github1012/Graph - Fusion.git上免费获取。本研究的优势与局限:图融合;疾病预测;噪声。