Adaptive Federated Learning with Generative AI for Privacy-Preserving Healthcare
Author(s): PRAYAGA, Lakshmi; PRAYAGA, Chandra
Author(s) keywords: Adaptive Aggregation, Federated Learning, Federated Optimization, Generative AI, Privacy-Preserving Machine Learning, Synthetic Healthcare Data
Reference keywords: artificial intelligence, data privacy, health
Abstract:
Advancements in healthcare monitoring have led to significant data generation, raising urgent concerns about privacy and data sharing. Federated Learning (FL) has emerged as a privacy-preserving solution by enabling decentralized model training across institutions. This study presents an Adaptive Federated Learning framework across four simulated hospital clients with heterogeneous healthcare data. We evaluate three FL strategies - FedAvg, FedProx, and an adaptive FedOpt - highlighting the efficacy of dynamically tuned aggregation using the Adam optimizer. To improve model generalization while preserving privacy, we integrate synthetic datasets generated via Generative AI (ChatGPT and Mostly AI), simulating realistic yet anonymized clinical scenarios. The system includes continuous performance monitoring, particularly addressing the underperformance of one client (Client A4 and B4), ensuring fairness through personalized tuning. Our approach achieves up to 91.3% accuracy and 0.88 F1-score, demonstrating robust performance in non-IID data settings. This work uniquely combines Generative AI with adaptive FL, contributing a scalable, privacy-conscious learning paradigm for future healthcare AI deployments.
References:
[1]. Systematic Review of Privacy-Preserving Distributed Machine Learning From Federated Databases in Health Care - PubMed Central, accessed March 29, 2025, https://pmc.ncbi.nlm.nih.gov/articles/PMC7113079/ Ref: Zerka, F., Barakat, S., Walsh, S., Bogowicz, M., Leijenaar, R. T. H., Jochems, A., Miraglio, B., Townend, D., & Lambin, P. (2020). Systematic Review of Privacy-Preserving Distributed Machine Learning From Federated Databases in Health Care. JCO clinical cancer informatics, 4, 184–200. https://doi.org/10.1200/CCI.19.00047
[2]. Privacy preservation for federated learning in health care - PMC - PubMed Central, accessed March 29, 2025, https://pmc.ncbi.nlm.nih.gov/articles/PMC11284498/ Pati, S., Kumar, S., Varma, A., Edwards, B., Lu, C., Qu, L., Wang, J. J., Lakshminarayanan, A., Wang, S. H., Sheller, M. J., Chang, K., Singh, P., Rubin, D. L., Kalpathy-Cramer, J., & Bakas, S. (2024). Privacy preservation for federated learning in health care. Patterns (New York, N.Y.), 5(7), 100974. https://doi.org/10.1016/j.patter.2024.100974
[3]. Privacy Preserving Federated Learning in Medical Imaging with Uncertainty Estimation, accessed March 29, 2025, https://arxiv.org/html/2406.12815v1 ref: Koutsoubis, N., Yilmaz, Y., Ramachandran, R. P., Schabath, M., & Rasool, G. (2025). Privacy Preserving Federated Learning in Medical Imaging with Uncertainty Estimation. [Conference Name or Journal Name, if applicable].
[4]. Zhang, F., Kreuter, D., Chen, Y., Dittmer, S., Tull, S., Shadbahr, T., BloodCounts! consortium, Preller, J., Rudd, J. H. F., Aston, J. A. D., Schönlieb, C. B., Gleadall, N., & Roberts, M. (2024). Recent methodological advances in federated learning for healthcare. Patterns (New York, N.Y.), 5(6), 101006. https://doi.org/10.1016/j.patter.2024.101006
[5]. Bonawitz, K., Ivanov, V., Kreuter, B., Marcedone, A., McMahan, H. B., Patel, S., ... & Song, S. (2019). Towards federated learning at scale: System design. arXiv preprint arXiv:1907.09693.
[6]. Tian L., Gu Q. In: Proceedings of the 20th International Conference on Artificial Intelligence and Statistics vol. 54 of Proceedings of Machine Learning Research. Singh A., Zhu J., editors. PMLR; 2017. Communication-efficient Distributed Sparse Linear Discriminant Analysis; pp. 1178–1187. https://proceedings.mlr.press/v54/tian17a.html [Google Scholar]
[7]. 16.Yang Q., Liu Y., Chen T., Tong Y. Federated machine learning: Concept and applications. ACM Trans. Intell. Syst. Technol. 2019;10:1–19. doi: 10.1145/3298981. [DOI] [Google Scholar]
[8]. 17.Li T., Sahu A.K., Talwalkar A., Smith V. Federated learning: Challenges, methods, and future directions. IEEE Signal Process. Mag. 2020;37:50–60. doi: 10.1007/978-3-030-85559-8_13. [DOI] [Google Scholar]
[9]. Silva S., Altmann A., Gutman B., Lorenzi M. Domain Adaptation and Representation Transfer, and Distributed and Collaborative Learning. Springer; 2020. Fed-biomed: A general open-source frontend framework for federated learning in healthcare; pp. 201–210. [DOI] [Google Scholar]
[10]. Yang Q., Liu Y., Chen T., Tong Y. Federated machine learning: Concept and applications. ACM Trans. Intell. Syst. Technol. 2019;10:1–19. doi: 10.1145/3298981. [DOI] [Google Scholar]
[11]. A federated learning architecture for secure and private neuroimaging analysis - PMC, accessed March 29, 2025, https://pmc.ncbi.nlm.nih.gov/articles/PMC11 368680/ ref: Stripelis, D., Gupta, U., Saleem, H., Dhinagar, N., Ghai, T., Anastasiou, C., Sánchez, R., Steeg, G. V., Ravi, S., Naveed, M., Thompson, P. M., & Ambite, J. L. (2024). A federated learning architecture for secure and private neuroimaging analysis. Patterns (New York, N.Y.), 5(8), 101031. https://doi.org/10.1016/j.patter.2024.101031
[12]. Reddi, S., Charles, Z., Zaheer, M., Garrett, Z., Rush, K., Konečný, J., Kumar, S., & McMahan, H. B. (2020, February). Adaptive federated optimization. arXiv e-prints. Retrieved from https://doi.org/10.48550/arXiv.2003.00295
[13]. Zhang, C., Xie, Y., Bai, H., Yu, B., Li, W., & Gao, Y. (2021). A survey on federated learning. Knowledge-Based Systems, 216, 106775. https://doi.org/10.1016/j.knosys.2021.106775
[14]. Mammen, P. M. (2021). Federated Learning: Opportunities and Challenges. arXiv preprint arXiv:2101.05428. https://doi.org/10.48550/arXiv.2101.05428
Article Title: Adaptive Federated Learning with Generative AI for Privacy-Preserving Healthcare
Author(s): PRAYAGA, Lakshmi; PRAYAGA, Chandra
Date of Publication: 2025-06-30
Publication: International Journal of Information Security and Cybercrime
ISSN: 2285-9225 e-ISSN: 2286-0096
Digital Object Identifier: 10.19107/IJISC.2025.01.02
Issue: Volume XIV, Issue 1, Year 2025
Section: Advances in Information Security Research
Page Range: 28-36 (9 pages)
Copyright ©2012-2025
The International Journal of Information Security and Cybercrime (IJISC)
All rights reserved
The International Journal of Information Security and Cybercrime is a trademark of the Romanian Association for Information Security Assurance (RAISA).
No part of this publication may be reproduced, stored in a retrieval system, photocopied, recorded or archived, without the written permission from RAISA. When authors submit their papers for publication, they agree that the copyright for their article be transferred to the Romanian Association for Information Security Assurance, if the articles are accepted for publication. The copyright covers the exclusive rights to reproduce and distribute the article, including reprints and translations.