FEMBA on the Edge: Physiologically-Aware Pre-Training, Quantization, and Deployment of a Bidirectional Mamba EEG Foundation Model on an Ultra-low Power Microcontroller
On the map: State-space sequence models (S4, Mamba) → Physiological signal modeling
Abstract (arXiv)
Objective: To enable continuous, long-term neuro-monitoring on wearable devices by overcoming the computational bottlenecks of Transformer-based Electroencephalography (EEG) foundation models and the quantization challenges inherent to State-Space Models (SSMs). Methods: We present FEMBA, a bidirectional Mamba architecture pre-trained on over 21,000 hours of EEG. We introduce a novel Physiologically-Aware pre-training objective, consisting of a reconstruction with low-pass filtering, to prioritize neural oscillations over high-frequency artifacts. To address the activation outliers common in SSMs, we employ Quantization-Aware Training (QAT) to compress the model to 2-bit weights. The framework is deployed on a parallel ultra-low-power RISC-V microcontroller (GAP9) using a custom double-buffered memory streaming scheme. Results: The proposed low-pass pre-training improves downstream AUROC on TUAB from 0.863 to 0.893 and AUPR from 0.862 to 0.898 compared to the best contrastive baseline. QAT successfully compresses weights with negligible performance loss, whereas standard post-training quantization degrades accuracy by approximately 30%. The embedded implementation achieves deterministic real-time inference (1.70~s per 5~s window) and reduces the memory footprint by 74% (to ≈2~MB), achieving competitive accuracy with up to 27× fewer FLOPs than Transformer benchmarks. Conclusion: FEMBA demonstrates that Mamba-based foundation models can be effectively quantized and deployed on extreme-edge hardware without sacrificing the representation quality required for robust clinical analysis. Significance: This work establishes the first full-stack framework for deploying large-scale EEG foundation models on ultra-low-power wearables, facilitating continuous, SSM based monitoring for epilepsy and sleep disorders.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Physiological signal modeling, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 32 of its 65 references that are in the map. First posted on arXiv 18 Mar 2026.
Builds on
32 of 65 references.
Closest ancestors · 3
Same direction, newest first
- 2025FEMBA: Efficient and Scalable EEG Analysis with a Bidirectional Mamba Foundation Model10 Feb
- 2024EEGMamba: Bidirectional State Space Model with Mixture of Experts for EEG Multi-task Classification20 Jul
- 2022Modeling Multivariate Biosignals With Graph Neural Networks and Structured State Space Models21 Nov
Other references · 29
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2016EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfacescited by 616
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 390
- 2024Large Brain Model for Learning Generic Representations with Tremendous EEG Data in BCIcited by 311
- 2021BENDR: using transformers and a contrastive self-supervised learning task to learn from massive amounts of EEG datacited by 194
- 2024CBraMod: A Criss-Cross Brain Foundation Model for EEG Decodingcited by 188
- 2022Unsupervised Time-Series Representation Learning with Iterative Bilinear Temporal-Spectral Fusioncited by 96
- 2021Transformer-based Spatial-Temporal Feature Learning for EEG Decodingcited by 92
- 2023BrainBERT: Self-supervised representation learning for intracranial recordingscited by 70
- 2022A Comprehensive Survey on Model Quantization for Deep Neural Networks in Image Classificationcited by 64
- 2020EEG-TCNet: An Accurate Temporal Convolutional Network for Embedded Motor-Imagery Brain-Machine Interfacescited by 62
17 more
- 2022Transformer Convolutional Neural Networks for Automated Artifact Detection in Scalp EEGcited by 54
- 2024EEGFormer: Towards Transferable and Interpretable Large-Scale EEG Foundation Modelcited by 51
- 2024EEG2Rep: Enhancing Self-supervised EEG Representation Through Informative Masked Inputscited by 48
- 2019PULP-NN: Accelerating Quantized Neural Networks on Parallel Ultra-Low-Power RISC-V Processorscited by 34
- 2022Data augmentation for learning predictive models on EEG: a systematic comparisoncited by 32
- 2021Self-supervised EEG Representation Learning for Automatic Sleep Stagingcited by 30
- 2021EEG-GNN: Graph Neural Networks for Classification of Electroencephalogram (EEG) Signalscited by 28
- 2025MambaQuant: Quantizing the Mamba Family with Variance Aligned Rotation Methodscited by 26
- 2025LUNA: Efficient and Topology-Agnostic Foundation Model for EEG Signal Analysiscited by 24
- 2021CADDA: Class-wise Automatic Differentiable Data Augmentation for EEG Signalscited by 22
- 2021ECG-TCN: Wearable Cardiac Arrhythmia Detection with a Temporal Convolutional Networkcited by 14
- 2025CEReBrO: Compact Encoder for Representations of Brain Oscillations Using Efficient Alternating Attentioncited by 13
- 2018Addressing Class Imbalance in Classification Problems of Noisy Signals by using Fourier Transform Surrogatescited by 10
- 2024xTern: Energy-Efficient Ternary Neural Network Inference on RISC-V-Based Edge Systemscited by 4
- 2024GAPses: Versatile smart glasses for comfortable and fully-dry acquisition and parallel ultra-low-power processing of EEG and EOGcited by 3
- 2025Myna: Masking-Based Contrastive Learning of Musical Representationscited by 3
- 2025Lightweight Software Kernels and Hardware Extensions for Efficient Sparse Deep Neural Networks on Microcontrollerscited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026LuMamba: Latent Unified Mamba for Electrode Topology-Invariant and Efficient EEG Modeling
- 2024Mamba-based Deep Learning Approach for Sleep Staging on a Wireless Multimodal Wearable System without Electroencephalography
- 2025Mentality: A Mamba-based Approach towards Foundation Models for EEG
- 2026CaMBRAIN: Real-time, Continuous EEG Inference with Causal State Space Models
- 2025SAMBA: Toward a Long-Context EEG Foundation Model via Spatial Embedding and Differential Mamba
- 2024BiT-MamSleep: Bidirectional Temporal Mamba for EEG Sleep Staging
- 2025BioMamba: Leveraging Spectro-Temporal Embedding in Bidirectional Mamba for Enhanced Biosignal Classification
- 2026MedMamba: Recasting Mamba for Medical Time Series Classification