Modified Artificial Bee Colony Based Feature Optimized Federated Learning for Heart Disease Diagnosis in Healthcare
Source abstract
Abstract
Heart disease is one of the lethal diseases causing millions of fatalities every year. The Internet of Medical Things (IoMT) based healthcare effectively enables a reduction in death rate by early diagnosis and detection of disease. The biomedical data collected using IoMT contains personalized information about the patient and this data has serious privacy concerns. To overcome data privacy issues, several data protection laws are proposed internationally. These privacy laws created a huge problem for techniques used in traditional machine learning. We propose a framework based on federated matched averaging with a modified Artificial Bee Colony (M-ABC) optimization algorithm to overcome privacy issues and to improve the diagnosis method for the prediction of heart disease in this paper. The proposed technique improves the prediction accuracy, classification error, and communication efficiency as compared to the state-of-the-art federated learning algorithms on the real-world heart disease dataset.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Privacy-Preserving Technologies in Data · Blockchain Technology Applications and Security · Artificial Intelligence in Healthcare
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, University of Central Punjab, Lahore 54782, 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 Technology, College of Computer and Information Sciences, Princess Nourah bint Abdulrahman University, P.O. Box 84428, Riyadh 11671, Saudi Arabia
Funding
Princess Nourah bint Abdulrahman University Researchers Supporting · Princess Nourah bint Abdulrahman University, Riyadh, Saudi Arabia
References (50)
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 Von dem Bussche, A. (2017). Scope of application of the GDPR. The EU General Data Protection Regulation, Springer.10.1007/978-3-319-57959-7 ↗
7. Wagner, J. (2022, July 20). China’s Cybersecurity Law: What you need to know. The Diplomat, 1 June 2017. Available online: https://thediplomat.com/2017/06/chinas-cybersecurity-law-what-you-need-to-know/.
8. De la Torre, L. (2022, July 21). A Guide to the California Consumer Privacy Act of 2018. SSRN. Available online: https://papers.ssrn.com/sol3/papers.cfm?abstract_id=3275571.10.2139/ssrn.3275571 ↗
9. McMahan, B., and Ramage, D. (2022, August 02). Federated Learning: Collaborative Machine Learning without Centralized Training Data. Google AI Blog. Available online: https://ai.googleblog.com/2017/04/federated-learning-collaborative.html.
10. McMahan, B., Moore, E., Ramage, D., Hampson, S., and Aguera y Arcas, B. (2017, January 20–22). Communication-Efficient Learning of Deep Networks from Decentralized Data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, Lauderdale, FL, USA.
11. Monarch butterfly optimization10.1007/s00521-015-1923-y ↗
12. Slime mould algorithm: A new method for stochastic optimization10.1016/j.future.2020.03.055 ↗
13. Task scheduling in cloud computing based on hybrid moth search algorithm and differential evolution10.1016/j.knosys.2019.01.023 ↗
14. Hunger games search: Visions, conception, implementation, deep analysis, perspectives, and towards performance shifts10.1016/j.eswa.2021.114864 ↗
15. RUN beyond the metaphor: An efficient optimization algorithm based on Runge Kutta method10.1016/j.eswa.2021.115079 ↗
16. Harris hawks optimization: Algorithm and applications10.1016/j.future.2019.02.028 ↗
17. Improving the security of internet of things using cryptographic algorithms: A case of smart irrigation systems10.1007/s12652-020-02303-5 ↗
18. Security of internet of things based on cryptographic algorithms: A survey10.1007/s11276-020-02535-5 ↗
19. Federated Learning: A Survey on Enabling Technologies, Protocols, and Applications10.1109/ACCESS.2020.3013541 ↗
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). 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. Fast-Convergent Federated Learning10.1109/JSAC.2020.3036952 ↗
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. (2020, January 19). Bayesian Inference Federated Learning for Heart Rate Prediction. Proceedings of the International Conference on Wireless Mobile Communication and Healthcare, Virtual Event.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. Artificial bee colony algorithm10.4249/scholarpedia.6915 ↗
31. Zaman, S.K.U., Jehangiri, A.I., Maqsood, T., Haq, N.U., Umar, A.I., Shuja, J., Ahmad, Z., Ben Dhaou, I., and Alsharekh, M.F. (2022). LiMPO: Lightweight mobility prediction and offloading framework using machine learning for mobile edge computing. Clust. Comput., 1–19.10.1007/s10586-021-03518-7 ↗
32. Zaman, S.K.U., Jehangiri, A.I., Maqsood, T., Umar, A.I., Khan, M.A., Jhanjhi, N.Z., Shorfuzzaman, M., and Masud, M. (2022). COME-UP: Computation Offloading in Mobile Edge Computing with LSTM Based User Direction Prediction. Appl. Sci., 12.10.3390/app12073312 ↗
33. Mobility-aware computational offloading in mobile edge networks: A survey10.1007/s10586-021-03268-6 ↗
34. COVID-19 open source data sets: A comprehensive survey10.1007/s10489-020-01862-6 ↗
35. Manimurugan, S., Almutairi, S., Aborokbah, M.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 ↗
36. A Stable AI-Based Binary and Multiple Class Heart Disease Prediction Model for IoMT10.1109/TII.2021.3098306 ↗
37. A Healthcare Monitoring System for the Diagnosis of Heart Disease in the IoMT Cloud Envi-ronment Using MSSO-ANFIS10.1109/ACCESS.2020.3006424 ↗
38. Adaptive Multi-Cost Routing Protocol to Enhance Lifetime for Wireless Body Area Network
39. The IoT-based heart disease monitoring system for pervasive healthcare service10.1016/j.procs.2017.08.265 ↗
40. An IoT Framework for Heart Disease Prediction Based on MDCNN Classifier10.1109/ACCESS.2020.2974687 ↗
41. An Efficient IoT-Based Patient Monitoring and Heart Disease Prediction System Using Deep Learning Modified Neural Network10.1109/ACCESS.2020.3007561 ↗
42. Utilizing IoT wearable medical device for heart disease prediction using higher order Boltzmann model: A classification approach10.1016/j.measurement.2019.07.043 ↗
43. 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 ↗
44. Development of IoT-based mhealth framework for various cases of heart disease patients10.1007/s12553-021-00579-x ↗
45. 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 ↗
46. An IoT based efficient hybrid recommender system for cardiovascular disease10.1007/s12083-019-00733-3 ↗
47. HDPF: Heart Disease Prediction Framework Based on Hybrid Classifiers and Genetic Algorithm10.1109/ACCESS.2021.3122789 ↗
48. Shin, S., Kang, M., Zhang, G., Jung, J., and Kim, Y.T. (2022). Lightweight Ensemble Network for Detecting Heart Disease Using ECG Signals. Appl. Sci., 12.10.3390/app12073291 ↗
49. Ashfaq, Z., Mumtaz, R., Rafay, A., Zaidi, S.M.H., Saleem, H., Mumtaz, S., Shahid, A., De Poorter, E., and Moerman, I. (2022). Embedded AI-Based Digi-Healthcare. Appl. Sci., 12.10.3390/app12010519 ↗
50. A Modified Artificial Bee Colony Algorithm with Firefly Algorithm Strategy for Continuous Optimization Problems10.1155/2018/1237823 ↗