Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

THD-BAR: Topology Hierarchical Derived Brain Autoregressive Modeling for EEG Generic Representations

About

Large-scale pre-trained models hold significant potential for learning universal EEG representations. However, most existing methods, particularly autoregressive (AR) frameworks, primarily rely on straightforward temporal sequencing of multi-channel EEG data, which fails to capture the rich physiological characteristics inherent to EEG signals. Moreover, their time-centered modeling approach also limits the effective representation of the dynamic spatial topology of brain activity. To address these challenges and fully exploit the potential of large-scale EEG models, we propose a novel Topology Hierarchical Derived Brain Autoregressive Modeling (THD-BAR) for EEG generic representations. The core innovation of THD-BAR lies in the introduction of the Brain Topology Hierarchy (BTH), which establishes a multi-scale spatial order for EEG channels. This hierarchical structure enables a redefinition of autoregressive learning as a "next-scale-time prediction" problem, effectively capturing both spatial and temporal dynamics. Based on BTH, we design a Topology-Hierarchical Vector Quantized-Variational Autoencoder (THVQ-VAE) for multi-scale tokenization and develop an enhanced Brain Autoregressive (BAR) module with specialized masking strategies for prediction. Through extensive large-scale pre-training on 17 datasets, followed by rigorous validation on 10 downstream datasets spanning 5 distinct tasks, THD-BAR consistently outperforms existing methods. These results highlight the superior generalization and modeling capabilities of our proposed approach.

Wenchao Yang, Weidong Yan, Wenkang Liu, Yulan Ma, Yang Li• 2025

Related benchmarks

TaskDatasetResultRank
Binary classification of normal versus abnormal EEG signalsTUAB
Balanced Accuracy82.2
113
Event Type ClassificationTUEV
Balanced Accuracy65.3
50
EEG ClassificationWorkload
Balanced Accuracy67.1
31
sleep stages classificationHMC
Balanced Accuracy0.684
30
Abnormality DetectionTUAB
Balanced Accuracy82.2
27
EEG ClassificationSEED
B-Acc73.9
12
Emotion RecognitionSEED
Balanced Accuracy (B-Acc)73.9
12
EEG ClassificationTUEV
Balanced Accuracy65.3
12
EEG ClassificationTUSL
Balanced Accuracy69.2
12
EEG InterpretationTUSL
Balanced Accuracy (B-Acc)69.2
12
Showing 10 of 11 rows

Other info

Follow for update