Personalized Federated Learning with Collaborative Aggregation Networks for Multi-Site Brain Disorder Diagnosis
Qian Si, Yang Li · 2024
In multi-site brain disease diagnosis studies, traditional centralized training methods necessitate sharing medical data, posing significant privacy risks. Federated learning (FL) offers a privacy-preserving solution by enabling global model training through aggregating locally trained models from multiple data centers without sharing raw data. However, current FL approaches rely on a server-based network topology, where central server failure disrupts training. Additionally, data heterogeneity across sites often slows convergence and reduces accuracy. To overcome these issues, we introduce a decentralized personalized federated learning collaborative aggregation network (pFedCAN). This framework has two core components: (1) separating local models into shared and personalized layers, and (2) forming a collaborative aggregation network via similarity detection in the shared layers. Specifically, each center trains its local model, then separates it into shared and personalized layers. The shared layer is exchanged with other centers, while the personalized layer remains local. Data centers analyze similarities in received shared layers to build a collaborative network, where shared layers from similar centers are aggregated to refine the model. This approach flexibly adapts to varying levels of data heterogeneity, enhancing model training efficiency. Validation on public datasets, ABIDE I and ADHD, shows that the proposed method outperforms current leading techniques.