Standardized Learning for Applicable Local Medical Systems: A Pilot Assessment

Ahmed Abdalaali

doi:10.30526/39.3.4430

Authors

Ahmed Abdalaali Software Computing and Artificial Intelligence, Computer Science, Islamic Azad University, Qazvin Branch, Iran https://orcid.org/0009-0001-3486-1170

DOI:

https://doi.org/10.30526/39.3.4430

Keywords:

Federated Learning, Medical Systems, Data Privacy, Machine Learning, Experimental Study

Abstract

Federated learning (FL) has become a feasible approach to developing medical prediction models by using distributed institutions without transferring the patient’s raw data to the central server. The pilot experiment presented herewith confirms the usability of FL in local medical systems using the Wisconsin Breast Cancer Diagnostic dataset, which is divided into five clients simulating independent medical institutions under IID and Non-IID settings. A lightweight multilayer perceptron was trained with the Federated Averaging (FedAvg) algorithm and compared to a centralized baseline under the same training conditions. Assessment Metrics for Model Performance: Accuracy, Precision, Recall, F1-Score, and Training Time. The accuracy of the centralized model was 96.49% while the federated model’s accuracy was 94.74% under IID partitioning and 93.86% under Non-IID partitioning. This shows that the performance of the models dropped by an overall 2.6 percentage points in the heterogeneous setting. Notably, recall remained stable at 98.61% across all configurations, suggesting consistent sensitivity in picking up malignant cases. These findings suggest that FL is able to achieve reliable predictive performance while preserving the locality of data and can thus be employed as a method for privacy-aware learning in a collaborative local medical system.

Author Biography

Ahmed Abdalaali, Software Computing and Artificial Intelligence, Computer Science, Islamic Azad University, Qazvin Branch, Iran

-

References

1. Rieke N, Hancox J, Li W, Milletari F, Roth HR, Albarqouni S, Bakas S, Galtier MN, Landman BA, Maier-Hein KH, Ourselin S, Sheller MJ, Summers RM, Trask A, Xu D, Baust M, Cardoso MJ. The future of digital health with federated learning. npj Digit Med. 2020;3(1):119. https://doi.org/10.1038/s41746-020-00323-1

2. Kairouz P, McMahan HB, Avent B, Bellet A, Bennis M, Bhagoji AN, Bonawitz K, Charles Z, Cormode G, Cummings R, D’Oliveira R, Eichner H, El Rouayheb S, Evans D, Gardner J, Garrett Z, Gascón A, Ghazi B, Gibbons PB, Gruteser M, Harchaoui Z, He C, He L, Huo Z, Hutchinson B, Hsu J, Jaggi M, Javidi T, Joshi G, Khodak M, Konecny J, Korolova A, Koushanfar F, Koyejo S, Lepoint T, Liu Y, Mittal P, Mohri M, Nock R, Özgür A, Pagh R, Qi H, Ramage D, Raskar R, Raykova M, Song D, Song W, Stich SU, Sun Z, Suresh AT, Tramèr F, Vepakomma P, Wang J, Xiong L, Xu Z, Yang Q, Yu FX, Yu H, Zhao S. Advances and Open Problems in Federated Learning. Found Trends Mach Learn. 2021;14(1–2):1–210. https://doi.org/10.1561/2200000083

3. McMahan HB, Moore E, Ramage D, Hampson S, Agüera y Arcas B. Communication-efficient learning of deep networks from decentralized data. In: Proc 20th Int Conf Artif Intell Stat (AISTATS). PMLR; 2017. p. 1273–1282. URL: https://proceedings.mlr.press/v54/mcmahan17a.html

4. Sheller MJ, Edwards B, Reina GA, Martin J, Pati S, Kotrotsou A, Milchenko M, Xu W, Marcus D, Colen RR, Bakas S. Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data. Sci Rep. 2020;10:12598. https://doi.org/10.1038/s41598-020-69250-1

5. Dayan I, Roth HR, Zhong A, Harouni A, Gentili A, Abidin AZ. Federated learning for predicting clinical outcomes in patients with COVID-19. Nat Med. 2021;27:1735–1743. https://doi.org/10.1038/s41591-021-01506-3

6. Antunes RS, da Costa CA, Küderle A, Yari IA, Eskofier BM. Federated Learning for Healthcare: Systematic Review and Architecture Proposal. ACM Trans Intell Syst Technol. 2022;13(4):54:1–54:23. https://doi.org/10.1145/3501813

