Skip to main content
SpringerLink
Log in

An improved ensemble model of hyper parameter tuned ML algorithms for fetal health prediction

  • Original Research
  • Published:
International Journal of Information Technology Aims and scope Submit manuscript

Abstract

Fetal health is a critical concern during pregnancy as it can impact the well-being of both the mother and the baby. Regular monitoring and timely interventions are necessary to ensure the best possible outcomes. While there are various methods to monitor fetal health in the mother's womb, the use of artificial intelligence can improve the accuracy, efficiency, and speed of diagnosis. In this study, we propose a robust ensemble model called ensemble of tuned Support Vector Machine and ExtraTrees (ETSE) for predicting fetal health. Initially, we employed various data preprocessing techniques such as outlier rejection, missing value imputation, data standardization, and data sampling. Then, seven machine learning classifiers including Support Vector Machine, XGBoost, Light Gradient Boosting Machine, Decision Tree, Random Forest, ExtraTrees, and K-Neighbors were implemented. These models were evaluated and then optimized by hyperparameter tuning using the grid search technique. Finally, we analyzed the performance of our proposed ETSE model. The performance analysis of each model revealed that our proposed ETSE model outperformed the other models with 100% precision, 100% recall, 100% F1-score, and 99.66% accuracy. This indicates that the ETSE model can effectively predict fetal health, which can aid in timely interventions and improve outcomes for both the mother and the baby.

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

Access this article

Log in via an institution

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

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3

Similar content being viewed by others

Data availability

The data will be provided upon request.

