STAMP: Spatial-Temporal Adapter with Multi-Head Pooling
About
Time series foundation models (TSFMs) pretrained on data from multiple domains have shown strong performance on diverse modeling tasks. Various efforts have been made to develop foundation models specific to electroencephalography (EEG) data, which records brain electrical activity as time series. However, no comparative analysis of EEG-specific foundation models (EEGFMs) versus general TSFMs has been performed on EEG-specific tasks. We introduce a novel Spatial-Temporal Adapter with Multi-Head Pooling (STAMP), which leverages univariate embeddings produced by a general TSFM, implicitly models spatial-temporal characteristics of EEG data, and achieves performance comparable to state-of-the-art EEGFMs. A comprehensive analysis is performed on 8 benchmark datasets of clinical tasks using EEG for classification, along with ablation studies. Our proposed adapter is lightweight in trainable parameters and flexible in the inputs it can accommodate, supporting easy modeling of EEG data using TSFMs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| EEG Classification | TUEV (test) | Balanced Accuracy56.4 | 24 | |
| EEG Classification | MentalArithmetic | Balanced Accuracy64.38 | 18 | |
| Emotion Recognition | FACED 9-Class | Balanced Accuracy36.06 | 17 | |
| EEG Classification | BCIC-IV-2a | Balanced Accuracy55.64 | 5 | |
| EEG Classification | PhysioNet-MI 4-Class 9,837 Samples | Balanced Accuracy60.98 | 5 | |
| EEG Classification | Mumtaz 2-Class 2016 (test) | Balanced Accuracy91.72 | 5 | |
| Emotion Recognition | SEED-V 5-Class | Balanced Accuracy36.7 | 5 | |
| EEG Classification | SHU-MI 2-Class 11,988 Samples | Balanced Accuracy59.83 | 5 |