Temporal Context and Architecture: A Benchmark for Naturalistic EEG Decoding
Mehmet Ergezer
TL;DR
This work benchmarks five architectures (CNN, LSTM, EEGXF, S4, S5) for naturalistic EEG decoding on the HBN movie-watching dataset across long temporal contexts (8–128 s). It reveals a practical trade-off: S5 delivers high accuracy with far fewer parameters and faster training, whereas EEGXF offers more robust uncertainty handling under frequency shifts and OOD inputs. Across generalization tests, S5 demonstrates stronger cross-subject performance but can be overconfident on unseen tasks, while EEGXF remains conservative, better calibrated for OOD scenarios. The findings guide practical choices between peak performance and robust uncertainty in real-world EEG decoding applications and motivate validation on additional benchmarks.
Abstract
We study how model architecture and temporal context interact in naturalistic EEG decoding. Using the HBN movie-watching dataset, we benchmark five architectures, CNN, LSTM, a stabilized Transformer (EEGXF), S4, and S5, on a 4-class task across segment lengths from 8s to 128s. Accuracy improves with longer context: at 64s, S5 reaches 98.7%+/-0.6 and CNN 98.3%+/-0.3, while S5 uses ~20x fewer parameters than CNN. To probe real-world robustness, we evaluate zero-shot cross-frequency shifts, cross-task OOD inputs, and leave-one-subject-out generalization. S5 achieves stronger cross-subject accuracy but makes over-confident errors on OOD tasks; EEGXF is more conservative and stable under frequency shifts, though less calibrated in-distribution. These results reveal a practical efficiency-robustness trade-off: S5 for parameter-efficient peak accuracy; EEGXF when robustness and conservative uncertainty are critical.
