Machine Learning for Motor Imagery Wrist Dorsiflexion Prediction in Brain-Computer Interface Assisted Stroke Rehabilitation

Stroke is a life-changing event that can affect the survivors' physical, cognitive and emotional state. Stroke care focuses on helping the survivors to regain their strength; recover as much functionality as possible and return to independent living through rehabilitation therapies. Automated training protocols have been reported to improve the efficiency of the rehabilitation process. These protocols also decrease the dependency of the process on a professional trainer. Brain-Computer Interface (BCI) based systems are examples of such systems where they make use of the motor imagery (MI) based electroencephalogram (EEG) signals to drive the rehabilitation protocols. In this paper, we have proposed the use of well-known machine learning (ML) algorithms, such as, the decision tree (DT), Naive Bayesian (NB), linear discriminant analysis (LDA), support vector machine (SVM), ensemble learning classifier (ELC), and artificial neural network (ANN) for MI wrist dorsiflexion prediction in a BCI assisted stroke rehabilitation study conducted on eleven stroke survivors with either the left or right paresis. The doubling sub-band selection filter bank common spatial pattern (DSBS-FBCSP) has been proposed as feature extractor and it is observed that the ANN based classifier produces the best results.