Abstract:Although the neonatal mortality rate has been diminished over time in Bangladesh, the rate is still very high. The day of birth is the most vulnerable period for newborns. This study assessed to predict and detect associated risk predictors of the first-day neonatal mortality through Machine Learning (ML) algorithm. We investigated the potential risk factors on the information collected from 43,772 evermarried women of different backgrounds extracted from the 2014 Bangladesh Demographic and Health Survey (BDHS). Based on this demographic and socio-economic risk factors, our goal is to predict the prevalence of first-day neonatal mortality in Bangladesh. Analysis was done using different ML algorithms. Boruta algorithm and Support Vector Machine (SVM) were used to extract the relevant risk factors of the first-day neonatal mortality. Prediction of the prevalence of the first-day neonatal mortality was executed using Decision Tree (DT), Random Forest (RF), SVM, and Logistic Regression (LR), and their performances were appraised using different parameters of confusion matrix, Receiver Operating Characteristics (ROC) curves, and k-fold cross-validation techniques. About two-thirds of the firstday neonatal mortality occurred in the rural area. Male children had high neonatal mortality, with a rate of 51.2%. Mother Age at 1st birth, Husband/partner’s education level, Type of cooking fuel, Total children ever born, Wealth index, Mother’s education level, Access to media, and Type of place of residence were selected as significant risk predictors for predicting the first-day neonatal mortality. Results found that the SVM with Gaussian kernel (Accuracy = 0.8358, Sensitivity = 0. 8637, Specificity = 0.3333, Precision = 0.9588, area under the ROC curve (AUC) = 0.6596, k-fold accuracy=0.8530) performed better among the four machine learning models to predict the firstday neonatal mortality in Bangladesh. Machine learning framework can detect significant predictors of the first-day neonatal mortality, therefore may help the health-policymakers, stakeholders, and family members to understand and prevent this public health problem. Keywords: infant health; decision tree; random forest; support vector machine; feature selection; logistic regression; boruta algorithm; confusion matrix; ROC; k-fold cross-validation.

On the interpretability of the SVM model for predicting infant mortality in Bangladesh

Risk predictors selection and predict for the first-day neonatal mortality in Bangladesh using machine learning techniques

Using Interpretable Machine Learning to Predict Maternal and Fetal Outcomes

Understanding and predicting pregnancy termination in Bangladesh: A comprehensive analysis using a hybrid machine learning approach

Machine Learning Model Interpretability for Precision Medicine

Interpretability of machine learning‐based prediction models in healthcare

Interpretable Predictive Models to Understand Risk Factors for Maternal and Fetal Outcomes

Predictive Modeling for Breast Cancer Classification in the Context of Bangladeshi Patients: A Supervised Machine Learning Approach with Explainable AI

Statistical Insights Into Machine Learning-Based Box Models for Pregnancy Care and Maternal Mortality Reduction: A Literature Survey

Global and local interpretability techniques of supervised machine learning black box models for numerical medical data

Algorithms for interpretable machine learning

Predicting Unintended Pregnancy in Senegal by using Machine Learning Models: Evidence from Senegal DHS 2023

The Influences of Pre-birth Factors in Early Assessment of Child Mortality using Machine Learning Techniques

Interpretable Machine Learning for Survival Analysis

Interpretable machine learning for dermatological disease detection: Bridging the gap between accuracy and explainability

Comparative analysis of explainable machine learning prediction models for hospital mortality

Study of Interpretability in ML Algorithms for Disease Prognosis

In Pursuit of Interpretable, Fair and Accurate Machine Learning for Criminal Recidivism Prediction

A Machine Learning-Based Prediction Model for Preterm Birth in Rural India

What is Interpretable? Using Machine Learning to Design Interpretable Decision-Support Systems

An Explainable Machine Learning Approach to Traffic Accident Fatality Prediction