FetalNet 1.0: TOPSIS-Guided Ensemble Learning with Genetic Feature Selection and SHAP Explainability for Fetal Health Classification from Cardiotocography

Fetal health assessment is a crucial aspect of prenatal care, aimed at the early detection of potential complications to ensure optimal outcomes for both mother and child. Traditional methods, such as the visual analysis of cardiotocography (CTG) data by healthcare professionals, are valuable but often subjective and time-consuming. This work investigates the application of machine learning techniques, with a focus on ensemble learning, to enhance the accuracy and efficiency of fetal health classification based on CTG data. Genetic Algorithm (GA) is employed for optimal feature selection, identifying the most discriminative subset of CTG attributes to improve model performance and reduce computational complexity. We employ a combination of advanced machine learning models, including AdaBoost, Gaussian Naïve Bayes, Decision Tree, k-nearest neighbors (KNN), and Logistic Regression. The top two models were selected based on comprehensive performance metrics using the TOPSIS (Technique for Order Preference by Similarity to Ideal Solution) method. These models were then integrated through ensemble learning approaches, such as stacking, Particle Swarm Optimization (PSO) weighted averaging, and soft voting, to improve prediction reliability. Our proposed stacking ensemble model achieves a remarkable accuracy of 97.9%, demonstrating its potential as a robust, data-driven tool for fetal health monitoring and the early identification of at-risk pregnancies. The results indicate that machine learning can effectively complement traditional fetal health assessment methods by providing an objective framework to support clinical decision-making.