Constructing a prediction model for stroke patients’activities of daily living risk based on interpretable machine learning
CSTR:
Author:
Affiliation:

Clc Number:

R743.3

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Objective:To utilize machine learning algorithms to predict risk factors affecting the activities of daily living(ADL)of stroke patients,providing references for their ADL management decisions. Methods:A retrospective analysis was conducted on 423 stroke patients treated at the Rehabilitation Medicine Center of the First Affiliated Hospital of Nanjing Medical University from January 2015 to February 2019. Patients were categorized into a better ADL group(BI ≥ 60 points)and a poorer ADL group(BI <60 points) based on the Barthel Index(BI)assessment scale,and data preprocessing was performed. Feature variables were selected using colinearity diagnostics and the least absolute shrinkage and selection operator(LASSO). Logistic regression(LR),support vector machine(SVM),random forest(RF),extreme gradient boosting(XGBoost),and K nearest neighbor(KNN)were selected as the five machine learning algorithms for predictive modeling. Afterten-fold cross-validation,the models were comprehensively evalutated using receiver operating characteristic(ROC)curves,area under aerue(AUC),precision recall(PR)curves,area under the precision recall curve(PRAUC),accuracy,sensitivity,and specificity. The Shapley additive interpretation(SHAP)was introduced to interpret the optimal machine learning model. Results:After LASSO regression analysis,16 feature variables were identified for constructing the machine learning model. The RF model demonstrated superior performance with the highest AUC(0.74),PRAUC(0.64),accuracy (0.97),sensitivity(0.75),and specificity(0.97). Interpretive analysis of the SHAP model revealed that among the top 5 features contributing to ADL,Brunnstrom stage(lower limb)exerted the most significant effect,followed by Brunnstrom stage(upper limb),D-dimer,serum albumin level,and age. Conclusion:The RF model emerged as the most effective in predicting ADL in stroke patient, providing valuable references for ADL management decisions in stroke patients.

    Reference
    Related
    Cited by
Get Citation

叶倩,杨云,徐文韬,刘玲玲.基于可解释机器学习构建脑卒中患者日常生活自理能力风险预测模型[J].南京医科大学学报(自然科学版英文版),2024,(5):672-680.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:January 04,2024
  • Revised:
  • Adopted:
  • Online: May 15,2024
  • Published:
Article QR Code