A Comparative Machine Learning Analysis of Neural Networks and XGBoost for Predicting Maternal Mortality in Maiduguri, Nigeria

Abstract Maternal mortality continues to pose a major public health challenge in low resource and conflict affected settings. This study builds and compares two machine learning classifiers, an Artificial Neural Network (ANN) and Extreme Gradient Boosting (XGBoost), for predicting maternal mortality outcomes using 158 maternal health records collected in Maiduguri, Nigeria (118 survivals and 40 deaths). During data preparation, the Severe Morbidity (SM) variable was found to behave as a deterministic proxy for the outcome and was removed to avoid data leakage, together with a Health Facility Type variable that showed no variation across the sample. Model performance was assessed through repeated stratified 5-fold cross-validation (20 repetitions, 100 folds in total), using accuracy, precision, recall, F1-score, and the area under the receiver operating characteristic curve (AUC). XGBoost attained a mean cross-validated accuracy of 0.980 (precision 1.000, recall 0.920, F1-score 0.953, AUC 1.000), while the ANN reached a mean accuracy of 0.999 (precision 0.997, recall 1.000, F1-score 0.998, AUC 1.000); both exceeded a logistic regression baseline (accuracy 0.902, AUC 0.958). Once the leaked variable was removed, eclampsia, mode of delivery, parity, and gravidity emerged as the strongest predictors of maternal mortality. Although both models classified the available data very well, the limited sample size relative to the number of predictors means the results are best read as evidence of a strong fit to this particular dataset rather than proof of generalizable performance, and external validation is recommended before any clinical application. The findings nonetheless suggest that XGBoost and ANN hold real promise as maternal-mortality risk-stratification tools in resource-constrained settings, pending further validation.

Authors

Institutions

Publication Details

Journal
Research Square
Published
2026-11-04
DOI
https://doi.org/10.21203/rs.3.rs-11173830/v1
Primary Topic
Artificial Intelligence in Healthcare
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

A Comparative Machine Learning Analysis of Neural Networks and XGBoost for Predicting Maternal Mortality in Maiduguri, Nigeria

Muhammad Lefami Zarma, Abubakar Masha
Research Square
Artificial Intelligence in Healthcare
preprint

A Comparative Machine Learning Analysis of Neural Networks and XGBoost for Predicting Maternal Mortality in Maiduguri, Nigeria

Muhammad Lefami Zarma, Abubakar Masha
preprint en

Abstract

Abstract Maternal mortality continues to pose a major public health challenge in low resource and conflict affected settings. This study builds and compares two machine learning classifiers, an Artificial Neural Network (ANN) and Extreme Gradient Boosting (XGBoost), for predicting maternal mortality outcomes using 158 maternal health records collected in Maiduguri, Nigeria (118 survivals and 40 deaths). During data preparation, the Severe Morbidity (SM) variable was found to behave as a deterministic proxy for the outcome and was removed to avoid data leakage, together with a Health Facility Type variable that showed no variation across the sample. Model performance was assessed through repeated stratified 5-fold cross-validation (20 repetitions, 100 folds in total), using accuracy, precision, recall, F1-score, and the area under the receiver operating characteristic curve (AUC). XGBoost attained a mean cross-validated accuracy of 0.980 (precision 1.000, recall 0.920, F1-score 0.953, AUC 1.000), while the ANN reached a mean accuracy of 0.999 (precision 0.997, recall 1.000, F1-score 0.998, AUC 1.000); both exceeded a logistic regression baseline (accuracy 0.902, AUC 0.958). Once the leaked variable was removed, eclampsia, mode of delivery, parity, and gravidity emerged as the strongest predictors of maternal mortality. Although both models classified the available data very well, the limited sample size relative to the number of predictors means the results are best read as evidence of a strong fit to this particular dataset rather than proof of generalizable performance, and external validation is recommended before any clinical application. The findings nonetheless suggest that XGBoost and ANN hold real promise as maternal-mortality risk-stratification tools in resource-constrained settings, pending further validation.

Research Square
Kashim Ibrahim University (NG), Kashim Ibrahim University (NG)
Artificial Intelligence in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon