Objective <p>This study aims to develop, validate, and visualize a novel machine learning (ML)-based predictive model for depression risk in patients with sleep disorders.</p> Methods <p>Using data from the NHANES (2005–2020), 11 machine learning models were constructed, including Least Absolute Shrinkage and Selection Operator (LASSO), Ridge Regression (Ridge), Elastic Net (ENet), Light Gradient Boosting Machine(LightGBM), Support Vector Machine (SVM), Random Forest (RF), Decision Tree (DT), XGBoost, Logistic Regression (LR), K-Nearest Neighbors (KNN), and Multilayer Perceptron (MLP). Model performance was evaluated using multiple metrics. Decision Curve Analysis (DCA) and calibration curves were used to assess the clinical applicability of the models. SHAP values were applied for model interpretation, and an online web calculator was developed for further model visualization.</p> Results <p>Among the 11 machine learning models, the LightGBM model demonstrated the best performance with an AUC of 0.73. Calibration curves for both the training and test sets confirmed the model’s good calibration. The SHAP summary plot showed that the top three important features in the model were age, poverty-income ratio (PIR), and marital status. The model was integrated into an interactive web application that allows clinicians to predict depression risk based on 10 key clinical variables.</p> Conclusion <p>This study successfully developed a predictive model for depression risk in patients with sleep disorders, demonstrating strong discriminatory ability and good clinical applicability. The online application provides clinicians with a user-friendly tool to assess depression risk and guide targeted prevention and intervention strategies.</p> Clinical trial number <p>Not applicable.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Development, validation, and visualization of a machine learning-based predictive model for depression risk in sleep disorder patients

  • Huiying Wang,
  • Chunyu Zhang,
  • Bo Chen,
  • Yulei Xie,
  • Peng Tian

摘要

Objective

This study aims to develop, validate, and visualize a novel machine learning (ML)-based predictive model for depression risk in patients with sleep disorders.

Methods

Using data from the NHANES (2005–2020), 11 machine learning models were constructed, including Least Absolute Shrinkage and Selection Operator (LASSO), Ridge Regression (Ridge), Elastic Net (ENet), Light Gradient Boosting Machine(LightGBM), Support Vector Machine (SVM), Random Forest (RF), Decision Tree (DT), XGBoost, Logistic Regression (LR), K-Nearest Neighbors (KNN), and Multilayer Perceptron (MLP). Model performance was evaluated using multiple metrics. Decision Curve Analysis (DCA) and calibration curves were used to assess the clinical applicability of the models. SHAP values were applied for model interpretation, and an online web calculator was developed for further model visualization.

Results

Among the 11 machine learning models, the LightGBM model demonstrated the best performance with an AUC of 0.73. Calibration curves for both the training and test sets confirmed the model’s good calibration. The SHAP summary plot showed that the top three important features in the model were age, poverty-income ratio (PIR), and marital status. The model was integrated into an interactive web application that allows clinicians to predict depression risk based on 10 key clinical variables.

Conclusion

This study successfully developed a predictive model for depression risk in patients with sleep disorders, demonstrating strong discriminatory ability and good clinical applicability. The online application provides clinicians with a user-friendly tool to assess depression risk and guide targeted prevention and intervention strategies.

Clinical trial number

Not applicable.