FedBrain: A robust multi-site brain network analysis framework based on federated learning for brain disease diagnosis | |
Zhang, Chang1; Meng, Xiangzhu2; Liu, Qiang2; Wu, Shu2; Wang, Liang2; Ning, Huansheng1 | |
发表期刊 | NEUROCOMPUTING |
ISSN | 0925-2312 |
2023-11-28 | |
卷号 | 559页码:13 |
摘要 | In recent years, deep learning models have shown their advantages in neuroimage analysis, such as brain disease diagnosis. Unfortunately, it is usually difficult to acquire numerous brain networks at a single centralized site to effectively train a high-quality deep learning model. To address this issue, federated learning (FL) has gained popularity in brain disease diagnosis, which allows deep learning models to be trained without centralizing data. However, most FL-based works might still face two following challenges. Firstly, the high -dimensional features of brain networks are often far larger than sample size, which might lead to poor performance due to the curse of dimensionality. Secondly, differences in data distributions across different sites can impact the communication efficiency and performance of FL models. To overcome these challenges, we design a novel FL framework for diagnosing brain disorders, named FedBrain. Firstly, FedBrain proposes data augmentation based on L1 regularization to select significant features shared by all clients. The domain alignment loss based on the maximum mean discrepancy criterion is introduced to minimize differences in the marginal and conditional distributions between local clients. Furthermore, FedBrain proposes a personalized predictor based on mixture of experts to adapt to different clients, using a global and private predictor as two experts. Eventually, FedBrain integrates the above modules with differential privacy and homomorphic encryption into a unified FL framework. Experimental results on the Autism Brain Imaging Data Exchange (ABIDE) dataset demonstrate its effectiveness and robustness, which shows that FedBrain can reduce the communication burden of FL and achieve the highest average accuracy of 79% against other counterparts. |
关键词 | Functional magnetic resonance image Brain network Federated learning Deep neural networks Brain disease diagnosis |
DOI | 10.1016/j.neucom.2023.126791 |
关键词[WOS] | FMRI ; CONNECTIVITY ; IDENTIFICATION ; MRI ; EEG |
收录类别 | SCI |
语种 | 英语 |
资助项目 | National Natural Science Foundation of China[62141608] ; National Natural Science Foundation of China[62206291] ; National Natural Science Foundation of China[62372454] |
项目资助者 | National Natural Science Foundation of China |
WOS研究方向 | Computer Science |
WOS类目 | Computer Science, Artificial Intelligence |
WOS记录号 | WOS:001079155400001 |
出版者 | ELSEVIER |
七大方向——子方向分类 | 数据挖掘 |
国重实验室规划方向分类 | 智能计算与学习 |
是否有论文关联数据集需要存交 | 否 |
引用统计 | |
文献类型 | 期刊论文 |
条目标识符 | http://ir.ia.ac.cn/handle/173211/53066 |
专题 | 模式识别实验室 多模态人工智能系统全国重点实验室 |
通讯作者 | Meng, Xiangzhu |
作者单位 | 1.Univ Sci & Technol Beijing, Sch Comp & Commun Engn, Beijing 100083, Peoples R China 2.Chinese Acad Sci, Inst Automat, Ctr Res Intelligent Percept & Comp, Beijing, Peoples R China |
通讯作者单位 | 中国科学院自动化研究所 |
推荐引用方式 GB/T 7714 | Zhang, Chang,Meng, Xiangzhu,Liu, Qiang,et al. FedBrain: A robust multi-site brain network analysis framework based on federated learning for brain disease diagnosis[J]. NEUROCOMPUTING,2023,559:13. |
APA | Zhang, Chang,Meng, Xiangzhu,Liu, Qiang,Wu, Shu,Wang, Liang,&Ning, Huansheng.(2023).FedBrain: A robust multi-site brain network analysis framework based on federated learning for brain disease diagnosis.NEUROCOMPUTING,559,13. |
MLA | Zhang, Chang,et al."FedBrain: A robust multi-site brain network analysis framework based on federated learning for brain disease diagnosis".NEUROCOMPUTING 559(2023):13. |
条目包含的文件 | 下载所有文件 | |||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 | ||
1-s2.0-S092523122300(3483KB) | 期刊论文 | 作者接受稿 | 开放获取 | CC BY-NC-SA | 浏览 下载 |
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。
修改评论