Finding Neural Biomarkers for Motor Learning and Rehabilitation Using an Explainable Graph Neural Network
Jinpei Han, A. Embs, Federico Nardi, Shlomi Haar, A. Aldo Faisal
Abstract
Human motor learning is a neural process essential for acquiring new motor skills and adapting existing ones, which is fundamental to everyday activities. Neurological disorders such as Parkinson's Disease (PD) and stroke can significantly affect human motor functions. Identifying neural biomarkers for human motor learning is essential for advancing therapeutic strategies for such disorders. However, identifying specific neural biomarkers associated with motor learning has been challenging due to the complex nature of brain activity and the limitations of traditional data analysis techniques. In response to these challenges, we developed a novel Spatial Graph Neural Network (SGNN) model to predict motor learning outcomes from electroencephalogram (EEG) data using the spatial-temporal dynamics of brain activity. We used it to analyse EEG data collected during a visuomotor rotation (VMR) task designed to elicit distinct types of learning: error-based and reward-based. By doing so, we establish a controlled environment that allows for precisely investigating neural signatures associated with these learning processes. To understand the features learned by the SGNN, we used a set of spatial, spectral, and temporal explainability methods to identify the brain regions and temporal dynamics crucial for learning. These approaches offer comprehensive insights into the neural biomarkers, aligning with current literature and ablation studies, and pave the way for applying this methodology to find biomarkers from various brain signals and tasks.