References

  1. O’Neill E, Thorp J (2012) Antepartum evaluation of the fetus and fetal well being. Clin Obstet Gynecol 55(3):722–730. https://doi.org/10.1097/GRF.0b013e318253b318.PMID:22828105;PMCID:PMC3684248. Accessed 1 Jan 2023

    Article  PubMed  PubMed Central  Google Scholar 

  2. Ocak H, Ertunc H (2012) Prediction of fetal state from the cardiotocogram recordings using adaptive neuro-fuzzy inference systems. Neural Comput Appl 23(6):1583–1589. https://doi.org/10.1007/s00521-012-1110-3

    Article  Google Scholar 

  3. Sundar C (2012) An analysis on the performance of K-means clustering algorithm for cardiotocogram data clustering. Int J Comput Sci Appl 2(5):11–20. https://doi.org/10.5121/ijcsa.2012.2502

    Article  Google Scholar 

  4. Arif M (2015) Classification of cardiotocograms using random forest classifier and selection of important features from cardiotocogram signal. Biomater Biomech Bioeng 2(3):173–183. https://doi.org/10.12989/bme.2015.2.3.173

    Article  Google Scholar 

  5. Ingemarsson I (2009) Fetal monitoring during labor. Neonatology 95(4):342–346

    Article  PubMed  Google Scholar 

  6. News FIGO (1997) Report of the FIGO study group on the assessment of new technology. Int J Gynecol Obstet 59(2):169–173

    Article  Google Scholar 

  7. Miao GJ, Miao KH, Miao JH (2012) Neural pattern recognition model for breast cancer diagnosis. Multidiscip J Sci Technol J Select Areas Bioinform 2(8):1–8

    Google Scholar 

  8. Miao KH, Miao JH, Miao GJ (2016) Diagnosing coronary heart disease using ensemble machine learning. Int J Adv Comput Sci Appl 7(10):30–39

    Google Scholar 

  9. Miao KH, Miao GJ (2013) Mammographic diagnosis for breast cancer biopsy predictions using neural network classification model and receiver operating characteristic (ROC) curve evaluation. Multidiscip J Sci Technol J Select Areas Bioinform 3(9):1–10

    Google Scholar 

  10. Miao JH, Miao KH, Miao GJ (2015) Breast cancer biopsy predictions based on mammographic diagnosis using support vector machine learning. Multidiscip J Sci Technol J Select Areas Bioinform 5(4):1–9

    Google Scholar 

  11. Ayres-de-Campos D, Bernardes J, Garrido A, Marques-de-Sá J, Pereira-Leite L (2000) Sisporto 2.0: a program for automated analysis of cardiotocograms. J Matern Fetal Med 9(5):311–318

    CAS  PubMed  Google Scholar 

  12. Hossen MDA, Tazin T, Khan S, Alam E, Sojib HA, Monirujjaman Khan M, Alsufyani A (2021) Supervised machine learning-based cardiovascular disease analysis and prediction. Math Probl Eng 2021:1–10. https://doi.org/10.1155/2021/1792201

    Article  Google Scholar 

  13. Saboor A, Usman M, Ali S, Samad A, Abrar MF, Ullah N (2022) A method for improving prediction of human heart disease using machine learning algorithms. Mob Inf Syst 2022:1–9. https://doi.org/10.1155/2022/1410169

    Article  Google Scholar 

  14. Sundar C, Chitradevi M, Chitradevi M, Geetharamani G (2012) Classification of cardiotocogram data using neural network based machine learning technique. Int J Comput Appl 47(14):19–25. https://doi.org/10.5120/7256-0279

    Article  Google Scholar 

  15. Ocak H, Ertunc HM (2013) Prediction of fetal state from the cardiotocogram recordings using adaptive neuro-fuzzy inference systems. Neural Comput Appl 23(6):1583–1589

    Article  Google Scholar 

  16. Ocak H (2013) A medical decision support system based on support vector machines and the genetic algorithm for the evaluation of fetal well-being. J Med Syst 37(2):1–9

    Article  Google Scholar 

  17. Peterek T, Gajdoš P, Dohnálek P, Krohová J (2014) Human fetus health classification on cardiotocographic data using random forests. Adv Intell Syst Comput. https://doi.org/10.1007/978-3-319-07773-4_19

    Article  Google Scholar 

  18. Mehbodniya A, Lazar AJP, Webber J, Sharma DK, Jayagopalan SKK, Singh P, Rajan R, Pandya S, Sengan S (2021) Fetal health classification from cardiotocographic data using machine learning. Expert Syst. https://doi.org/10.1111/exsy.12899

    Article  Google Scholar 

  19. Rahmayanti N, Pradani H, Pahlawan M, Vinarti R (2022) Comparison of machine learning algorithms to classify fetal health using cardiotocogram data. Procedia Comput Sci 197:162–171. https://doi.org/10.1016/j.procs.2021.12.130

    Article  Google Scholar 

  20. Alam MT, Khan MAI, Dola NN, Tazin T, Khan MM, Albraikan AA, Almalki FA (2022) Comparative analysis of different efficient machine learning methods for fetal health classification. Appl Bion Biomech 2022:1–12. https://doi.org/10.1155/2022/6321884

    Article  Google Scholar 

  21. The UCI, Machine learning repository, cardiotocography dataset. https://archive.ics.uci.edu/ml/datasets/cardiotocography. Accessed 1 Jan 2023

  22. Chawla NV, Bowyer KW, Hall LO, Kegelmeyer WP (2002) SMOTE: synthetic minority over sampling technique. J Artif Intell Res 16:321–357

    Article  Google Scholar 

  23. Cao XH, Stojkovic I, Obradovic Z (2016) A robust data scaling algorithm to improve classification accuracies in biomedical data. BMC Bioinform 17(1):1–10

    Article  Google Scholar 

  24. Hyperparameter tuning. Grid search and random search. https://www.yourdatateacher.com/2021/05/19/hyperparameter-tuning-grid-search-and-random-search/#:~:text=Grid%20search%20is%20the%20simplest,performance%20metrics%20using%20cross%2Dvalidation. Accessed 1 Jan 2023

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

    MathSciNet  Google Scholar 

  26. Talukder MSH, Sarkar AK (2023) Nutrients deficiency diagnosis of rice crop by weighted average ensemble learning. Smart Agric Technol 4:100155

    Article  Google Scholar 

  27. Kumari S, Kumar D, Mittal M (2021) An ensemble approach for classification and prediction of diabetes mellitus using soft voting classifier. Int J Cogn Comput Eng 2:40–46. https://doi.org/10.1016/j.ijcce.2021.01.001

    Article  Google Scholar 

Download references

Funding

This research received no external funding.

Author information

Authors and Affiliations

  1. Department of Electrical and Electronic Engineering, Rajshahi University of Engineering and Technology, Rajshahi, Bangladesh

    Md. Simul Hasan Talukder

  2. Department of Biomedical Engineering, Jashore University of Science and Technology, Jashore, Bangladesh

    Sharmin Akter

Authors
  1. Md. Simul Hasan Talukder
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Sharmin Akter
    View author publications

    You can also search for this author in PubMed Google Scholar

Contributions

MSHT: Implementation and drafting; SA: Writing, editing and correcting.

Corresponding author

Correspondence to Md. Simul Hasan Talukder.

Ethics declarations

Conflict of interest

The authors declare no conflict of interest.

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

Talukder, M.S.H., Akter, S. An improved ensemble model of hyper parameter tuned ML algorithms for fetal health prediction. Int. j. inf. tecnol. 16, 1831–1840 (2024). https://doi.org/10.1007/s41870-023-01447-9

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s41870-023-01447-9

Keywords

Search

Navigation