cs.AIMar 19, 2026

LuMamba: Latent Unified Mamba for Electrode Topology-Invariant and Efficient EEG Modeling

Authors: Danaé Broustail, Anna Tegon, Thorir Mar Ingolfsson, Yawei Li, Luca Benini

Organizations: Integrated Systems Laboratory, ETH Zürich, Zürich, Switzerland · School of Electrical and Electronic Engineering, Nanyang Technological University, Singapore · DEI, University of Bologna, Bologna, Italy

Abstract

Electroencephalography (EEG) enables non-invasive monitoring of brain activity across clinical and neurotechnology applications, yet building foundation models for EEG remains challenging due to differing electrode topologies and computational scalability, as Transformer architectures incur quadratic sequence complexity. As a joint solution, we propose LuMamba (Latent Unified Mamba), a self-supervised framework combining topology-invariant encodings with linear-complexity state-space modeling, using LUNA's learned-query cross-attention mechanism for channel unification, and FEMBA's bidirectional Mamba blocks for efficient temporal modeling. Within this architecture, we provide the first systematic investigation of the Latent-Euclidean Joint-Embedding Predictive Architecture (LeJEPA) for biosignal learning. Pre-trained on over 21,000 hours of unlabeled EEG from the TUEG corpus, LuMamba is evaluated on five downstream tasks spanning abnormality detection, artifact recognition, and mental condition classification across electrode configurations ranging from 16 to 26 channels. In the pre-training objective, masked reconstruction alone yields structured but less generalizable representations, while LeJEPA alone produces diffuse embeddings; combining both objectives achieves the most robust performance. With only 4.6M parameters, LuMamba attains 80.99% balanced accuracy on TUAB and achieves state-of-art performance on Alzheimer's detection (0.97 AUPR), while requiring 377x fewer FLOPS than state-of-art models at equivalent sequence lengths and scaling to 12x longer sequences before reaching typical GPU memory limits. Code is available at https://github.com/pulp-bio/biofoundation.

Figures & tables

Explore similar work

CardsList
  1. Laya: A LeJEPA Approach to EEG via Latent Prediction over Reconstruction

    Mar 17, 2026Saarang Panchavati, Uddhav Panchavati, Hiroki Nariai +2EEG Foundation ModelsSelf-Supervised Time Series Representation Learning

  2. FEMBA on the Edge: Physiologically-Aware Pre-Training, Quantization, and Deployment of a Bidirectional Mamba EEG Foundation Model on an Ultra-low Power Microcontroller

    Mar 18, 2026Anna Tegon, Nicholas Lehmann, Yawei Li +3EEG Foundation ModelsEdge ML

  3. EEG-JEPA: Structured Latent Prediction for EEG Foundation Models

    Jul 31, 2026Jinhao Li, Zhiyuan Ma, Xueqiao Han +8Self-Supervised Time Series Representation LearningJoint-Embedding Predictive Architecture