Early Detection of Fetal Health Conditions Using Machine Learning for Classifying Imbalanced Cardiotocographic Data

Abstract

Background: Cardiotocography (CTG) is widely used in obstetrics to monitor fetal heart rate and uterine contractions. It helps detect early signs of fetal distress. However, manual interpretation of CTG can be time-consuming and may vary between clinicians. Recent advances in machine learning provide more efficient and consistent alternatives for analyzing CTG data. Objectives: This study aims to investigate the classification of fetal health using various machine learning models to facilitate early detection of fetal health conditions. Methods: This study utilized a tabular dataset comprising 2126 patient records and 21 features. To classify fetal health outcomes, various machine learning algorithms were employed, including CatBoost, Decision Tree, ExtraTrees, Gradient Boosting, KNN, LightGBM, Random Forest, SVM, ANN and DNN. To address class imbalance and enhance model performance, the Synthetic Minority Oversampling Technique (SMOTE) was employed. Results: Among the tested models, the LightGBM algorithm achieved the highest performance, boasting a classification accuracy of 90.73% and, more notably, a balanced accuracy of 91.34%. This superior balanced accuracy highlights LightGBM's effectiveness in handling imbalanced datasets, outperforming other models in ensuring fair classification across all classes. Conclusions: This study highlights the potential of machine learning models as reliable tools for fetal health classification. The findings emphasize the transformative impact of such technologies on medical diagnostics. Additionally, the use of SMOTE effectively addressed dataset imbalance, further enhancing the reliability and applicability of the proposed approach.