A Federated Deep Learning Framework for 3D Brain MRI Images.

Zhipeng Fan,Jianpo Su,Kai Gao,Dewen Hu,Ling-Li Zeng
DOI: https://doi.org/10.1109/ijcnn52387.2021.9534376
2021-01-01
Abstract:Deep learning has shown unique advantages in multiple applications. However, medical image data such as 3D brain MRI scans contain private information of subjects and cannot be shared and used without special permits. This makes it difficult for aggregated analysis of medical images from multiple sites. Federated learning provides an approach to aggregate learned features from different sites without transferring raw data, ensuring the security of subject information. Traditional federated learning algorithms such as FEDAVG has been proved to be effective, but the performance is always poor due to high data heterogeneity across sites. Moreover, federated learning for 3D medical images has rarely been explored. Here we proposed a Guide-Weighted Federated Deep Learning framework for multisite 3D brain MRI images. We used weighted aggregation on gradients and then used the guide gradients to update federated models to deal with cross-site heterogeneity. The framework was validated on the publicly accessible multi-site brain MRI database of Autism Spectrum Disorder (ASD) - ABIDE I&II. Results showed that federated models outperform local models with improvements in accuracy of 0.92-4.02%. As the first federated learning framework for 3D MRI data, the proposed framework is expected to facilitate imaging data sharing across sites and accelerate the Medical Imaging Alliance's formation. In this way, a robust federated model with better performance can then be generated for future diagnostic classification tasks and assisted in the diagnosing and screening of neuropsychiatric disorders.
What problem does this paper attempt to address?