Related Experiment Video
Updated: Mar 27, 2026
A Machine Learning Approach to Design an Efficient Selective Screening of Mild Cognitive Impairment
Published on: January 11, 2020
Explainable machine learning for long-term cardiovascular disease risk prediction in Chinese middle-aged and older
Xing-Yu Zhu1, Wei Li2, Xu-Yang Pan1
1Department of Cardiovascular Medicine, ShuYang Hospital of Traditional Chinese Medicine, Shu Yang, 223600, Jiangsu Province, China.
Insights
A new machine learning model accurately predicts cardiovascular disease risk in Chinese adults. Waist circumference, triglycerides, age, and hypertension are key predictors, enabling personalized prevention strategies.
Area of Science:
- Cardiovascular epidemiology
- Machine learning in healthcare
- Public health in China
Background:
- Cardiovascular disease (CVD) is the leading cause of death in China.
- Existing CVD risk models are Western-centric and poorly calibrated for Chinese populations.
- Machine learning offers superior prediction but lacks interpretability.
Purpose of the Study:
- Develop an interpretable machine learning tool for long-term CVD risk prediction in Chinese adults.
- Compare ten ML algorithms using the China Health and Retirement Longitudinal Study (CHARLS) data.
- Create an individualized CVD risk assessment tool for Chinese residents aged 45+.
Main Methods:
- Utilized the CHARLS longitudinal dataset (2011-2020) with 8,080 participants aged ≥45 without baseline CVD.
- Employed logistic regression to identify 11 key predictors from 77 variables.
- Trained and validated ten ML models, including random forest, evaluating performance via AUROC, calibration, and decision curves.
- Interpreted model features using SHapley Additive exPlanations (SHAP).
Main Results:
- Incident CVD occurred in 22.0% of the training cohort.
- Key predictors identified: hypertension, waist circumference, dyslipidaemia, and liver disease.
- Random forest model showed superior performance (validation AUC 0.829).
- SHAP analysis highlighted waist circumference, triglycerides, age, and hypertension as primary contributors.
- Psychobehavioral factors (depression, sleep duration) showed independent predictive value.
Conclusions:
- The interpretable random forest model accurately predicts 9-year CVD risk in Chinese middle-aged and elderly individuals.
- Waist circumference is the most critical predictor, followed by triglycerides, age, and hypertension.
- A web-based risk calculator facilitates community screening and personalized CVD prevention, especially in resource-limited settings.
Abstract:
Cardiovascular disease represents the leading cause of mortality in China, accounting for over 40% of all deaths. Existing risk prediction models predominantly derive from Western populations, rendering them suboptimally calibrated for the Chinese middle-aged and elderly demographic. Conventional statistical approaches inadequately capture non-linear associations within high-dimensional data, whilst machine learning models, despite superior performance, suffer from insufficient interpretability. This study leveraged a nationally representative cohort to develop an interpretable machine learning-based tool for long-term cardiovascular risk prediction tailored to the Chinese population. To compare the predictive performance of ten machine learning algorithms using data from the China Health and Retirement Longitudinal Study (CHARLS), identify the optimal model, achieve transparent interpretation through SHapley Additive exPlanations (SHAP) methodology, and develop an individualized cardiovascular risk assessment tool for Chinese residents aged 45 years and above. The study enrolled 8,080 participants aged ≥ 45 years without baseline cardiovascular disease from the CHARLS 2011-2020 longitudinal dataset, with 9-year follow-up. The primary outcome was incident cardiovascular disease. From 77 candidate variables, logistic regression analysis identified 11 predictors: geographical region, hypertension, dyslipidaemia, liver disease, asthma, depression score, age, sleep duration, triglycerides, high-density lipoprotein cholesterol, and waist circumference. The cohort was randomly partitioned into training (n = 5,657, 70%) and validation (n = 2,423, 30%) sets. Ten predictive models were constructed, including random forest, gradient boosting machine, and extreme gradient boosting. Model performance was evaluated using area under the receiver operating characteristic curve (AUROC), calibration plots, and decision curve analysis. Feature contributions were elucidated using SHAP values. Incident cardiovascular disease occurred in 1,246 participants (22.0%) within the training cohort. Multivariable analysis identified hypertension (adjusted OR 1.80), waist circumference (adjusted OR 1.05 per 1-cm increment), dyslipidaemia (adjusted OR 1.42), and liver disease (adjusted OR 1.60) as principal independent predictors. Among ten algorithms evaluated, random forest demonstrated superior performance: validation set AUC 0.829 (95% CI 0.809-0.848), accuracy 0.770, sensitivity 0.681, specificity 0.795. The model exhibited excellent calibration and yielded maximal net clinical benefit across the 10%-85% risk threshold spectrum. SHAP analysis revealed waist circumference as the predominant contributor, followed by triglycerides, age, and hypertension. Psychobehavioural factors (depression, sleep duration) demonstrated independent predictive value. A web-based risk calculator was developed, providing real-time individual 9-year cardiovascular disease probability estimates. The random forest model accurately predicts cardiovascular disease risk in the Chinese middle-aged and elderly population, with waist circumference emerging as the most critical predictor. Translated into an online assessment tool, this model facilitates community-based screening and individualized prevention, offering a pragmatic risk stratification approach for resource-constrained settings.
