Prediction of adolescent weight status by machine learning: a population-based study

Abstract

Abstract Background Adolescent weight problems have become a growing public health concern, making early prediction of non-normal weight status crucial for effective prevention. However, few temporal prediction tools for adolescent four weight status have been developed. This study aimed to predict the short- and long-term weight status of Hong Kong adolescents and assess the importance of predictors. Methods A population-based retrospective cohort study of adolescents was conducted using data from a territory-wide voluntary annual health assessment service provided by the Department of Health in Hong Kong. Using diet habits, physical activity, psychological well-being, and demographics, we generated six prediction models for successive weight status (normal, overweight, obese and underweight) using multiclass Decision Tree, Random Forest, k-Nearest Neighbor, eXtreme gradient boosting, support vector machine, logistic regression. Model performance was evaluated by multiple standard classifier metrics and the overall accuracy. Predictors’ importance was assessed using Shapley values. Results 442,898 Primary 4 (P4, Grade 4 in the US) and 344,186 in Primary 6 (P6, Grade 6 in the US) students, with followed up until their Secondary 6 (Grade 12 in the US) during the academic years 1995/96 to 2014/15 were included. The XG Boosts model outperformed consistently all other models, predicting the long-term weight status at S6 from P4 or P6 with an overall accuracy of 0.89 or 0.90, respectively. XG Boost also achieved better model performance on each prediction window, and yielded obese group 0.96 precision, 0.98 recall, and 0.97 F-1 score both at S6 by P4 or P6. Weight, height, sex, age, frequency and hours of aerobic exercise were consistently the most important predictors for both cohorts. Conclusions The machine learning approaches accurately predict adolescent weight status in both short- and long-term. The developed multiclass model that utilizing easy-assessed variables enables accurate long-term prediction on weight status, which can be used by adolescents and parents for self-prediction when applied in health care system. The interpretable models may help to provide the early and individualized interventions suggestions for adolescents with weight problems particularly.