Hybrid Classifier-Based Federated Learning in Health Service Providers for Cardiovascular Disease Prediction
Source abstract
Abstract
One of the deadliest diseases, heart disease, claims millions of lives every year worldwide. The biomedical data collected by health service providers (HSPs) contain private information about the patient and are subject to general privacy concerns, and the sharing of the data is restricted under global privacy laws. Furthermore, the sharing and collection of biomedical data have a significant network communication cost and lead to delayed heart disease prediction. To address the training latency, communication cost, and single point of failure, we propose a hybrid framework at the client end of HSP consisting of modified artificial bee colony optimization with support vector machine (MABC-SVM) for optimal feature selection and classification of heart disease. For the HSP server, we proposed federated matched averaging to overcome privacy issues in this paper. We tested and evaluated our proposed technique and compared it with the standard federated learning techniques on the combined cardiovascular disease dataset. Our experimental results show that the proposed hybrid technique improves the prediction accuracy by 1.5%, achieves 1.6% lesser classification error, and utilizes 17.7% lesser rounds to reach the maximum accuracy.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Artificial Intelligence in Healthcare · Privacy-Preserving Technologies in Data · AI in cancer detection
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Computer Science, HITEC University Taxila, Taxila 47080, Pakistan
Department of Computer Science, COMSATS University Islamabad Abbottabad Campus, Abbottabad 22060, Pakistan
Computer Science Department, Faculty of Computer Sciences, ILMA University, Karachi 75190, Pakistan
Department of Information Systems, College of Computer and Information Sciences, Princess Nourah bint Abdulrahman University, Riyadh 11671, Saudi Arabia
Funding
Princess Nourah bint Abdulrahman University, Riyadh, Saudi Arabia.
References (43)
Show source-provided references
1. Intelligence in the Internet of Medical Things era: A systematic review of current and future trends10.1016/j.comcom.2019.12.030 ↗
2. Big data in healthcare: Management, analysis and future prospects10.1186/s40537-019-0217-0 ↗
3. Rheumatic heart disease worldwide: JACC scientific expert panel10.1016/j.jacc.2018.06.063 ↗
4. Effective Heart Disease Prediction Using Hybrid Machine Learning Techniques10.1109/ACCESS.2019.2923707 ↗
5. Heart Disease Identification Method Using Machine Learning Classification in E-Healthcare10.1109/ACCESS.2020.3001149 ↗
6. Voigt, P., and dem Bussche, A.V. (2017). The EU General Data Protection Regulation, Springer.10.1007/978-3-319-57959-7 ↗
7. Wagner, J. (2022, October 10). China’s Cybersecurity Law: What You Need to Know. The Diplomat. Available online: https://thediplomat.com/2017/06/chinas-cybersecurity-law-what-you-need-to-know/.
8. de la Torre, L. (2022, October 10). A Guide to the California Consumer Privacy Act of 2018. Available online: http://dx.doi.org/10.2139/ssrn.3275571.10.2139/ssrn.3275571 ↗
9. McMahan, B., and Ramage, D. (2022, October 11). Federated Learning: Collaborative Machine Learning without Centralized Training Data. Google AI Blog 2017. Available online: https://ai.googleblog.com/2017/04/federated-learning-collaborative.html.
10. McMahan, B., Moore, E., Ramage, D., Hampson, S., and Areas, B.A.Y. (2017, January 20–22). Communication-Efficient Learning of Deep Networks from Decentralized Data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, Fort Lauderdale, FL, USA.
11. Yaqoob, M.M., Nazir, M., Yousafzai, A., Khan, M.A., Shaikh, A.A., Algarni, A.D., and Elmannai, H. (2022). Modified Artificial Bee Colony Based Feature Optimized Federated Learning for Heart Disease Diagnosis in Healthcare. Appl. Sci., 12.10.3390/app122312080 ↗
12. Psdf: Privacy-aware iov service deployment with federated learning in cloud-edge computing
13. Efficient Federated Learning for Cloud-Based AIoT Applications10.1109/TCAD.2020.3046665 ↗
14. Security of federated learning for cloud-edge intelligence collaborative computing10.1002/int.22992 ↗
15. Task scheduling in cloud computing based on hybrid moth search algorithm and differential evolution10.1016/j.knosys.2019.01.023 ↗
16. Hunger games search: Visions, conception, implementation, deep analysis, perspectives, and towards performance shifts10.1016/j.eswa.2021.114864 ↗
17. A Modified Artificial Bee Colony Algorithm with Firefly Algorithm Strategy for Continuous Optimization Problems10.1155/2018/1237823 ↗
18. A support vector machine classifier with rough set-based feature selection for breast cancer diagnosis10.1016/j.eswa.2011.01.120 ↗
19. Yadav, D.P., Saini, P., and Mittal, P. (2021, January 22–23). Feature Optimization Based Heart Disease Prediction using Machine Learning. Proceedings of the 2021 5th IEEE International Conference on Information Systems and Computer Networks (ISCON), Mathura, India.10.1109/ISCON52037.2021.9702410 ↗
20. A Survey on Federated Learning: The Journey From Centralized to Distributed On-Site Learning and Beyond10.1109/JIOT.2020.3030072 ↗
21. Federated learning of predictive models from federated electronic health records10.1016/j.ijmedinf.2018.01.007 ↗
22. Nilsson, A., Smith, S., Ulm, G., Gustavsson, E., and Jirstrand, M. (2018, January 10–11). A performance evaluation of federated learning algorithms. Proceedings of the Second Workshop on Distributed Infrastructures for Deep Learning (DIDL), Rennes France.10.1145/3286490.3286559 ↗
23. Wang, H., Yurochkin, M., Sun, Y., Papailiopoulos, D., and Khazaeni, Y. (2020). Federated learning with matched averaging. arXiv.
24. Federated Learning: A Survey on Enabling Technologies, Protocols, and Applications10.1109/ACCESS.2020.3013541 ↗
25. Fast-convergent federated learning with class-weighted aggregation10.1016/j.sysarc.2021.102125 ↗
26. Salam, M.A., Taha, S., and Ramadan, M. (2021). COVID-19 detection using federated machine learning. PLoS ONE, 16.10.1371/journal.pone.0252573 ↗
27. A Privacy-Protection Model for Patients10.1155/2020/6647562 ↗
28. Fang, L., Liu, X., Su, X., Ye, J., Dobson, S., Hui, P., and Tarkoma, S. Bayesian Inference Federated Learning for Heart Rate Prediction. Proceedings of the International Conference on Wireless Mobile Communication and Healthcare, Virtual Event, 19 November 2020.10.1007/978-3-030-70569-5_8 ↗
29. Intelligent Computation Offloading for IoT Applications in Scalable Edge Computing Using Artificial Bee Colony Optimization10.1155/2021/5563531 ↗
30. Fast-Convergent Federated Learning10.1109/JSAC.2020.3036952 ↗
31. Manimurugan, S., Almutairi, S., Aborokbah, M., Narmatha, C., Ganesan, S., Chilamkurti, N., Alzaheb, R.A., and Almoamari, H. (2022). Two-Stage Classification Model for the Prediction of Heart Disease Using IoMT and Artificial Intelligence. Sensors, 22.10.3390/s22020476 ↗
32. A Stable AI-Based Binary and Multiple Class Heart Disease Prediction Model for IoMT10.1109/TII.2021.3098306 ↗
33. A Healthcare Monitoring System for the Diagnosis of Heart Disease in the IoMT Cloud Environment Using MSSO-ANFIS10.1109/ACCESS.2020.3006424 ↗
34. Multi-criteria HPC task scheduling on IaaS cloud infrastructures using meta-heuristics10.1007/s10586-020-03168-1 ↗
35. The IoT-based heart disease monitoring system for pervasive healthcare service10.1016/j.procs.2017.08.265 ↗
36. An IoT Framework for Heart Disease Prediction Based on MDCNN Classifier10.1109/ACCESS.2020.2974687 ↗
37. An Efficient IoT-Based Patient Monitoring and Heart Disease Prediction System Using Deep Learning Modified Neural Network10.1109/ACCESS.2020.3007561 ↗
38. Utilizing IoT wearable medical device for heart disease prediction using higher order Boltzmann model: A classification approach10.1016/j.measurement.2019.07.043 ↗
39. Ganesan, M., and Sivakumar, N. (2019, January 29–30). IoT based heart disease prediction and diagnosis model for healthcare using machine learning models. Proceedings of the 2019 IEEE International Conference on System, Computation, Automation and Networking (ICSCAN), Pondicherry, India.10.1109/ICSCAN.2019.8878850 ↗
40. Development of IoT-based mhealth framework for various cases of heart disease patients10.1007/s12553-021-00579-x ↗
41. Gupta, A., Yadav, S., Shahid, S., and Venkanna, U. (2019, January 19–21). HeartCare: IoT Based Heart Disease Prediction System. Proceedings of the 2019 International Conference on Information Technology (ICIT), Bhubaneswar, India.10.1109/ICIT48102.2019.00022 ↗
42. An IoT based efficient hybrid recommender system for cardiovascular disease10.1007/s12083-019-00733-3 ↗
43. Fedesoriano (2022, November 28). Heart Failure Prediction Dataset. Available online: https://www.kaggle.com/fedesoriano/heart-failure-prediction.