MANAS-2: Constrained Reconstruction for EEG Foundation Models
Arvasu Kulkarni ⋅ Aditya Ray Mishra ⋅ Mahir Jain ⋅ Jeet Bandhu Lahiri ⋅ Parshva Runwal ⋅ Lakshya Saini ⋅ Sandeep Singh ⋅ Siddharth Panwar
Abstract
Masked reconstruction is widely used for EEG foundation models, but optimizing fidelity to noisy observed waveforms does not necessarily produce the most useful latent representation. We introduce MANAS-2, a new EEG foundation model that combines a Raw-band Hybrid (RBH) masked autoencoder with Constrained Reconstruction (ConRec), a physically motivated output-space regularizer. RBH jointly reconstructs temporal waveform patches and compact spectral-band targets, while ConRec acts only on the temporal decoder output, penalizing differences in RMS energy across regularly spaced boundaries. ConRec is intended to shape the encoder by biasing it toward the organization of oscillatory-envelope information. Across seven held-out EEG datasets, adding ConRec to an otherwise identical RBH model increases frozen ridge recovery of six-band spectral power from mean $R^2=0.860$ to $0.906$ and recovery of inter-patch band-energy dynamics from $R^2=0.283$ to $0.354$, without sacrificing encoded temporal waveform information. Applied to a temporal-only masked autoencoder, ConRec also improves frozen downstream transfer and frequency-dependent latent geometry despite receiving no spectral targets. I.e., the effects of ConRec are architecture-independent. MANAS-2 also outperforms leading EEG Foundation Models on most downstream knowledge-transfer tasks. From the effects of ConRec, we see that a physically motivated constraint imposed through the decoder can make for a more spectrally organized and transferable latent space. MANAS-2 therefore provides a new EEG foundation model built around constrained reconstruction as a mechanism for shaping representation -- rather than reconstruction -- quality.
Chat is not available.
Successful Page Load