One EEG Diffusion Model Decodes Motor Imagery, Hemiplegic Side and Recovery
Researchers have proposed a unified EEG-based framework that simultaneously performs motor imagery classification, hemiplegic side detection and functional recovery prediction, aimed at motor rehabilitation after stroke. Stroke remains one of the leading causes of long-term motor disability worldwide, the authors note, and motor imagery brain-computer interfaces are seen as a way to accelerate recovery. Current MI-BCI methods, however, generalize poorly across patients, lack an effective functional assessment step, and are limited by scarce patient data and a shortage of suitable augmentation approaches. The framework introduces a diffusion model tailored to the spatio-temporal characteristics of EEG, built on a decoupled neural architecture with rotary spatial encoding and autoregressive temporal fusion. To offset data scarcity, the team designed two augmentation strategies adapted to stroke EEG. Experiments across multiple MI-BCI tasks show superior performance and generalizability, the authors say, supporting the method's potential for personalized stroke rehabilitation.