Bidirectional Mamba-2 boosts EEG super-resolution via regression and diffusion.

Abstract

MOTIVATIONS: Electroencephalography (EEG) is a non-invasive method that records brain electrical activity from scalp electrodes, offering millisecond temporal resolution but limited spatial detail due to sparse sensor layouts. RESULTS: We present DiBiMa-EEGSR, a bidirectional Mamba-2 diffusion framework for spatio-temporal EEG super-resolution that reconstructs high-resolution signals from standard low-density recordings without additional hardware. The method formulates super-resolution as conditional generative inference and integrates a diffusion process with a bidirectional state-space backbone to model long-range temporal dependencies with linear complexity. Conditioning on low-resolution inputs, electrode positions and task labels enables anatomically coherent and context-aware reconstruction. A one-step sampling strategy substantially reduces inference time while preserving fidelity. Across two public benchmarks, the approach improves reconstruction accuracy, spatial coherence and spectral preservation over convolutional, transformer-based and prior diffusion models in both spatial and temporal upsampling tasks, providing a scalable pathway toward high-resolution electrophysiological imaging. AVAILABILITY AND IMPLEMENTATION: Code to reproduce ablation experiments, training and evaluation of the proposed BiMa and DiBiMa EEGSR models are available at https://github.com/UgoLomoio/DiBiMa-EEGSR.git. Model weights are available at https://huggingface.co/Ugo96/DiBiMa-EEGSR while an interactive demo for EEG spatial super-resolution using our models can be found at https://huggingface.co/spaces/Ugo96/DiBiMa-EEGSR-Demo.