Laya: A LeJEPA Approach to EEG via Latent Prediction over Reconstruction
Organizations: Department of Medical Informatics UCLA Los Angeles, USA · Department of Cognitive Science UCSD San Diego, USA · David Geffen School of Medicine UCLA Mattel Children’s Hospital Los Angeles, USA
Abstract
Electroencephalography (EEG) is a widely used tool for studying brain function, with applications in clinical neuroscience, diagnosis, and brain-computer interfaces (BCIs). Recent EEG foundation models trained on large unlabeled corpora aim to learn transferable representations, but their effectiveness remains unclear; reported improvements over smaller task-specific models are often modest, sensitive to downstream adaptation and fine-tuning strategies, and limited under linear probing. We hypothesize that one contributing factor is the reliance on signal reconstruction as the primary self-supervised learning (SSL) objective, which biases representations toward high-variance artifacts rather than task-relevant neural structure. To address this limitation, we explore an SSL paradigm based on Joint Embedding Predictive Architectures (JEPA), which learn by predicting latent representations instead of reconstructing raw signals. We introduce Laya, the first EEG foundation model based on LeJEPA. We show that latent prediction yields representations that encode semantic structure in EEG: Laya embeddings track clinically meaningful state changes such as seizure onset, are resilient to noise, and achieve the strongest mean clinical accuracy under frozen linear probing, with particular gains on tasks where relevant neural patterns are subtle and easily obscured by artifacts. Controlled ablations against matched MAE variants confirm that the choice of pretraining objective, rather than architecture or data, is the primary driver of these gains.
Figures & tables
| Task | LaBraM | LUNA | CBraMod | REVE | Laya-S | Laya-B |
|---|---|---|---|---|---|---|
| 5-Finger MI | 0.208 0.005 | 0.202 0.005 | 0.204 0.005 | 0.234 0.004 | 0.210 0.007 | 0.205 0.005 |
| LH vs RH MI | 0.471 0.019 | 0.493 0.007 | 0.491 0.015 | 0.513 0.017 | 0.510 0.003 | 0.507 0.004 |
| 4-Class MI | 0.289 0.005 | 0.259 0.009 | 0.271 0.017 | 0.280 0.007 | 0.273 0.005 | 0.288 0.007 |
| RH vs Feet MI | 0.524 0.017 | 0.501 0.002 | 0.515 0.011 | 0.508 0.009 | 0.586 0.002 | 0.578 0.002 |
| Mean | 0.373 | 0.364 | 0.370 | 0.384 | 0.395 | 0.394 |
| Task | LaBraM | LUNA | CBraMod | REVE | Laya-S | Laya-B |
|---|---|---|---|---|---|---|
| Abnormal | 0.781 0.013 | 0.733 0.018 | 0.605 0.011 | 0.500 0.000 | 0.798 0.007 | 0.778 0.006 |
| Artifact (Binary) | 0.725 0.003 | 0.707 0.023 | 0.690 0.001 | 0.720 0.003 | 0.716 0.001 | 0.737 0.001 |
| Epilepsy | 0.696 0.019 | 0.647 0.128 | 0.557 0.017 | 0.667 0.018 | 0.657 0.023 | 0.596 0.012 |
| mTBI | 0.549 0.133 | 0.613 0.052 | 0.499 0.002 | 0.682 0.053 | 0.699 0.030 | 0.815 0.048 |
| Artifact (Multiclass) | 0.374 0.006 | 0.288 0.019 | 0.298 0.004 | 0.431 0.058 | 0.375 0.002 | 0.382 0.008 |
| OCD | 0.697 0.175 | 0.546 0.063 | 0.600 0.173 | 0.633 0.000 | 0.627 0.144 | 0.567 0.037 |
| Variant | Objective | BCI Mean | Clinical Mean |
|---|---|---|---|
| Laya-MAE-Decoder | Raw signal reconstruction | 0.390 | 0.529 |
| Laya-MAE | Channel-mixed token prediction | 0.389 | 0.567 |
| Laya-S | Latent prediction | 0.395 | 0.603 |
| Laya-B | Latent prediction | 0.394 | 0.597 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Laya-S | Laya-B | |
| Data | ||
| Training data fraction | 10% | 100% |
| Max input duration | 16 s (160 timesteps) | 16 s (160 timesteps) |
| Batch size | 256 | 256 |
| Model | ||
| Encoder dim | 384 | 384 |
| Configuration | StopGrad | BCI Mean | Clinical Mean |
|---|---|---|---|
| Laya-S | ✓ | 0.395 | 0.603 |
| Laya-S | 0.391 | 0.559 | |
| Laya-B (20K) | ✓ | 0.394 | 0.597 |
| Laya-B (20K) | 0.353 | 0.466 |
EEG Foundation Models
Electroencephalography (EEG) foundation models learn reusable representations of brain signals for transfer across people, tasks, and recording setups.
See all 100 papers