Development and validation of an explainable machine learning-based risk prediction model for obesity in Chinese

Zekai Chen1, Lin Zhu2,3, Peijie Chen1

  • 1School of Exercise and Health, Shanghai University of Sport, Shanghai, China.

Insights

A new machine learning model accurately predicts childhood obesity risk in China. Key factors include parental BMI, screen time, and physical activity, enabling early intervention for this public health challenge.

Area of Science:

  • Public Health
  • Pediatrics
  • Machine Learning

Background:

  • Childhood obesity is a major global health concern requiring effective prediction tools.
  • Limited interpretable models exist for identifying obesity risk in children and adolescents using national data.
  • This study aimed to develop and validate a predictive model for childhood obesity in China.

Purpose of the Study:

  • To develop and validate an interpretable and user-friendly obesity risk prediction model for Chinese children and adolescents.
  • To utilize nationally representative data for accurate risk assessment.
  • To facilitate early prevention and intervention strategies for childhood obesity.

Main Methods:

  • Utilized cross-sectional data (2017-2018) and temporal validation (2020) from the Physical Activity and Fitness in China-The Youth Study (PAFCTYS).
  • Employed Least Absolute Shrinkage and Selection Operator (LASSO) with recursive feature elimination (RFE) to identify key predictors from 38 candidate variables.
  • Developed and compared eight machine learning algorithms, selecting the Random Forest (RF) model for its superior performance and interpreting it using SHapley Additive exPlanation (SHAP).

Main Results:

  • The Random Forest (RF) model demonstrated excellent performance with an AUC of 0.946 on the test set and 0.810 on the temporal validation set.
  • Key predictors identified by SHAP analysis include parental BMI, weekday mobile device use, moderate-to-vigorous physical activity (MVPA), weekday TV watching, and sex.
  • A web-based risk calculator based on the RF model was successfully deployed.

Conclusions:

  • Developed and validated an explainable machine learning model for predicting childhood obesity risk in China using a large, national sample.
  • The model accurately assesses current obesity risk, aiding healthcare, schools, and parents in large-scale screening.
  • The findings support early identification and intervention for childhood obesity.
Abstract