A deep learning model with interpretable squeeze-and-excitation for automated rehabilitation exercise assessment.

Abstract

Rehabilitation exercises are critical for recovering from motor dysfunction caused by neurological conditions like stroke, back pain, Parkinson's disease, and spinal cord injuries. Traditionally, these exercises require constant monitoring by therapists, which is time-consuming and costly, often leading to therapist shortages. This paper introduces a deep learning model, convolutional neural network - squeeze excitation (CNN-SE), to automate rehabilitation exercise assessment. By optimizing its parameters with the grey wolf optimization algorithm, the model was fine-tuned for optimal performance. The model's effectiveness was tested on both healthy and unhealthy participants with motor dysfunction, providing a comprehensive evaluation of its capabilities. To interpret the model's decisions and understand its inner workings, we employed Shapley additive explanations (SHAP) to analyze feature importance at each time step. Our CNN-SE model achieved a state-of-the-art mean absolute deviation of 0.127 on the KIMORE dataset and a comparable MAD of 0.014 on the UI-PRMD dataset across various exercises, demonstrating its potential to provide a cost-effective, efficient alternative to traditional therapist-led evaluations.