Toward Precision Psychological Rehabilitation: Predicting CBT Efficacy in Post-Stroke Depression Using Machine Learning

Jingyuan Lin^1*Jiansong Yu²

• ¹Fujian Provincial Geriatric Hospital, Fuzhou, China
• ²Taizhou hospital of Zhejiang province affiliated to Wenzhou medical university, Taizhou, China

Objective: This study retrospectively examined the potential benefits of cognitive behavioral therapy (CBT) for post-stroke depression (PSD) and developed an interpretable machine learning model to predict individual treatment response. Methods: Clinical and psychological data from 120 PSD patients receiving CBT and 123 patients in a control group were analyzed. Changes in PHQ-9, GAD-7, and General Self-Efficacy Scale (GSE) scores were compared between groups. Within the CBT cohort, a random forest classifier was trained to predict treatment response and compared with logistic regression and gradient boosting models. SHAP values and ablation analyses were used to assess feature contributions and model interpretability. Results: Baseline characteristics were comparable between groups. The CBT group showed greater improvement in depressive symptoms than the control group. Among predictive models, the random forest classifier demonstrated the highest performance (AUC = 0.897; accuracy = 0.861). SHAP and ablation analyses consistently highlighted baseline depressive severity (PHQ-9), anxiety (GAD-7), self-efficacy (GSE), and social support (SSRS) as the most influential predictors of CBT response. Conclusion: CBT was associated with greater improvement in depressive symptoms among patients with post-stroke depression; however, causal inferences should be made cautiously given the retrospective design. The proposed machine learning model shows preliminary promise for predicting treatment response, but further validation in prospective and multi-center studies is needed before clinical implementation.