Predicting grip strength-related frailty in middle-aged and older Chinese adults using interpretable machine learning models: a prospective cohort study

Lisheng Yu,Shunshun Cao,Botian Song,Yangyang Hu
DOI: https://doi.org/10.3389/fpubh.2024.1489848
IF: 5.2
2024-12-18
Frontiers in Public Health
Abstract:Introduction: Frailty is an emerging global health burden, and there is no consensus on the precise prediction of frailty. We aimed to explore the association between grip strength and frailty and interpret the optimal machine learning (ML) model using the SHapley Additive exPlanation (SHAP) to predict the risk of frailty. Methods: Data for the study were extracted from the China Health and Retirement Longitudinal Study (CHARLS) database. Socio-demographic, medical history, anthropometric, psychological, and sleep parameters were analyzed in this study. We used the least absolute shrinkage and selection operator (LASSO) regression to filter the model for the best predictor variables and constructed six ML models for predicting frailty. The feature performance of six ML models was compared based on the area under the receiver operating characteristic curve (AUROC) and the light gradient boosting machine (LightGBM) model was selected as the best predictive frailty model. We used SHAP to interpret the LightGBM model and to reveal the decision-making process by which the model predicts frailty. Results: A total of 10,834 eligible participants were included in the study. Using the lowest quartile of grip strength as a reference, grip strength was negatively associated with the risk of frailty when grip strength was >29.00 kg for males or >19.00 kg for females ( p < 0.001). The LightGBM model predicted frailty with optimal performance with an AUROC of 0.768 (95% CI 0.741 ~ 0.795). The SHAP summary plot showed that all features predicted frailty in order of importance, with cognitive function being considered the most important predictive feature. The poorer the cognitive function, nighttime sleep duration, body mass index (BMI), and grip strength, the higher the risk of frailty in middle-aged and older adults. The SHAP individual force plot clearly shows that the LightGBM model predicts frailty in the individual decision-making process. Conclusion: The grip strength-related LightGBM prediction model based on SHAP has high accuracy and robustness in predicting the risk of frailty. Increasing grip strength, cognitive function, nighttime sleep duration, and BMI reduce the risk of frailty and may provide strategies for individualized management of frailty.
public, environmental & occupational health
What problem does this paper attempt to address?