Skip to main content

Advertisement

Log in

Optimizing Heart Disease Prediction Using a Hybrid Dynamic Swarm Evolution Approach

  • Original Research
  • Published:
SN Computer Science Aims and scope Submit manuscript

Abstract

Early identification of Cardiovascular issues is crucial due to the high mortality rate associated with heart disease. This research assesses machine learning algorithms for heart disease prediction using tabular datasets. Previous researchers suggested Tree-based models like Random Forest, XGBoost, and Decision Trees and various hyper-parameter optimization methods including Grid Search, Random Search, Swarm and Evolutionary Algorithms which excel in accuracy and robustness. However, they are computationally inefficient and less effective in dynamic settings. A novel Hybrid Swarm Evolution Optimization (HySEOpt) is introduced, which adjusts mutation rates based on performance curves and utilizes parallel processing for faster optimization achieving 98.01% accuracy. HySEOpt enhances model’s quality and robustness, addressing limitations of existing methods and contributes to hyper-parameter optimization in predictive healthcare modeling.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Algorithm 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8

Similar content being viewed by others

Data Availibility Statement

The datasets used in this research is openly accessible for academic use on Kaggle [30].

Materials Availability

Not Applicable.

Code Availability

Not Applicable.

References

  1. Salhi D, Dhai TAE, Kechadi T. Using machine learning for heart disease prediction 2020.https://doi.org/10.1007/978-3-030-69418-0_7

  2. Hossain MI, MARKea MH, Maruf A. Heart disease prediction using distinct artificial intelligence techniques: performance analysis and comparison. Iran J Comput Sci 2023;6:397–417. https://doi.org/10.1007/s42044-023-00148-7.

  3. Muhammad Y, MHea MT. Early and accurate detection and diagnosis of heart disease using intelligent computational model. Sci Rep. 2020;10:19747. https://doi.org/10.1038/s41598-020-76635-9.

  4. Sharma V, S Y, Gupta M. Heart disease prediction using machine learning techniques. In: 2020 2nd International Conference on Advances in Computing, Communication Control and Networking (ICACCCN), 2020;177–181. https://doi.org/10.1109/ICACCCN51052.2020.9362842

  5. Bhatt CM, Patel TGP, Mazzeo PL. Effective heart disease prediction using machine learning techniques. Algorithms. 2023;16(2):88. https://doi.org/10.3390/a16020088.

    Article  Google Scholar 

  6. Chandrasekhar N, Peddakrishna S. Enhancing heart disease prediction accuracy through machine learning techniques and optimization. Processes. 2023;11(4):1210. https://doi.org/10.3390/pr11041210.

    Article  Google Scholar 

  7. Patidar S, J A, Gupta A. Comparative analysis of machine learning algorithms for heart disease predictions. In: 2022 6th International Conference on Intelligent Computing and Control Systems (ICICCS), 2022;1340–1344. https://doi.org/10.1109/ICICCS53718.2022.9788408

  8. Nayeem MJN, Islam MR. Prediction of heart disease using machine learning algorithms. Eur J Artif Intell Mach Learn 2022;1(3):22–26. https://doi.org/10.24018/ejai.2022.1.3.13

  9. Marelli A, Li AC. Machine learning informed diagnosis for congenital heart disease in large claims data source. JACC Adv. 2024;3:2. https://doi.org/10.1016/j.jacadv.2023.100801.

    Article  Google Scholar 

  10. Nissa N, Mohammad S. Heart disease prediction using machine learning techniques. Wesleyan J Res. 2021;13:67.

    Google Scholar 

  11. Valarmathi R, Sheela T. Heart disease prediction using hyper parameter optimization (hpo) tuning. Biomed Signal Process Control. 2021. https://doi.org/10.1016/j.bspc.2021.103033.

    Article  Google Scholar 

  12. Asif D, Bibi MSAM, Mukheimer A. Enhancing heart disease prediction through ensemble learning techniques with hyperparameter optimization. Algorithms. 2023;16(6):308. https://doi.org/10.3390/a16060308.

    Article  Google Scholar 

  13. Bergstra J, Bengio Y. Random search for hyper-parameter optimization. J Mach Learn Res. 2012;13(10):281–305.

    MathSciNet  Google Scholar 

  14. Maclaurin D, Adams RP. Gradient-based hyperparameter optimization through reversible learning 2015. arXiv preprint arXiv:https://arxiv.org/abs/1502.03492

  15. Swersky K, Adams RP. Multi-task bayesian optimization. In: Advances in Neural Information Processing Systems, vol. 26. https://proceedings.neurips.cc/paper/2013/file/f33ba15effa5c10e873bf3842afb46a6-Paper.pdf

  16. Hazan E,K A, Yuan Y. Hyperparameter optimization: A spectral approach 2017. arXiv preprint arXiv: https://arxiv.org/abs/1706.00764

  17. Freitas A. A review of evolutionary algorithms for data mining. 2010. https://doi.org/10.1007/978-0-387-09823-4_19.

  18. Jaderberg M, Dalibard SV. Population based training of neural networks 2017. arXiv preprint arXiv:https://arxiv.org/abs/1711.09846

  19. Ho D, Liang XCIS E, Abbeel P. Population based augmentation: efficient learning of augmentation policy schedules. In: International Conference on Machine Learning, 2019;2731–2741.https://doi.org/10.48550/arXiv.2003.03384

  20. Real E, Liang DSC, Le Q. Automl-zero: evolving machine learning algorithms from scratch. International Conference on Machine Learning, 2020;8007–8019. https://doi.org/10.48550/arXiv.1611.01578

  21. Zoph B, Le QV. Neural architecture search with reinforcement learning 2016. arXiv preprint arXiv:https://arxiv.org/abs/1611.01578

  22. Shahhosseini M, G H, Pham H. Optimizing ensemble weights and hyperparameters of machine learning models for regression problems. Mach Learn Appl 2022 https://doi.org/10.1016/j.mlwa.2022.100251

  23. Folch JP, Lee BSDWCTMvdW R M, Misener R. Combining multi-fidelity modelling and asynchronous batch bayesian optimization. Comput Chem Eng 2023;172:108194. https://doi.org/10.1016/j.compchemeng.2023.108194

  24. Diaz GI, Fokoue-Nkoutche GNA, Samulowitz H. An effective algorithm for hyperparameter optimization of neural networks. IBM J Res Dev 2017;61(4): 9–1. https://doi.org/10.48550/arXiv.1705.08520

  25. Liu H,S K, Yang Y. Darts: differentiable architecture search 2018. arXiv preprint arXiv:https://arxiv.org/abs/1806.09055

  26. Li L, Jamieson GDARK, Talwalkar A. Hyperband: a novel bandit-based approach to hyperparameter optimization. J Mach Learn Res. 2018;18(185):1–52.

    MathSciNet  Google Scholar 

  27. Luo L, Xiong Y, Liu Y, Sun X. Adaptive gradient methods with dynamic bound of learning rate 2019. arXiv preprint arXiv:https://arxiv.org/abs/1902.09843

  28. Ali Y, Awwad E, Al-Razgan M, Maarouf A. Hyperparameter search for machine learning algorithms for optimizing the computational complexity. Processes 2023;11:349. https://doi.org/10.3390/pr11020349

  29. Ibrahim A. Wind speed ensemble forecasting based on deep learning using adaptive dynamic optimization algorithm. IEEE Access 2021;9:125787–125804. https://doi.org/10.1109/ACCESS.2021.3111408

  30. Indicators of Heart Disease (2022 UPDATE). https://www.kaggle.com/datasets/kamilpytlak/personal-key-indicators-of-heart-disease (2022)

Download references

Funding

Not applicable.

Author information

Authors and Affiliations

Authors

Contributions

V.P. conceived the research idea, designed experiments, implemented the various hyper-parameter optimization algorithms, performed data analysis, interpreted results, wrote the manuscript, and created visualizations. B. S. contributed to investigation, provided resources, participated in writing, reviewing, and editing the manuscript, and contributed to data analysis and interpretation of results. A.B. curated and prepared datasets for training and evaluation, and participated in writing, reviewing, and editing the manuscript. A.M. supervised the project, provided assistance in research, reviewed and edited the manuscript, and administered the project.

Corresponding author

Correspondence to Amarjit Malhotra.

Ethics declarations

Conflict of interest

The authors declare no Conflict of interest.

Ethics approval and consent to participate

Not Applicable

Consent of Publication

Not applicable.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Parikh, V., Sharma, B., Byotra, A. et al. Optimizing Heart Disease Prediction Using a Hybrid Dynamic Swarm Evolution Approach. SN COMPUT. SCI. 5, 1104 (2024). https://doi.org/10.1007/s42979-024-03484-y

Download citation

  • Received:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s42979-024-03484-y

Keywords

Navigation