A Federated Learning Approach to Privacy-Preserving Medical Image Classification Across Distributed Healthcare Systems
Keywords:
Federated Learning, Medical Image Classification, Privacy Preservation, Distributed Healthcare, Differential Privacy, Convolutional Neural NetworksAbstract
The surge in medical imaging data and the expansion of distributed healthcare systems have emphasized the need for privacy-preserving machine learning solutions. Traditional centralized approaches to training deep learning models pose risks related to data leakage and non-compliance with health data privacy regulations. Federated Learning (FL) has emerged as a powerful paradigm enabling collaborative model training without raw data sharing. This paper presents a federated deep learning architecture for privacy-preserving classification of medical images, particularly across hospital systems with heterogeneous imaging modalities.
We propose a federated convolutional neural network (CNN) framework using federated averaging (FedAvg) and incorporate differential privacy techniques to enhance data protection. Experimental results on benchmark datasets (e.g., BraTS, ChestXray14) demonstrate that FL achieves near-centralized accuracy while maintaining data locality. The study also explores challenges such as data heterogeneity, communication overhead, and defense against adversarial attacks in federated settings
References
Rieke, N., Hancox, J., Li, W., Milletari, F., Roth, H. R., Albarqouni, S., ... & Cardoso, M. J. (2020). The future of digital health with federated learning. NPJ Digital Medicine, 3(1), 1–7.
Sheller, M. J., Reina, G. A., Edwards, B., Martin, J., & Bakas, S. (2020). Multi-institutional deep learning modeling without sharing patient data: A feasibility study on brain tumor segmentation. Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, 92–104.
Li, T., Sahu, A. K., Zaheer, M., Sanjabi, M., Talwalkar, A., & Smith, V. (2020). Federated optimization in heterogeneous networks. Proceedings of MLSys 2020.
Kaissis, G. A., Makowski, M. R., Rückert, D., & Braren, R. F. (2021). Secure, privacy-preserving and federated machine learning in medical imaging. Nature Machine Intelligence, 3(6), 473–484.
McMahan, H. B., Moore, E., Ramage, D., Hampson, S., & y Arcas, B. A. (2017). Communication-efficient learning of deep networks from decentralized data. Proceedings of AISTATS.
Yang, Q., Liu, Y., Chen, T., & Tong, Y. (2019). Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology (TIST), 10(2), 12.
Kairouz, P., McMahan, H. B., Avent, B., et al. (2021). Advances and open problems in federated learning. Foundations and Trends® in Machine Learning, 14(1–2), 1–210.
Brisimi, T. S., Chen, R., Mela, T., Olshevsky, A., Paschalidis, I. C., & Shi, W. (2018). Federated learning of predictive models from federated Electronic Health Records. Scientific Reports, 8(1), 1–8.
Sheller, M. J., Edwards, B., Reina, G. A., Martin, J., Pati, S., Kotrotsou, A., & Bakas, S. (2020). Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data. Scientific Reports, 10(1), 12598.
Li, X., Gu, Y., Dvornek, N., Staib, L. H., Ventola, P., & Duncan, J. S. (2021). Multi-site fMRI analysis using privacy-preserving federated learning and domain adaptation: ABIDE results. Medical Image Analysis, 65, 101765.
Xu, J., Glicksberg, B. S., Su, C., Walker, P., Bian, J., Wang, F., & Chen, Y. (2021). Federated learning for healthcare informatics. Journal of Healthcare Informatics Research, 5(1), 1–19.
Abadi, M., Chu, A., Goodfellow, I., McMahan, H. B., Mironov, I., Talwar, K., & Zhang, L. (2016). Deep learning with differential privacy. Proceedings of the 2016 ACM SIGSAC Conference on Computer and Communications Security, 308–318.
Vepakomma, P., Gupta, O., Swedish, T., & Raskar, R. (2018). Split learning for health: Distributed deep learning without sharing raw patient data. Proceedings of NeurIPS Workshop on Machine Learning for Health (ML4H).
Dayan, I., Roth, H. R., Zhong, A., Harouni, A., Gentili, A., Abidin, A. Z., ... & Li, W. (2021). Federated learning for predicting clinical outcomes in patients with COVID-19. Nature Medicine, 27(10), 1735–1743.
Choudhury, O., Gkoulalas-Divanis, A., Salonidis, T., Sylla, I., Das, A., Bellet, A., ... & Madabhushi, A. (2020). Differential privacy-enabled federated learning for sensitive health data. arXiv preprint arXiv:2001.10500.
Yang, H., Yu, H., & Yang, Q. (2019). Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology (TIST), 10(2), 1–19.
Liu, Q., Yang, R., Ding, M., & Shikh-Bahaei, M. (2021). Federated learning for intelligent healthcare: A survey. IEEE Access, 9, 103638–10365
