Abstract
Advance assessment of the potential functional improvement of patients undergoing a rehabilitation program is crucial in developing precision medicine tools and patient-oriented rehabilitation programs, as well as in better allocating resources in hospitals. In this work, we propose a novel approach to this problem using machine learning algorithms focused on assessing the modified Barthel index (mBI) as an indicator of functional ability. We build four tree-based ensemble machine learning models and train them on a private training cohort of orthopedic (OP) and neurological (NP) hospital discharges. Moreover, we evaluate the models using a validation set for each category of patients using root mean squared error (RMSE) as an absolute error indicator between the predicted mBI and the actual values. The best results obtained from the study are an RMSE of 6.58 for OP patients and 8.66 for NP patients, which shows the potential of artificial intelligence in predicting the functional improvement of patients undergoing rehabilitation.