Abstract
This study develops explainable artificial intelligence for predicting safe balance using hospital data, including clinical, neurophysiological, and diffusion tensor imaging properties. Retrospective data from 92 first-time stroke patients from January 2016 to June 2023 was analysed. The dependent variables were independent mobility scores, i.e., Berg Balance Scales with 0 (45 or below) vs. 1 (above 45) measured after three and six months, respectively. Twenty-nine predictors were included. Random forest variable importance was employed for identifying significant predictors of the Berg Balance Scale and testing its associations with the predictors, including Berg Balance Scale after one month and corticospinal tract diffusion tensor imaging properties. Shapley Additive Explanation values were calculated to analyse the directions of these associations. The random forest registered a higher or similar area under the curve compared to logistic regression, i.e., 91% vs. 87% (Berg Balance Scale after three months), 92% vs. 92% (Berg Balance Scale after six months). Based on random forest variable importance values and rankings: (1) Berg Balance Scale after three months has strong associations with Berg Balance Scale after one month, Fugl-Meyer assessment scale, ipsilesional corticospinal tract fractional anisotropy, fractional anisotropy laterality index and age; (2) Berg Balance Scale after six months has strong relationships with Fugl-Meyer assessment scale, Berg Balance Scale after one month, ankle plantar flexion muscle strength, knee extension muscle strength and hip flexion muscle strength. These associations were positive in the SHAP summary plots. Including Berg Balance Scale after one month, Fugl-Meyer assessment scale or ipsilesional corticospinal tract fractional anisotropy in the random forest will increase the probability of Berg Balance Scale after three months being above 45 by 0.11, 0.08, or 0.08. In conclusion, safe balance after stroke strongly correlates with its initial motor function, Fugl-Meyer assessment scale, and ipsilesional corticospinal tract fractional anisotropy. Diffusion tensor imaging information aids in developing explainable artificial intelligence for predicting safe balance after stroke.