bioRxivBioinformaticsPreprint
Pre-trained Vision Transformers for Seizure Prediction: A Reproducible Baseline with Event-Based Evaluation and Statistical Validation
Abstract
BackgroundScalp electroencephalography (EEG) based seizure prediction plays a critical role in improving the quality of life for patients with drug-resistant epilepsy, offering the potential for real-time warnings and timely interventions. Despite its clinical significance and decades of research, the field still lacks an open benchmark with reproducible baselines and deployment-oriented event-level evaluation. Most prior work relies on the small and outdated Childrens Hospital Boston (CHB-MIT) dataset and reports window-level metrics only, leaving the false-alarm burden of a real warning system underspecified. In seizure prediction, the cost of false alarm is significantly high since patients may receive painful electrical stimulation to suppress seizure. Hence, false alarms per hour (FA/h) and partial AUC (pAUC) are the most deployment-relevant metrics, reflecting alarm burden and discriminability in the low-false-alarm operating region that a usable warning system can realistically tolerate. However, few studies have systematically reported such metrics. In addition, vision transformers event-level performance under deployable FA/h constraints remains underexplored, and newer backbones such as MambaVision have yet to be evaluated under this setting. MethodsIn this work, we introduce a reproducible 5-fold benchmark derived from the Temple University Hospital EEG Seizure Corpus (TUSZ) dataset, and evaluate models using a pseudo-real-time event pipeline, reporting event-level sensitivity, false alarms per hour (FA/h) and partial AUC (pAUC). All models are compared to random predictors for statistical validation. We benchmark pre-trained vision transformers (SegFormer and MambaVision) under three EEG-to-image encoding methods, including a self-proposed Temporal-Patchify encoding for SegFormer. ResultsOur proposed Temporal-Patchify encoding method achieves state-of-the-art performance. We achieved 0.61 pAUC, which is 16.2% higher than the baseline Temporal-Tile SegFormer of Parani et al. The false-alarm burden (0.40{+/-}0.28 FA/h) is 44.4% lower than the Temporal-Tile SegFormer baseline while maintaining clinically usable sensitivity (60.7%{+/-}5.0%). We further perform statistical validation against a matched Poisson random predictor, confirming performance exceeds chance. Finally, we report end-to-end inference through-put up to 920 windows/s, confirming MambaVisions fastest inference speed, exceeding SegFormer by over 20%. ConclusionsThis work bridges the gap between seizure prediction algorithms and clinically usable seizure prediction systems in real-world settings. Our findings indicate that pre-trained vision transformers, when coupled with appropriate EEG encoding methods, can achieve robust performance in low-false-alarm operating regimes, which is critical for real-world deployment. This benchmark and evaluation framework may facilitate more clinically meaningful and reproducible seizure prediction research.
- Citations
- 0
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- CViT-ESP: Lightweight Pre-trained Vision Transformers for EEG-based Epileptic Seizure Prediction
Umair Mohammad, Paras Parani, Fahad Saeed·bioRxiv·Aug 26, 2026
- Utilizing Pretrained Vision Transformers and LargeLanguage Models for Epileptic Seizure Prediction
Paras Parani, Umair Mohammad, Fahad Saeed·International Conference on Data Science and Machine Learning Applications (CDMA)Peer-reviewed·Feb 16, 2025
- MLSPred-Bench: ML-Ready Benchmark LeveragingSeizure Detection EEG data for Predictive Models
Umair Mohammad, Fahad Saeed·bioRxiv·Jul 17, 2024
- TSD: Transformers for Seizure Detection
Y. Ma, C. Liu, … O. Kavehei·bioRxiv·Jan 24, 2023
- Multiscale predictive modeling robustly improves the accuracy of pseudo-prospective seizure forecasting in drug-resistant epilepsy
Gagan Acharya, Erin C. Conrad, … Erfan Nozari·Journal of Neural EngineeringPeer-reviewed·Jun 2, 2026
- ScoreNet: A Neural network-based post-processing model for identifying epileptic seizure onset and offset in EEGs
Poomipat Boonyakitanont, Apiwat Lek-uthai, Jitkomut Songsiri·IEEE Transactions on Neural Systems and Rehabilitation EngineeringPeer-reviewed·Jan 1, 2021