Privacy-Preserving Federated Learning Framework for Cardiovascular Disease Risk Prediction under Non-IID Data

Jiazhide Liu

doi:10.54097/kvaaqx37

Authors

Jiazhide Liu Aberdeen School of Artificial Intelligence and Data Science, South China Normal University, Foshan, China

DOI:

https://doi.org/10.54097/kvaaqx37

Keywords:

Federated Learning (FL), Cardiovascular Disease (CVD), K-Means-based, non-IID.

Abstract

With the rapid growth of digital healthcare data and increasing concerns over data security and regulatory compliance, the need for privacy-preserving collaborative learning has become more urgent than ever. Nowadays, Cardiovascular Disease (CVD) has become the leading cause of global mortality, while traditional centralized medical model training is facing problems like severe data privacy barriers and data island. This work designs a privacy-compliant federated learning architecture to realize clinical heart disease risk prediction, FedAvg algorithm will be used to enable cross-institutional collaborative training without sharing raw patient data. At the same time, the research will adopt K-Means based non-IID to simulate real-world medical data heterogeneity. Experimental results show that the proposed framework achieves competitive performance compared with centralized training, the optimal test accuracy is 0.8704, exceeding the result of conventional centralized training. Therefore, this framework can provide a feasible privacy-preserving solution for cross-hospital clinical collaboration and offer a practical approach for future distributed medical risk prediction.

References

[1] Roth, G. A., Mensah, G. A., Johnson, C. O., et al. Global Burden of Cardiovascular Diseases and Risk Factors, 1990-2019: Update From the GBD 2019 Study. Journal of the American College of Cardiology, 2020, 76(25): 2982-3021.

[2] Rieke, N., Hancox, J., Li, W., et al. The future of digital health with federated learning. npj Digital Medicine, 2020, 3: 119.

[3] Sheller, M. J., Edwards, B., Reina, G. A., et al. Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data. Scientific Reports, 2020, 10(1): 12598.

[4] Chen, Y., Shi, Y., Zhang, L., et al. Federated learning for cardiovascular disease risk prediction: A multisite study. Journal of the American Medical Informatics Association, 2023, 30(5): 879-888.

[5] Li, Y., Jiang, X., Wang, H., et al. Federated learning for privacy-preserving healthcare predictive modeling. IEEE Journal of Biomedical and Health Informatics, 2020, 24(11): 3219-3229.

[6] Zhang, L., Wang, B., Li, X., et al. Federated learning for COVID-19 prognosis from chest CT scans across multiple institutions. Nature Communications, 2021, 12(1): 6015.

[7] Dou, Q., So, C. W., Jiang, M., et al. Privacy-preserving federated learning for medical imaging. Nature Machine Intelligence, 2022, 4(3): 239-251.

[8] McMahan, H. B., Moore, E., Ramage, D., et al. Communication-efficient learning of deep networks from decentralized data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, 2017: 1273-1282.

[9] Dua, D. and Graff, C. UCI Machine Learning Repository: Heart Disease Dataset. Irvine, CA: University of California, School of Information and Computer Sciences, 2019. [Online]. Available: https://archive.ics.uci.edu/ml/datasets/heart+disease

[10] Linardos, A., Kushibar, K., Walsh, S., et al. Federated learning for multi-center imaging diagnostics: a simulation study in cardiovascular disease. Scientific Reports, 2022, 12(1): 4743.

[11] Dubey, M., Tembhurne, J. V., Makhijani, R. Enhancing federated learning through differential privacy: Introducing FedHybrid for multicenter diverse heart disease datasets. IEEE Transactions on Emerging Topics in Computational Intelligence, 2025.

[12] Xu, J., Glicksberg, B. S., Su, C., et al. Federated learning for healthcare informatics. Journal of Healthcare Informatics Research, 2021, 5(1): 1-19.

[13] Alom, M. S., Akhi, S. S., Borsha, S. N., et al. Federated machine learning for cardiovascular risk assessment: A decentralized XGBoost approach. 2025 International Conference on Quantum Photonics, Artificial Intelligence, and Networking (QPAIN), 2025.

[14] Zhou, Y., Ram, P., Salonidis, T., et al. Single-shot Hyper-parameter Optimization for Federated Learning: A General Algorithm & Analysis. arXiv preprint arXiv:2202.08338, 2022.

[15] Wang, Z., Kuang, W., Zhang, C., et al. FEDHPO-BENCH: A Benchmark Suite for Federated Hyperparameter Optimization. Proceedings of the 40th International Conference on Machine Learning, 2023: 23451-23468.