School of Mathematics Science, Liaocheng University, Liaocheng, Shandong, 252000, China.
Department of Radiology and BRIC, University of North Carolina at Chapel Hill, Chapel Hill, NC, 27599, USA.
Neural Netw. 2024 Jan;169:584-596. doi: 10.1016/j.neunet.2023.11.004. Epub 2023 Nov 7.
Resting-state functional magnetic resonance imaging (rs-fMRI) offers a non-invasive approach to examining abnormal brain connectivity associated with brain disorders. Graph neural network (GNN) gains popularity in fMRI representation learning and brain disorder analysis with powerful graph representation capabilities. Training a general GNN often necessitates a large-scale dataset from multiple imaging centers/sites, but centralizing multi-site data generally faces inherent challenges related to data privacy, security, and storage burden. Federated Learning (FL) enables collaborative model training without centralized multi-site fMRI data. Unfortunately, previous FL approaches for fMRI analysis often ignore site-specificity, including demographic factors such as age, gender, and education level. To this end, we propose a specificity-aware federated graph learning (SFGL) framework for rs-fMRI analysis and automated brain disorder identification, with a server and multiple clients/sites for federated model aggregation and prediction. At each client, our model consists of a shared and a personalized branch, where parameters of the shared branch are sent to the server while those of the personalized branch remain local. This can facilitate knowledge sharing among sites and also helps preserve site specificity. In the shared branch, we employ a spatio-temporal attention graph isomorphism network to learn dynamic fMRI representations. In the personalized branch, we integrate vectorized demographic information (i.e., age, gender, and education years) and functional connectivity networks to preserve site-specific characteristics. Representations generated by the two branches are then fused for classification. Experimental results on two fMRI datasets with a total of 1218 subjects suggest that SFGL outperforms several state-of-the-art approaches.
静息态功能磁共振成像 (rs-fMRI) 提供了一种非侵入性的方法来检查与脑疾病相关的异常脑连接。图神经网络 (GNN) 在 fMRI 表示学习和脑疾病分析中具有强大的图表示能力,因此越来越受欢迎。训练一个通用的 GNN 通常需要来自多个成像中心/站点的大规模数据集,但集中多站点数据通常面临与数据隐私、安全和存储负担相关的固有挑战。联邦学习 (FL) 可以在没有集中多站点 fMRI 数据的情况下进行协作模型训练。不幸的是,以前用于 fMRI 分析的 FL 方法通常忽略了站点特异性,包括年龄、性别和教育水平等人口统计学因素。为此,我们提出了一种用于 rs-fMRI 分析和自动脑疾病识别的感知特异性的联邦图学习 (SFGL) 框架,该框架具有一个服务器和多个客户端/站点,用于联邦模型聚合和预测。在每个客户端,我们的模型由一个共享分支和一个个性化分支组成,其中共享分支的参数被发送到服务器,而个性化分支的参数则保持本地。这可以促进站点之间的知识共享,也有助于保留站点特异性。在共享分支中,我们采用时空注意力图同构网络来学习动态 fMRI 表示。在个性化分支中,我们整合了矢量化的人口统计学信息(即年龄、性别和受教育年限)和功能连接网络,以保留站点特异性特征。然后将两个分支生成的表示融合进行分类。在包含 1218 名受试者的两个 fMRI 数据集上的实验结果表明,SFGL 优于几种最先进的方法。