Machine learning algorithms to predict depression in older adults in China: a cross-sectional study

ObjectiveThe 2-fold objective of this research is to investigate machine learning's (ML) predictive value for the incidence of depression among China's older adult population and to determine the noteworthy aspects resulting in depression.MethodsThis research selected 7,880 older adult peo...

Full description

Saved in:
Bibliographic Details
Main Authors: Yan Li Qing Song, Lin Chen, Haoqiang Liu, Yue Liu
Format: Article
Language:English
Published: Frontiers Media S.A. 2025-01-01
Series:Frontiers in Public Health
Subjects:
Online Access:https://www.frontiersin.org/articles/10.3389/fpubh.2024.1462387/full
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841556877978107904
author Yan Li Qing Song
Lin Chen
Haoqiang Liu
Yue Liu
author_facet Yan Li Qing Song
Lin Chen
Haoqiang Liu
Yue Liu
author_sort Yan Li Qing Song
collection DOAJ
description ObjectiveThe 2-fold objective of this research is to investigate machine learning's (ML) predictive value for the incidence of depression among China's older adult population and to determine the noteworthy aspects resulting in depression.MethodsThis research selected 7,880 older adult people by utilizing data from the 2020 China Health and Retirement Longitudinal Study. Thereafter, the dataset was classified into training and testing sets at a 6:4 ratio. Six ML algorithms, namely, logistic regression, k-nearest neighbors, support vector machine, decision tree, LightGBM, and random forest, were used in constructing a predictive model for depression among the older adult. To compare the differences in the ROC curves of the different models, the Delong test was conducted. Meanwhile, to evaluate the models' performance, this research performed decision curve analysis (DCA). Thereafter, the Shapely Additive exPlanations values were utilized for model interpretation on the bases of the prediction results' substantial contributions.ResultsThe range of the area under the curve (AUC) of each model's ROC curves was 0.648–0.738, with significant differences (P < 0.01). The DCA results indicate that within various probability thresholds, LightGBM's net benefit was the highest. Self-rated health, nighttime sleep, gender, age, and cognitive function are the five most important characteristics of all models in terms of predicting the occurrence of depression.ConclusionThe occurrence of depression among China's older adult population and the critical factors leading to depression can be predicted and identified, respectively, by ML algorithms.
format Article
id doaj-art-918f70124c9e422c9cd37f5e3c172823
institution Kabale University
issn 2296-2565
language English
publishDate 2025-01-01
publisher Frontiers Media S.A.
record_format Article
series Frontiers in Public Health
spelling doaj-art-918f70124c9e422c9cd37f5e3c1728232025-01-07T05:23:46ZengFrontiers Media S.A.Frontiers in Public Health2296-25652025-01-011210.3389/fpubh.2024.14623871462387Machine learning algorithms to predict depression in older adults in China: a cross-sectional studyYan Li Qing Song0Lin Chen1Haoqiang Liu2Yue Liu3College of Sports, Nanjing Tech University, Nanjing, ChinaCollege of Sports, Nanjing Tech University, Nanjing, ChinaCollege of Sports, Nanjing Tech University, Nanjing, ChinaSchool of Athletic Performance, Shanghai University of Sport, Shanghai, ChinaObjectiveThe 2-fold objective of this research is to investigate machine learning's (ML) predictive value for the incidence of depression among China's older adult population and to determine the noteworthy aspects resulting in depression.MethodsThis research selected 7,880 older adult people by utilizing data from the 2020 China Health and Retirement Longitudinal Study. Thereafter, the dataset was classified into training and testing sets at a 6:4 ratio. Six ML algorithms, namely, logistic regression, k-nearest neighbors, support vector machine, decision tree, LightGBM, and random forest, were used in constructing a predictive model for depression among the older adult. To compare the differences in the ROC curves of the different models, the Delong test was conducted. Meanwhile, to evaluate the models' performance, this research performed decision curve analysis (DCA). Thereafter, the Shapely Additive exPlanations values were utilized for model interpretation on the bases of the prediction results' substantial contributions.ResultsThe range of the area under the curve (AUC) of each model's ROC curves was 0.648–0.738, with significant differences (P < 0.01). The DCA results indicate that within various probability thresholds, LightGBM's net benefit was the highest. Self-rated health, nighttime sleep, gender, age, and cognitive function are the five most important characteristics of all models in terms of predicting the occurrence of depression.ConclusionThe occurrence of depression among China's older adult population and the critical factors leading to depression can be predicted and identified, respectively, by ML algorithms.https://www.frontiersin.org/articles/10.3389/fpubh.2024.1462387/fulldepressionmachine learninghealth promotionCHARLSChina
spellingShingle Yan Li Qing Song
Lin Chen
Haoqiang Liu
Yue Liu
Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
Frontiers in Public Health
depression
machine learning
health promotion
CHARLS
China
title Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
title_full Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
title_fullStr Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
title_full_unstemmed Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
title_short Machine learning algorithms to predict depression in older adults in China: a cross-sectional study
title_sort machine learning algorithms to predict depression in older adults in china a cross sectional study
topic depression
machine learning
health promotion
CHARLS
China
url https://www.frontiersin.org/articles/10.3389/fpubh.2024.1462387/full
work_keys_str_mv AT yanliqingsong machinelearningalgorithmstopredictdepressioninolderadultsinchinaacrosssectionalstudy
AT linchen machinelearningalgorithmstopredictdepressioninolderadultsinchinaacrosssectionalstudy
AT haoqiangliu machinelearningalgorithmstopredictdepressioninolderadultsinchinaacrosssectionalstudy
AT yueliu machinelearningalgorithmstopredictdepressioninolderadultsinchinaacrosssectionalstudy