STST-JEPA: New Self-Supervised EEG Model Predicts Brain Age

Roy Segal, Yoni Svechinsky, Tomer Fekete· July 9, 2026 View original

Key takeaways

  • STST-JEPA is a self-supervised transformer for EEG, pretrained on a large, diverse dataset.
  • It achieves state-of-the-art brain age regression and other classification tasks.
  • The model addresses challenges like data heterogeneity and small labeled cohorts.
  • Its brain age residual correlates with cognitive efficiency, suggesting biomarker potential.

Who benefits

HealthcareMedical DevicesNeuroscience ResearchPharmaceuticalsWearable Tech

Summary

Researchers introduce STST-JEPA, a self-supervised transformer for EEG, pretrained on a large dataset to predict masked-token representations and reconstruct signals. This model achieves state-of-the-art performance in brain age regression, sex classification, and psychopathology composite regression across a wide age range.

A new self-supervised learning model, STST-JEPA (Shallow-Target Spatio-Temporal Joint Embedding Prediction Architecture), has been developed for Electroencephalography (EEG) data. This transformer-based model is pretrained on a massive dataset of 47,703 EEG sessions, covering a broad age range from 5 to 81 years, sourced from the brain.space and Healthy Brain Network corpora. The model's training involves a dual objective: predicting masked-token representations against an EMA-of-tokenizer target and an auxiliary signal-reconstruction term applied to 30-second multi-channel windows with spatiotemporal block masks. STST-JEPA demonstrates impressive performance, achieving a mean absolute error of 3.06 years for age regression on held-out validation data, significantly outperforming a baseline. With minimal fine-tuning, the pretrained encoder also secured top rankings on the NeuralBench x brain.space EEG leaderboard for sex classification, age prediction, and psychopathology composite regression. The model's ability to infer "brain age" and its correlation with cognitive efficiency positions it as a promising biomarker for neurological and psychiatric conditions, addressing challenges like cross-site heterogeneity and small labeled cohorts in EEG research.

Why it matters

This self-supervised EEG model offers a robust and data-efficient approach to developing biomarkers for neurological and psychiatric conditions, with broad applications in healthcare and research.

How to implement this in your domain

  1. 1Explore integrating STST-JEPA's pretrained encoder into your EEG analysis pipelines for tasks like brain age prediction or disease classification.
  2. 2Leverage the self-supervised pretraining approach to develop foundation models for other physiological signal data.
  3. 3Utilize the model's capabilities for identifying potential biomarkers for neurological and psychiatric disorders in clinical research.
  4. 4Adapt the model for real-world applications requiring robust EEG analysis, such as in wearable health devices.

Original post by Roy Segal, Yoni Svechinsky, Tomer Fekete

"arXiv:2607.06629v1 Announce Type: new Abstract: Brain age -- the age inferred from a physiological recording -- is an emerging biomarker whose deviation from chronological age tracks neurological and psychiatric burden, and EEG is an attractive substrate for it because it is chea…"

View on X

Originally posted by Roy Segal, Yoni Svechinsky, Tomer Fekete on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses