Noninvasive brain stimulation during EEG improves machine learning classification in chronic stroke

Abstract

Background: In individuals with chronic stroke and hemiparesis, noninvasive brain stimulation (NIBS) may be used as an adjunct to therapy for improving motor recovery. Specific states of movement during motor recovery are more responsive to brain stimulation than others, thus a system that could auto-detect movement state would be useful in correctly identifying the most effective stimulation periods. The aim of this study was to compare the performance of different machine learning models in classifying movement periods during EEG recordings of hemiparetic individuals receiving noninvasive brain stimulation. We hypothesized that transcranial direct current stimulation, a form of NIBS, would modulate brain recordings correlating with movement state and improve classification accuracies above those receiving sham stimulation. Methods: Electroencephalogram data were obtained from 10 participants with chronic stroke and 11 healthy individuals performing a motor task while undergoing transcranial direct current stimulation. Eight traditional machine learning algorithms and five ensemble methods were used to classify two movement states (a hold posture and an arm reaching movement) before, during and after stimulation. To minimize compute times, preprocessing and feature extraction were limited to z-score normalization and power binning into five frequency bands (delta through gamma). Results: Classification of disease state produced significantly higher accuracies in the stimulation (versus sham) group at 78.9% (versus 55.6%, p < 0.000002). We observed significantly higher accuracies when classifying stimulation state in the chronic stroke group (77.6%) relative to healthy controls (64.1%, p < 0.0095). In the chronic stroke cohort, classification of hold versus reach was highest during the stimulation period (75.2%) as opposed to the pre- and post-stimulation periods. Linear discriminant analysis, logistic regression, and decision tree algorithms classified movement state most accurately in participants with chronic stroke during the stimulation period (76.1%). For the ensemble methods, the highest classification accuracy for hold versus reach was achieved using low gamma frequency (30–50 Hz) as a feature (74.5%), although this result did not achieve statistical significance. Conclusions: Machine learning algorithms demonstrated sufficiently high movement state classification accuracy in participants with chronic stroke performing functional tasks during noninvasive brain stimulation. tDCS improved disease state and movement state classification in participants with chronic stroke.