Explainable machine learning for depression risk prediction in adults with obesity: development of an online tool

Read full text
Authors: Yong Xie, Yujia Huo, Chunyu Zhang, Jinyu He, Jian Feng
Cited by: 2
References: 59

Abstract

BACKGROUND: Obesity significantly increases the risk of depression, yet interpretable depression risk prediction tools specifically targeting obese individuals remain very limited. This study aims to develop and validate a depression risk prediction model for individuals with obesity using data from the United States National Health and Nutrition Examination Survey (NHANES). METHODS: A total of 6,271 individuals with obesity from the 2005–2020 NHANES cycles were included in this study. Feature selection was conducted using the least absolute shrinkage and selection operator regression and multivariable logistic regression to identify robust predictors. To address class imbalance, the Synthetic Minority Oversampling Technique (SMOTE) was applied to the training dataset. Nine machine-learning(ML) algorithms were developed and compared. Model interpretability was evaluated using SHapley Additive exPlanations (SHAP), and an interactive online calculator was developed based on the best-performing model to support practical clinical risk estimation. RESULTS: Among the nine models, the Stacking-ensemble model demonstrated the highest performance, achieving an AUC of 0.82, along with strong balanced accuracy, F1 score, and Matthews correlation coefficient. SHAP analysis revealed that sleep disturbance, poverty-income ratio (PIR), and gender were the most influential predictors of depression risk. CONCLUSION: ML models can accurately predict depression risk in individuals with obesity. The Stacking-ensemble model showed the highest predictive performance, and an associated online calculator provides clinicians with a practical tool to rapidly estimate individual risk, supporting informed clinical decision-making. CLINICAL TRIAL NUMBER: Not applicable.

Fields of Study & Subjects

Bariatric Surgery and Outcomes Mental Health via Writing Machine Learning in Healthcare

Publication details

Journal / source
BMC Medical Informatics and Decision Making
Publisher
Springer Science and Business Media LLC
Volume
26
Issue
1
Publication date
7 February 2026
Language
EN