Breastfeeding prediction using machine learning: insights into key predictors and model performance

Publication type

Journal Article

Authors

Publication date

July 15, 2026

Summary:

Background:
Breastfeeding shows persistent disparities shaped by complex maternal, infant, and household factors, making prediction challenging. As traditional models struggle to capture such complexity, this study aims to identify predictors of ever breastfeeding and longer breastfeeding duration using multiple machine learning models and to assess the potential of these methods in breastfeeding prediction.

Methods:
Using nationally representative data from the U.K. Household Longitudinal Study, we apply Lasso (Least Absolute Shrinkage and Selection Operator) logistic regression, decision tree, random forest, XGBoost, support vector machine, and neural network models to examine factors associated with whether infants were ever breastfed and whether breastfeeding stopped at or after 3 months.

Results:
Maternal, household, and birth-related factors all relate to breastfeeding, but maternal socioeconomic factors, especially education and ethnicity, emerge as the strongest predictors. Across the two breastfeeding outcomes, model performance is better for whether infants were ever breastfed than for breastfeeding duration, which is more difficult to predict. Within each outcome, the models show similar performance without substantial differences.

Conclusions:
Breastfeeding outcomes can be reasonably predicted, and the predictive signal is driven primarily by maternal socioeconomic characteristics, enabling targeted support at relatively low informational cost. However, prediction remains limited, especially for breastfeeding duration, suggesting that while key influences are captured, more nuanced or situational mechanisms remain difficult to anticipate. Lasso logistic regression, with its low complexity and transparent structure, performs comparably to more complex models and may therefore be a more appropriate methodological choice for breastfeeding prediction.

Published in

Breastfeeding Medicine

Volume and page numbers

Volume: 21 , p.359 -367

DOI

https://doi.org/10.1177/15568253261450552

ISSN

15568342

Subjects

#589097

News

Latest findings, new research

Publications search

Search all research by subject and author

Podcasts

Researchers discuss their findings and what they mean for society

Projects

Background and context, methods and data, aims and outputs

Events

Conferences, seminars and workshops

Survey methodology

Specialist research, practice and study

Themes

Key research themes and areas of interest