An Attention-based Bidirectional LSTM Model for Continuous Cross-Subject Estimation of Knee Joint Angle during Running from sEMG Signals

Abstract

Accurate and robust estimation of joint kinematics via surface electromyogram (sEMG) signals provides a human-machine interaction (HMI)-based method that can be used to adequately control rehabilitation robots while performing complex movements, such as running, for motor function restoration in affected individuals. To this end, this paper proposes a deep learning-based model (AM-BiLSTM) that integrates a bidirectional long short-term memory (BiLSTM) network and an attention mechanism (AM) for robust estimation of joint kinematics. The proposed model was appraised using knee joint kinematic and sEMG signals collected from fourteen subjects who performed running at the speed of 2 m/s. The proposed model's generalizability was tested for both within- and cross-subject scenarios and compared with long short-term memory (LSTM) and multi-layer perceptron (MLP) networks in terms of normalized root-mean-square error and correlation coefficient metrics. Based on the statistical tests, the proposed AM-BiLSTM model significantly outperformed the LSTM and MLP methods in both within- and cross-subject scenarios (p<0.05) and achieved state-of-the-art performance.Clinical Relevance- The promising results of this study suggest that the AM-BiLSTM model has the potential for continuous cross-subject estimation of lower limb kinematics during running, which can be used to control sEMG-driven exoskeleton robots oriented towards rehabilitation training.