Continuous Model Adaptation in Distributed Healthcare Systems: A MLOps Framework for Federated Learning in Multi-Institutional Cardiovascular Risk Assessment
DOI:
https://doi.org/10.63282/3050-922X.ICAILLMBA-101Keywords:
Mlops, Federated Learning, Healthcare AI, Model Drift Detection, Privacy-Preserving Machine Learning, Cardiovascular Risk PredictionAbstract
When deploying machine learning models across multiple hospitals, we face a critical challenge: patient data is scattered everywhere, models degrade with changing populations, and no one wants to centralize sensitive health information. This paper explores how Machine Learning Operations combined with federated learning can help by letting each institution train models on its own data while sharing only algorithmic improvements, not patient records. We reviewed literature on operational model maintenance and privacy-preserving data sharing in healthcare organizations. Our main finding is that this MLOps-federated approach addresses three essential needs simultaneously: maintaining model performance through automatic retraining when it drifts, protecting patient privacy since raw data never leaves hospitals, and catching bias across diverse patient populations. The challenge lies in handling messy, non-independent clinical data across institutions with different computing resources. The real operational headaches emerge when managing distributed computational resources and when hospitals disagree on measuring model performance fairly. Effective governance requires clear protocols for model versioning, regular performance checks with intervention thresholds, and data quality standards respecting each institution's circumstances. Healthcare systems must invest in infrastructure for automated model management before attempting federated approaches, and regulators need frameworks treating these systems appropriately. This work bridges the gap between theoretical federated learning and actual hospital implementation, providing practical guidance for deploying responsible AI across multiple institutions while maintaining patient privacy.
References
[1] A. Paleyes, R. G. Urma, and N. D. Lawrence, "Challenges in deploying machine learning: a survey of case studies," ACM Computing Surveys, vol. 55, no. 6, pp. 1-29, 2022.
[2] D. S. W. Ting, L. R. Pasquale, L. Peng, J. P. Campbell, A. Y. Lee, R. Raman, G. S. W. Tan, L. Schmetterer, P. A. Keane, and T. Y. Wong, "Artificial intelligence and deep learning in ophthalmology," British Journal of Ophthalmology, vol. 103, no. 2, pp. 167-175, 2019.
[3] S. Rabanser, S. Günnemann, and Z. Lipton, "Failing loudly: An empirical study of methods for detecting dataset shift," Advances in Neural Information Processing Systems, vol. 32, pp. 1396-1408, 2019.
[4] M. Chen, Y. Hao, K. Hwang, L. Wang, and L. Wang, "Disease prediction by machine learning over big data from healthcare communities," IEEE Access, vol. 5, pp. 8869-8879, 2017.
[5] T. Li, A. K. Sahu, A. Talwalkar, and V. Smith, "Federated learning: Challenges, methods, and future directions," IEEE Signal Processing Magazine, vol. 37, no. 3, pp. 50-60, 2020.
[6] N. Rieke, J. Hancox, W. Li, F. Milletari, H. R. Roth, S. Albarqouni, S. Bakas, M. N. Galtier, B. A. Landman, K. Maier-Hein, S. Ourselin, M. Sheller, R. M. Summers, A. Trask, D. Xu, M. Baust, and M. J. Cardoso, "The future of digital health with federated learning," NPJ Digital Medicine, vol. 3, no. 1, pp. 1-7, 2020.
[7] B. McMahan, E. Moore, D. Ramage, S. Hampson, and B. A. Arcas, "Communication-efficient learning of deep networks from decentralized data," Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, pp. 1273-1282, 2017.
[8] R. C. Geyer, T. Klein, and M. Nabi, "Differentially private federated learning: A client level perspective," arXiv preprint arXiv:1712.07557, 2017.
[9] N. Rieke, J. Hancox, W. Li, F. Milletarì, H. R. Roth, S. Albarqouni, S. Bakas, M. N. Galtier, B. A. Landman, K. Maier-Hein, S. Ourselin, M. Sheller, R. M. Summers, A. Trask, D. Xu, M. Baust, and M. J. Cardoso, "The future of digital health with federated learning," NPJ Digital Medicine, vol. 3, art. 119, 2020.
[10] Y. Liu, T. Fan, T. Chen, Q. Xu, and Q. Yang, "FATE: An industrial grade platform for collaborative learning with data protection," Journal of Machine Learning Research, vol. 22, no. 226, pp. 1-6, 2021.
[11] J. Lu, A. Liu, F. Dong, F. Gu, J. Gama, and G. Zhang, "Learning under concept drift: A review," IEEE Transactions on Knowledge and Data Engineering, vol. 31, no. 12, pp. 2346-2363, 2018.
[12] V. Mothukuri, R. M. Parizi, S. Pouriyeh, Y. Huang, A. Dehghantanha, and G. Srivastava, "A survey on security and privacy of federated learning," Future Generation Computer Systems, vol. 115, pp. 619-640, 2021.
[13] C. Zhang, Y. Xie, H. Bai, B. Yu, W. Li, and Y. Gao, "A survey on federated learning," Knowledge-Based Systems, vol. 216, art. 106775, 2021.
[14] D. C. Goff Jr., D. M. Lloyd-Jones, G. Bennett, S. Coady, R. B. D'Agostino, R. Gibbons, P. Greenland, D. T. Lackland, D. M. Levy, C. J. O'Donnell, J. G. Robinson, J. S. Schwartz, S. T. Shero, S. C. Smith Jr., P. Sorlie, N. J. Stone, and P. W. Wilson, "2013 ACC/AHA guideline on the assessment of cardiovascular risk," Circulation, vol. 129, no. 25 suppl 2, pp. S49-S73, 2014.
[15] I. Y. Chen, P. Szolovits, and M. Ghassemi, "Can AI help reduce disparities in general medical and mental health care?," AMA Journal of Ethics, vol. 21, no. 2, pp. 167-179, 2019.