Analyzing and forecasting under-5 mortality trends in Bangladesh using machine learning techniques

Background: Under-5 mortality remains a critical social indicator of a country’s development and economic sustainability, particularly in developing nations like Bangladesh. This study employs machine learning models, including Linear Regression, Ridge Regression, Lasso Regression, Bayesian Ridge, Decision Tree, Gradient Boosting, XGBoost, and CatBoost, to forecast future trends in under-5 mortality. By leveraging these models, the study aims to provide actionable insights for policymakers and health professionals to address persistent challenges. Methods: Data from the 1993–94 to 2017–18 Bangladesh Demographic and Health Survey (BDHS) was analyzed using advanced machine learning algorithms. Key metrics, including Mean Absolute Error (MAE), Root Mean Squared Error (RMSE), R-squared, and Mean Absolute Percentage Error (MAPE), were employed to evaluate model performance. Additionally, k-fold cross-validation was conducted to ensure robust model evaluation. Results: This study confirms a significant decline in under-5 mortality in Bangladesh over the study period, with machine learning models providing accurate predictions of future trends. Among the models, Linear Regression emerged as the most accurate, achieving the lowest MAE (4.05), RMSE (4.56), and MAPE (6.64%), along with the highest R-squared value (0.98). Projections indicate further reductions in under-5 mortality to 29.87 per 1,000 live births by 2030 and 26.21 by 2035. Conclusions: From 1994 to 2018, under-5 mortality in Bangladesh decreased by 76.72%. While the Linear Regression model demonstrated exceptional accuracy in forecasting trends, long-term predictions should be interpreted cautiously due to inherent uncertainties in socio-economic conditions. The forecasted rates fall short of the Sustainable Development Goal (SDG) target of 25 deaths per 1,000 live births by 2030, underscoring the need for intensified interventions in healthcare access and maternal health to achieve this target.