A Federated Learning Framework with Metaheuristic Optimization for Heart Disease Prediction

PDF (1014KB), PP.51-67

Views: 0 Downloads: 0

Author(s)

Bhaskar Adepu 1,* T. Archana 2

1. Dept. of CSE, Kakatiya University, Telangana, India

2. Dept. of CSE, UCE, Kakatiya University, Bhadradri Kothagudem, Telangana, India

* Corresponding author.

DOI: https://doi.org/10.5815/ijigsp.2026.02.04

Received: 8 May 2025 / Revised: 7 Sep. 2025 / Accepted: 21 Jan. 2026 / Published: 8 Apr. 2026

Index Terms

Cardiovascular Disease, Federated Learning, Lion Optimization

Abstract

Due to lifestyle changes and daily behavioural routines of people living across the globe, cardiovascular diseases (CVD) are increasing in the modern world. In the treatment process, the prediction level of CVD is significantly required. Incorporating machine learning algorithms into CVD prediction can provide advantages such as reduced time consumption in the diagnostic process and improved decision-making. Hence, this research aims to implement a novel Lion-based Federated Learning for Disease Prediction (LbFLDP) technique to predict CVD. The novel approach includes three local hospital models and one centralized global model. The local models are trained using CVD dataset obtained from the kaggle website. After the training phase, the local models are used to predict CVD. These prediction features are then updated in the global model from the local models to enhance the prediction features in the global model. The global model is then initiated for predicting CVD. At this time, the performance of the suggested technique is evaluated in terms of accuracy, F-score, Precision, recall, and error rate. The proposed approach has 98.41 recall, 99.6% accuracy, 98.57 F-score, 98.57 precision, and 0.4% error rate.

Cite This Paper

Bhaskar Adepu, T. Archana, "A Federated Learning Framework with Metaheuristic Optimization for Heart Disease Prediction", International Journal of Image, Graphics and Signal Processing(IJIGSP), Vol.18, No.2, pp. 51-67, 2026. DOI:10.5815/ijigsp.2026.02.04

Reference