7. Teo ZL, Jin L, Liu N, Li S, Miao D, Zhang X. Federated machine learning in healthcare: A systematic review on clinical applications and technical architecture. Cell Rep Med. 2024;5(2):101419. https://doi.org/10.1016/j.xcrm.2024.101419

8. Bonawitz K, Eichner H, Grieskamp W, Huba D, Ingerman A, Ivanov V, Kiddon C, Konečný J, Mazzocchi S, McMahan HB, Van Overveldt T. Towards federated learning at scale: System design. Proc Mach Learn Syst (MLSys). 2019;1:374–388. URL: https://proceedings.mlsys.org/paper/2019/file/7b8b965ad4bca0e41ab51de7b31363a1-Paper.pdf

9. Brisimi TS, Chen R, Mela T, Olshevsky A, Paschalidis IC, Shi W. Federated learning of predictive models from federated electronic health records. Int J Med Inform. 2018;112:59–67. https://doi.org/10.1016/j.ijmedinf.2018.01.007

10. Liu D, Chen J, Liu Q, Zhou B, Feng J. Federated learning for disease classification using multi-center clinical datasets. IEEE Access. 2020;8:113173–113181. https://doi.org/10.1109/ACCESS.2020.3003723

11. Kaissis GA, Makowski MR, Rückert D, Braren RF. Secure, privacy-preserving and federated machine learning in medical imaging. Nat Mach Intell. 2020;2(6):305–311. https://doi.org/10.1038/s42256-020-0186-1

12. Pfitzner B, Steckhan N, Arnrich B. Federated learning in a medical context: A systematic literature review. ACM Trans Internet Technol. 2021;21(2):1–31. https://doi.org/10.1145/3449390

13. Yang Q, Liu Y, Chen T, Tong Y. Federated machine learning: Concept and applications. ACM Trans Intell Syst Technol. 2019;10(2):1–19. https://doi.org/10.1145/3298981

14. Li T, Sahu AK, Zaheer M, Sanjabi M, Talwalkar A, Smith V. Federated Optimization in Heterogeneous Networks. Proc Mach Learn Syst. 2020;2.

15. Karimireddy SP, Kale S, Mohri M, Reddi S, Stich S, Suresh AT. SCAFFOLD: Stochastic Controlled Averaging for Federated Learning. In: Proc 37th Int Conf Mach Learn (ICML). PMLR. 2020;119:5132–5143.

16. Li X, Jiang M, Zhang X, Kamp M, Dou Q. FedBN: Federated Learning on Non-IID Features via Local Batch Normalization. In: Int Conf Learn Represent (ICLR). 2021.

17. Dua D, Graff C. UCI Machine Learning Repository. Irvine (CA): University of California, School of Information and Computer Sciences; 2019. https://archive.ics.uci.edu/ml

18. Wolberg WH, Street WN, Mangasarian OL. Machine learning techniques to diagnose breast cancer from fine-needle aspirates. Cancer Lett. 1994;77(2–3):163–171. https://doi.org/10.1016/0304-3835(94)90099-X

19. Oh W, Nadkarni GN. Federated Learning in Health care Using Structured Medical Data. Adv Kidney Dis Health. 2023;30(1):4–16. https://doi.org/10.1053/j.akdh.2022.11.007

20. Beutel DJ, Topal T, Mathur A, Qiu X, Fernandez-Marques J, Gao Y. Flower: A Friendly Federated Learning Research Framework. arXiv. 2020;arXiv:2007.14390.

21. Almufareh MF, Tariq N, Humayun M, Almas B. A federated learning approach to breast cancer prediction in a collaborative learning framework. Healthcare. 2023;11(24):3185. https://doi.org/10.3390/healthcare11243185

22. Shukla S, Rajkumar S, Sinha A, Esha M, Elango K, Sampath V. Federated learning with differential privacy for breast cancer diagnosis enabling secure data sharing and model integrity. Sci Rep. 2025;15:13061. https://doi.org/10.1038/s41598-025-95858-2

23. Briola E, Nikolaidis CC, Perifanis V, Pavlidis N, Efraimidis P. A federated explainable AI model for breast cancer classification. In: Proc Eur Interdiscip Cybersecur Conf. 2024. https://doi.org/10.1145/3655693.3660255

24. Jha M, Maitri S, Lohithdakshan M, Duela SJ, Raja K. PrivFED—A framework for privacy-preserving federated learning in enhanced breast cancer diagnosis. arXiv. 2024;2405.08084. Available from: https://arxiv.org/abs/2405.08084