Balanced Latent Semantics and Signal Fidelity for EEG representation learning
Van-Chien Nguyen ⋅ Trung-Hieu Tran ⋅ Tuan-Kiet Doan ⋅ Quang Hung Pham ⋅ Ngoc-Son Vu ⋅ Duc Han LE ⋅ Huy Phan ⋅ Phi Le Nguyen ⋅ Nikola Simidjievski ⋅ Samuel Tardieu ⋅ Van-Tam Nguyen
Abstract
Electroencephalography (EEG) is critical for neurological diagnosis but suffers from low SNR and subject variability. Current foundation models relying on raw signal reconstruction often overfit to local noise. We propose STELAR, a foundation model with a dual-space objective combining patch-level masked latent prediction for semantic stability with masked reconstruction for raw signal fidelity. To balance these objectives, we introduce MTPE-GB, a validation-driven gradient balancer that adaptively weights tasks without manual tuning or computational overhead. STELAR achieves state-of-the-art linear probing performance across diverse EEG benchmarks, demonstrating robust generalization. All source code will be released.
Chat is not available.
Successful Page Load