[1]S. Rani, A. Kataria, S. Kumar, P. Tiwari. "Federated learning for secure IoMT-applications in smart healthcare systems: A comprehensive review," Knowledge-Based System, vol. 274, pp. 110658, 2023. 
[2]S. Banabilah, M. Aloqaily, E. Alsayed, N. Malik, Y. Jararweh. "Federated learning review: Fundamentals, enabling technologies, and future applications," Information Processing & Management, vol. 59, no. 6, pp. 10306, 2022. 
[3]C. Zhang , Y. Xie, H. Bai, B. Yu, W. Li, Y. Gao. "A survey on federated learning", Knowledge-Based System, vol. 216, pp. 106775, 2021.
[4]A. Rahman, K. Hasan, D. Kundu, M.J. Islam, T. Debnath, S.S. Band, N. Kumar. "On the ICN-IoT with federated learning integration of communication: Concepts, security-privacy issues, applications, and future perspectives," Future Generation Computer Systems, vol. 138, pp. 61-88 2023. 
[5]N. Khalid, A. Qayyum, M. Bilal, A. Al-Fuqaha, J. Qadir. "Privacy-preserving artificial intelligence in healthcare: Techniques and applications," Computers in Biology and Medicine, vol. 158, pp. 106848, 2023. 
[6]R. Baumgartner, P. Arora, C. Bath, D. Burljaev, K. Ciereszko, B. Custers, J. Ding, W. Ernst, E. Fosch-Villaronga, V. Galanos, T. Gremsl. "Fair and equitable AI in biomedical research and healthcare: Social science perspectives," Artificial Intelligence in Medicine, vol. 144, pp. 102658, 2023. 
[7]K. Garikapati, S. Turnbull, R.G. Bennett, T.G. Campbell, J. Kanawati, M.S. Wong, S.P. Thomas, C.K. Chow, S. Kumar. "The role of contemporary wearable and handheld devices in the diagnosis and management of cardiac arrhythmias," Heart, Lung and Circulation, vol. 31, no. 11, pp. 1432-49, 2022. 
[8]S. Bag, P. Dhamija, R.K. Singh, M.S. Rahman, V.R. Sreedharan. "Big data analytics and artificial intelligence technologies based collaborative platform empowering absorptive capacity in health care supply chain: An empirical study," Journal of Business Research, vol. 154, pp. 113315, 2023.
[9]R. Yilmaz, F.G. Yilmaz. "The effect of generative artificial intelligence (AI)-based tool use on students' computational thinking skills, programming self-efficacy, and motivation," Computers and Education: Artificial Intelligence, vol. 4, pp. 100147, 2023. 
[10]S. Banabilah, M. Aloqaily, E. Alsayed, N. Malik, Y. Jararweh. "Federated learning review: Fundamentals, enabling technologies, and future applications," Information Processing & Management, vol. 59, no. 6, pp. 103061, 2022. 
[11]A. Atmakuru, A. Shahini, S. Chakraborty, S. Seoni, M. Salvi, A. Hafeez-Baig, S. Rashid, R. San Tan, P.D. Barua, F. Molinari, U.R. Acharya. "Artificial Intelligence-based Suicide Prevention and Prediction: A Systematic Review (2019-2023)," Information Fusion, vol. 114, pp. 102673, 2024. 
[12]A.V. Sujith, G.S. Sajja, V. Mahalakshmi, S. Nuhmani, B. Prasanalakshmi. "Systematic review of smart health monitoring using deep learning and Artificial intelligence," Neuroscience Informatics, vol. 2, no. 3, pp. 100028, 2022. 
[13]S. Agrawal, S. Sarkar, O. Aouedi, G. Yenduri, K. Piamrat, M. Alazab, S. Bhattacharya, P.K. Maddikunta, T.R. Gadekallu. "Federated learning for intrusion detection system: Concepts, challenges, and future directions," Comput Communication, vol. 195, pp. 346-61, 2022. 
[14]A. Lakhan, T.M. Grønli, G. Muhammad, P. Tiwari. "EDCNNS: Federated learning enabled evolutionary deep convolutional neural network for Alzheimer disease detection," Applied Soft Computing, vol. 147, pp. 110804, 2023. 
[15]N. Khalid, A. Qayyum, M. Bilal, A. Al-Fuqaha, J. Qadir. "Privacy-preserving artificial intelligence in healthcare: Techniques and applications," Computers in Biology and Medicine, vol. 158, pp. 106848, 2023. 
[16]J. Liu, J. Huang, Y. Zhou, X. Li, S. Ji, H. Xiong, D. Dou. "From distributed machine learning to federated learning: A survey," Knowledge and Information Systems, vol. 64, no. 4, pp. 885-917, 2022. 
[17]C. Thapa, S. Camtepe. "Precision health data: Requirements, challenges and existing techniques for data security and privacy," Computers in Biology and Medicine, vol. 129, pp. 104130, 2021. 
[18]F.A. Karim, A.H. Mohd Aman, R. Hassan, K. Nisar [Retracted]. "A Survey on Information‐Centric Networking with Cloud Internet of Things and Artificial Intelligence," Wireless Communications and Mobile Computing, vol. 2022, no. 1, pp. 7818712, 2022. 
[19]S.P. Ramu, P. Boopalan, Q.V. Pham, P.K. Maddikunta, T. Huynh-The, M. Alazab, T.T. Nguyen, T.R. Gadekallu. "Federated learning enabled digital twins for smart cities: Concepts, recent advances, and future directions," Sustainable Cities and Society, vol. 79, pp. 103663, 2022.
[20]M. Javaid, A. Haleem, R.P. Singh, R. Suman, S. Rab. "Significance of machine learning in healthcare: Features, pillars and applications," International Journal of Intelligent Networks, vol. 3, pp. 58-73, 2022. 
[21]A. Rahman, M.S. Hossain, G. Muhammad, D. Kundu, T. Debnath, M. Rahman, M.S. Khan, P. Tiwari, S.S. Band. "Federated learning-based AI approaches in smart healthcare: concepts, taxonomies, challenges, and open issues," Cluster computing, vol. 26, no. 4, pp. 2271-311, 2023. 
[22]D.C. Nguyen, Q.V. Pham, P.N. Pathirana, M. Ding, A. Seneviratne, Z. Lin, O. Dobre, W.J. Hwang. "Federated learning for smart healthcare: A survey," ACM Computing Surveys (Csur), vol. 55, no. 3, pp. 1-37, 2022. 
[23]A. Rauniyar, D.H. Hagos, D. Jha, J.E. Håkegård, U. Bagci, D.B. Rawat, V. Vlassov. "Federated learning for medical applications: A taxonomy, current trends, challenges, and future research directions," IEEE Internet of Things, vol. 11, no. 1, pp. 7374 – 7398, 2023. 
[24]J. Zhou, X. Wang, Y. Li, Y. Yang, J. Shi. "Federated-learning-based prognosis assessment model for acute pulmonary thromboembolism," BMC Medical Informatics and Decision Making, vol. 24, no. 1, pp. 141, 2024. 
[25]F. Efthymiadis, A. Karras, C. Karras, S. Sioutas. "Advanced optimization techniques for federated learning on non-iid data," Future Internet, vol. 16, no. 10, pp. 370, 2024.
[26]Q.T. Pham, H.D. Vu, D.D. Pham, H.H. Pham. "FedKDX: Federated Learning with Negative Knowledge Distillation for Enhanced Healthcare AI Systems," arXiv preprint arXiv: 2601.04587, 2026.
[27]F. Yesmin, "MedHE: Communication-Efficient Privacy-Preserving Federated Learning with Adaptive Gradient Sparsification for Healthcare," arXiv preprint arXiv: 2511.09043, 2025.
[28] G. Yan, L. Xie, X. Gao, W. Zhang, Q. Shen, Y. Fang, Z. Wu, "FedVCK: Non-IID Robust and Communication-Efficient Federated Learning via Valuable Condensed Knowledge for Medical Image Analysis," arXiv preprint arXiv:2412.18557, 2024.
[29]N.M. Eshwarappa, H. Baghban, C.H. Hsu, P.Y. Hsu, R.H. Hwang, M.Y. Chen, "Communication-efficient and privacy-preserving federated learning for medical image classification in multi-institutional edge computing." Journal of Cloud Computing, vol. 14, no. 1, pp. 44, 2025.