Beyond Flattened Tokens: Structure-Preserving EEG Decoding with Reusable TriDim Blocks
Summary
The paper proposes TriDim, a reusable block for EEG decoding that preserves spatial and temporal structures, achieving state-of-the-art performance across multiple datasets and improving existing models.
View Cached Full Text
Cached at: 09/18/26, 09:07 AM
# Structure-Preserving EEG Decodingwith Reusable TriDim Blocks
Source: [https://arxiv.org/html/2609.19842](https://arxiv.org/html/2609.19842)
## Beyond Flattened Tokens: Structure\-Preserving EEG Decoding with Reusable TriDim Blocks
Shiyue Su1,2, Song Wang2, Zekai Zhan1, Junjie Zeng1, Ziling Lu1,Zongsheng Li1,4, Xinyuan Ye1, Zhiyuan Ma5, Xinke Shen1\*, and Quanying Liu1,2,3\*1Department of Biomedical Engineering, Southern University of Science and Technology,Shenzhen, 518055, China\.2Omni\-Intelligence, Shenzhen, China\.3Shenzhen Loop Area Institute, Shenzhen, China\.4Department of Computer Science, The Chinese University of Hong Kong,Shenzhen, 518172, China\.5School of Biomedical Engineering, Tsinghua Medicine, Tsinghua University,Beijing, 100084, China\.shenxk@sustech\.edu\.cn; liuqy@sustech\.edu\.cn \*Co\-corresponding authors
###### Abstract
Effective EEG decoding requires representations that preserve organization among channels, local waveform dynamics, and long\-range temporal context\. Existing EEG architectures often capture these structures using separate specialized modules or collapse them into a single token sequence, making it difficult to maintain their distinct roles and coordinate their interactions throughout the backbone\. We propose TriDim, a reusable block that preserves the representation shape and keeps three EEG axes explicit: channel, sample position within each patch, and patch position across the recording\. These axes correspond to spatial, short\-term temporal, and long\-term temporal information, respectively\. Each TriDim block applies feed\-forward transformations along individual axes and cross\-axis attention to coordinate information exchange among them\. By stacking TriDim blocks with a multi\-level tri\-axis readout, we construct TriDimEEG, a standalone EEG decoder\. Under strict cross\-subject evaluation on eight datasets spanning clinical diagnosis, sleep staging, motor imagery, and emotion recognition, TriDimEEG achieves the best overall performance among fifteen evaluated models, with a 4\.3% relative improvement in average accuracy over the second\-best model\. Replacing Transformer blocks in three EEG foundation models with TriDim blocks yields an average relative improvement of 7\.4% in downstream accuracy while reducing parameter counts by 17\.0% to 47\.3%\. These results establish TriDim as an effective and reusable building block and TriDimEEG as a strong standalone EEG decoder\.
Code and parameters of TriDimEEG are available at[https://github\.com/ncclab\-sustech/TriDim\_model](https://github.com/ncclab-sustech/TriDim_model)\.
## Introduction
Electroencephalography \(EEG\) decoding supports diverse applications, such as clinical diagnosis, sleep monitoring, robotic control, and affective computing\([Craik et al\., 2019](https://arxiv.org/html/2609.19842#bib.bib20);[Roy et al\., 2019](https://arxiv.org/html/2609.19842#bib.bib21)\)\. EEG signals can be characterized by three complementary structural aspects \(Figure[1](https://arxiv.org/html/2609.19842#Sx1.F1)\): spatial organization across channels, waveform dynamics within local windows, and long\-range temporal context across windows\([Lawhern et al\., 2018](https://arxiv.org/html/2609.19842#bib.bib5);[Ding et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib6);[Song et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib8);[Wang et al\., 2025a](https://arxiv.org/html/2609.19842#bib.bib15)\)\. Spatial organization captures how neural activity is distributed and coordinated across recording channels\. Local waveform dynamics characterize oscillatory and transient patterns within short intervals\. Long\-range temporal context describes how these patterns evolve across extended periods\. The three information types are also strongly interdependent\([Wang et al\., 2025a](https://arxiv.org/html/2609.19842#bib.bib15);[Wang et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib16)\)\. For example, the same local waveform may indicate different neural states when expressed over different channel distributions, and the meaning of a spatial pattern may change according to the preceding and subsequent temporal context\. Effective EEG decoding therefore requires a backbone that progressively refines each information type and repeatedly coordinates interactions among the three information types\. Existing architectures, however, often prioritize only part of this structure and do not provide sufficiently deep and coordinated modeling of all three information types\.
Figure 1:Comparison of conventional flattened\-token EEG architecture and TriDim\. TriDim preserves channel, within\-patch temporal, and across\-patch temporal structures as three explicit axes for tri\-axis modeling\.The main limitation lies in how current EEG architectures organize computation across the spatial and temporal dimensions of EEG\. Compact convolutional networks use temporal kernels and spatial filters to capture local dynamics and channel interactions\([Lawhern et al\., 2018](https://arxiv.org/html/2609.19842#bib.bib5);[Ding et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib6);[Miao et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib7)\), but predefined receptive fields constrain their ability to model long\-range dependencies\. Transformer\-based models address this limitation in different ways\. CNN–Transformer hybrids combine convolutional feature extraction with self\-attention\([Song et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib8);[Wan et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib9);[Ding et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib10)\)\. Many pretrained EEG encoders divide multichannel recordings into channel\-wise temporal patches and arrange the resulting channel–time patches as a single token sequence for Transformer processing\([Yang et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib12);[Jiang et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib13);[Wang et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib14)\)\. As illustrated in Figure[1](https://arxiv.org/html/2609.19842#Sx1.F1), this conventional flattened\-token pipeline mixes spatial and temporal structures within a single sequence\-processing pathway\. Unlike language, EEG is a continuous multichannel signal with temporal structure at multiple scales\. A single token sequence assigns spatial organization and temporal dependencies to the same sequence\-processing pathway\. This design limits the depth and specificity with which the spatial, local temporal, and global temporal information and their interactions can be modeled\. Criss–cross Transformers partly preserve EEG structure by applying self\-attention along separate spatial and temporal directions\([Wang et al\., 2025a](https://arxiv.org/html/2609.19842#bib.bib15);[Wang et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib16)\)\. However, these models still assign fixed computational roles to the three EEG axes\. Channel and across\-patch time are treated as attention axes, whereas ordered within\-patch samples remain in the embedding dimension\. Consequently, within\-patch representations are repeatedly refined by feed\-forward networks without explicit relation modeling through self\-attention\. Channel and across\-patch representations are repeatedly mixed through attention without axis\-wise feed\-forward transformation\. Because self\-attention and feed\-forward networks provide complementary relational modeling and nonlinear representation refinement\([Dong et al\., 2021](https://arxiv.org/html/2609.19842#bib.bib2);[Geva et al\., 2020](https://arxiv.org/html/2609.19842#bib.bib3);[Xu et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib4)\), this fixed assignment limits the deep processing of the three information types and the progressive refinement of their interactions throughout the backbone\.
To enable deep processing of all three EEG information types and their interactions, we adopt a different design principle: channel, within\-patch time, and across\-patch time should remain explicit representation axes throughout the backbone\. Based on this principle, we propose TriDim, a shape\-preserving block for EEG modeling \(Figure[2](https://arxiv.org/html/2609.19842#Sx3.F2)\)\. TriDim constructs three axis\-conditioned views\. In each view, one axis serves as the embedding dimension, and self\-attention is applied along the other two axes\. The resulting views are combined through learnable weighted aggregation\. Stacking TriDim blocks therefore enables spatial, short\-term temporal, and long\-term temporal information to undergo repeated feed\-forward transformation, and interactions among the three information types are progressively updated through self\-attention\. A multi\-level tri\-axis readout further aggregates representations from different encoder depths\. The resulting TriDimEEG model can serve as a task\-specific EEG decoder, and the same TriDim block can also replace Transformer blocks in pretrained EEG encoders without redesigning the surrounding pipelines\. This work makes three main contributions:
- •We reinterpret temporally patched EEG representations using three explicit axes corresponding to channel, sample position within each patch, and patch position along the recording\. This organization preserves spatial, short\-term temporal, and long\-term temporal information throughout the backbone\.
- •We develop TriDim, a reusable block that combines feed\-forward transformations along individual axes with cross\-axis attention\. We further construct TriDimEEG by stacking TriDim blocks and introducing a multi\-level tri\-axis readout that aggregates representations across encoder depths\.
- •Strict cross\-subject experiments on eight EEG benchmarks establish TriDimEEG as a strong standalone decoder\. Replacing Transformer blocks in three EEG foundation models further demonstrates the reusability and parameter efficiency of TriDim\.
## Related Work
### Task\-Specific EEG Decoders
Early EEG decoders primarily relied on convolutional networks to capture local waveform patterns and inter\-channel relationships\. EEGNet combines temporal convolution with depthwise spatial filtering in a compact architecture\([Lawhern et al\., 2018](https://arxiv.org/html/2609.19842#bib.bib5)\)\. Subsequent supervised models introduce multi\-scale temporal kernels and specialized spatial processing for different EEG tasks\([Ding et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib6);[Miao et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib7);[Ma et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib11)\)\. Attention\-based architectures further extend this paradigm by modeling dependencies beyond local convolutional receptive fields\. EEGConformer applies self\-attention to features extracted by a convolutional temporal–spatial encoder\([Song et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib8)\)\. EEGDeformer introduces hierarchical coarse\-to\-fine Transformer processing and dense information purification to aggregate representations across multiple levels\([Ding et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib10)\)\. TeCh replaces conventional pairwise self\-attention with Core Token Aggregation\-Redistribution, which collects global context into a core token and redistributes it to the remaining tokens\([Yu et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib33)\)\. Despite their architectural differences, these methods generally convert EEG signals into feature sequences before applying attention\-based contextual modeling\.
### Pretrained EEG Representation Models
Large\-scale pretraining has further established sequence modeling as a prominent paradigm for transferable EEG representation learning\. BENDR\([Kostas et al\., 2021](https://arxiv.org/html/2609.19842#bib.bib34)\), BIOT\([Yang et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib12)\), LaBraM\([Jiang et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib13)\), and REVE\([El Ouahidi et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib39)\)learn contextual representations from segmented EEG recordings through convolutional or Transformer\-based encoders\. Recent pretrained models additionally explore language\-aligned representations\([Cui et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib35)\), state\-space sequence modeling\([Wang et al\., 2025b](https://arxiv.org/html/2609.19842#bib.bib36)\), frequency\-enhanced encoders\([Tegon et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib37)\), and sensor\-aware tokenization\([Xiao et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib17)\)\. These approaches have substantially improved transfer across datasets and downstream tasks\. Nevertheless, despite differences in backbone architectures and pretraining objectives, most of these models still arrange local EEG segments into a one\-dimensional token sequence, leaving spatial organization and across\-segment temporal dependencies to be modeled within the same sequence\-processing pathway\.
### Structure\-Preserving EEG Architectures
Recent studies have begun to design attention mechanisms more explicitly aligned with the structural organization of EEG\. CBraMod introduces a criss\-cross Transformer that separately performs spatial attention across electrodes and temporal attention across windows\([Wang et al\., 2025a](https://arxiv.org/html/2609.19842#bib.bib15)\)\. BraSTORM adopts separate spatial and temporal branches to learn complementary representations through input\-based spatiotemporal decomposition\([Wang et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib16)\)\. CSBrain further introduces cross\-scale spatiotemporal tokenization and structured sparse attention to model dependencies across temporal windows and anatomical regions\([Zhou et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib18)\)\. These methods demonstrate the value of preserving structural distinctions in EEG rather than treating all signal components as positions in a homogeneous sequence\. However, existing structure\-aware architectures still assign fixed and asymmetric computational roles to the EEG axes, typically treating channel and across\-patch time as attention axes and retaining within\-patch samples as the embedding dimension\. TriDim removes this fixed assignment by preserving channel, within\-patch time, and across\-patch time as three explicit axes, allowing each axis to undergo both axis\-specific feed\-forward transformation and cross\-axis attention throughout the backbone\.
## Methods
Figure 2:Overview of TriDimEEG, including tri\-axis input projections, stacked TriDim blocks, and a multi\-level tri\-axis readout\.TriDimEEG consists of a tri\-axis projection, an encoder constructed fromNNstacked TriDim blocks, and a multi\-level tri\-axis readout\.
### Tri\-axis Representation
We organize EEG representations along three axes: channel organization, short\-term temporal structure, and long\-term temporal context\. Given an input batchXraw∈ℝB×Cin×TrawX\_\{\\mathrm\{raw\}\}\\in\\mathbb\{R\}^\{B\\times C\_\{\\mathrm\{in\}\}\\times T\_\{\\mathrm\{raw\}\}\}withBBEEG samples,CinC\_\{\\mathrm\{in\}\}electrodes, andTrawT\_\{\\mathrm\{raw\}\}time points, we first applyInstanceTimeNorm\\mathrm\{InstanceTimeNorm\}independently to each channel over time\. This operation removes the temporal mean and normalizes the amplitude scale of each channel without changing the tensor shape:
X^∈ℝB×Cin×Traw\.\\hat\{X\}\\in\\mathbb\{R\}^\{B\\times C\_\{\\mathrm\{in\}\}\\times T\_\{\\mathrm\{raw\}\}\}\.
We then unfold the time axis intoLinL\_\{\\mathrm\{in\}\}patches of lengthSinS\_\{\\mathrm\{in\}\}with strideRR, applying zero\-padding at the end when necessary\. This operation separates the original temporal dimension into the ordered samples within each patch and the ordered patch positions along the segment:
X^∈ℝB×Cin×Traw⟶Xpatch∈ℝB×Cin×Sin×Lin\.\\hat\{X\}\\in\\mathbb\{R\}^\{B\\times C\_\{\\mathrm\{in\}\}\\times T\_\{\\mathrm\{raw\}\}\}\\;\\longrightarrow\\;X^\{\\mathrm\{patch\}\}\\in\\mathbb\{R\}^\{B\\times C\_\{\\mathrm\{in\}\}\\times S\_\{\\mathrm\{in\}\}\\times L\_\{\\mathrm\{in\}\}\}\.
Three basis projections, one for each representation axis, map\(Cin,Sin,Lin\)\(C\_\{\\mathrm\{in\}\},S\_\{\\mathrm\{in\}\},L\_\{\\mathrm\{in\}\}\)to the latent dimensions\(C,S,L\)\(C,S,L\):
ℝCin\\displaystyle\\mathbb\{R\}^\{C\_\{\\mathrm\{in\}\}\}→ℝC\\displaystyle\\to\\mathbb\{R\}^\{C\}\(1×1convolution\),\\displaystyle\\text\{\(\}1\{\\times\}1\\text\{ convolution\)\},ℝSin\\displaystyle\\mathbb\{R\}^\{S\_\{\\mathrm\{in\}\}\}→ℝS\\displaystyle\\to\\mathbb\{R\}^\{S\}\(linear projection\),\\displaystyle\\text\{\(linear projection\)\},ℝLin\\displaystyle\\mathbb\{R\}^\{L\_\{\\mathrm\{in\}\}\}→ℝL\\displaystyle\\to\\mathbb\{R\}^\{L\}\(shared convolution and pooling\)\.\\displaystyle\\text\{\(shared convolution and pooling\)\}\.
The channel projection applies a1×11\{\\times\}1convolution along the electrode axisCinC\_\{\\mathrm\{in\}\}\. The short\-term projection applies a linear transformation alongSinS\_\{\\mathrm\{in\}\}\. The long\-term projection applies a shared one\-dimensional convolution to the patch sequence at every channel and within\-patch position, followed by GELU, dropout, and adaptive average pooling alongLinL\_\{\\mathrm\{in\}\}\. The three projections produce
Xpatch∈ℝB×Cin×Sin×Lin⟶X0∈ℝB×C×S×L\.X^\{\\mathrm\{patch\}\}\\in\\mathbb\{R\}^\{B\\times C\_\{\\mathrm\{in\}\}\\times S\_\{\\mathrm\{in\}\}\\times L\_\{\\mathrm\{in\}\}\}\\;\\longrightarrow\\;X^\{0\}\\in\\mathbb\{R\}^\{B\\times C\\times S\\times L\}\.
We use𝒜=\{c,s,l\}\\mathcal\{A\}=\\\{c,s,l\\\}to denote the latent channel, short\-term, and long\-term axes, whose sizes areCC,SS, andLL, respectively\. The TriDimEEG encoder operates on the resulting\[B,C,S,L\]\[B,C,S,L\]representation, which is preserved throughout all TriDim blocks\.
### TriDim Block
LetXℓ∈ℝB×C×S×LX^\{\\ell\}\\in\\mathbb\{R\}^\{B\\times C\\times S\\times L\}denote the input to blockℓ\+1\\ell\+1, whereℓ=0,…,N−1\\ell=0,\\ldots,N\-1andX0X^\{0\}is defined above\. Each TriDim block contains two pre\-normalized residual sublayers: a cross\-axis attention sublayer followed by feed\-forward transformations along individual axes\. Both sublayers preserve the\[B,C,S,L\]\[B,C,S,L\]representation shape\.
Cross\-axis attention\.For each axisa∈𝒜a\\in\\mathcal\{A\}, we first apply Root Mean Square normalization\([Zhang and Sennrich, 2019](https://arxiv.org/html/2609.19842#bib.bib23)\)along that axis:
X~aℓ=AxisRMSNormaattn\(Xℓ\),a∈𝒜\.\\tilde\{X\}^\{\\ell\}\_\{a\}=\\mathrm\{AxisRMSNorm\}^\{\\mathrm\{attn\}\}\_\{a\}\\left\(X^\{\\ell\}\\right\),\\qquad a\\in\\mathcal\{A\}\.
For a tensorZZ, letAttnp\|q\(Z\)\\mathrm\{Attn\}\_\{p\|q\}\(Z\), wherep,q∈𝒜p,q\\in\\mathcal\{A\}andp≠qp\\neq q, denote multi\-head self\-attention along sequence axispp, with axisqqserving as the embedding dimension\. The remaining axis is treated as a collection of independent slices and folded into the batch dimension during attention computation\.
For each embedding axisqq, the two attention operations along the remaining axes share the same query, key, value, and output projection parameters\. They are applied independently to the same normalized representation, and their outputs are averaged\. The channel, short\-term, and long\-term views are computed as
Vcℓ\\displaystyle V\_\{c\}^\{\\ell\}=12\[Attns\|c\(X~cℓ\)\+Attnl\|c\(X~cℓ\)\],\\displaystyle=\\frac\{1\}\{2\}\\left\[\\mathrm\{Attn\}\_\{s\|c\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{c\}\\right\)\+\\mathrm\{Attn\}\_\{l\|c\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{c\}\\right\)\\right\],Vsℓ\\displaystyle V\_\{s\}^\{\\ell\}=12\[Attnc\|s\(X~sℓ\)\+Attnl\|s\(X~sℓ\)\],\\displaystyle=\\frac\{1\}\{2\}\\left\[\\mathrm\{Attn\}\_\{c\|s\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{s\}\\right\)\+\\mathrm\{Attn\}\_\{l\|s\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{s\}\\right\)\\right\],Vlℓ\\displaystyle V\_\{l\}^\{\\ell\}=12\[Attnc\|l\(X~lℓ\)\+Attns\|l\(X~lℓ\)\]\.\\displaystyle=\\frac\{1\}\{2\}\\left\[\\mathrm\{Attn\}\_\{c\|l\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{l\}\\right\)\+\\mathrm\{Attn\}\_\{s\|l\}\\left\(\\tilde\{X\}^\{\\ell\}\_\{l\}\\right\)\\right\]\.
Each view retains the\[B,C,S,L\]\[B,C,S,L\]shape and is named according to its embedding axis rather than the sequence axes over which attention is performed\.
To robustly integrate the three complementary attention views, we regularize each view before fusion\. Specifically, we apply DropPath independently to each viewVaℓV\_\{a\}^\{\\ell\}with dropout probabilitypaℓp\_\{a\}^\{\\ell\}\([Huang et al\., 2016](https://arxiv.org/html/2609.19842#bib.bib19)\)\. This enables view\-wise stochastic depth and discourages the model from relying excessively on any single EEG axis\. The details are reported in Appendix[G](https://arxiv.org/html/2609.19842#A7)\.
The regularized views are then combined using learnable fusion weights\. Let
𝜶ℓ=softmax\(𝜽attnℓ\)∈ℝ3\\boldsymbol\{\\alpha\}^\{\\ell\}=\\mathrm\{softmax\}\\left\(\\boldsymbol\{\\theta\}\_\{\\mathrm\{attn\}\}^\{\\ell\}\\right\)\\in\\mathbb\{R\}^\{3\}denote the learnable view\-fusion weights, andγattnℓ∈ℝC\\gamma\_\{\\mathrm\{attn\}\}^\{\\ell\}\\in\\mathbb\{R\}^\{C\}per\-channel learnable LayerScale parameters\([Touvron et al\., 2021](https://arxiv.org/html/2609.19842#bib.bib22)\)\. The attention residual update is
Xℓ\+12=Xℓ\+γattnℓ⊙∑a∈𝒜αaℓDropPath\(Vaℓ,paℓ\)\.X^\{\\ell\+\\frac\{1\}\{2\}\}=X^\{\\ell\}\+\\gamma\_\{\\mathrm\{attn\}\}^\{\\ell\}\\odot\\sum\_\{a\\in\\mathcal\{A\}\}\\alpha\_\{a\}^\{\\ell\}\\,\\mathrm\{DropPath\}\\left\(V\_\{a\}^\{\\ell\},p\_\{a\}^\{\\ell\}\\right\)\.
Axis\-wise feed\-forward sublayer\.After cross\-axis attention with multi\-view fusion, an independent feed\-forward networkFFNa\\mathrm\{FFN\}\_\{a\}is applied along each axisaa\. Each network contains two linear layers with an expansion ratio of22, a GELU activation, and dropout\. The intermediate representation is independently normalized along each target axis:
X¯aℓ\+12\\displaystyle\\bar\{X\}^\{\\ell\+\\frac\{1\}\{2\}\}\_\{a\}=AxisRMSNormaffn\(Xℓ\+12\),\\displaystyle=\\mathrm\{AxisRMSNorm\}^\{\\mathrm\{ffn\}\}\_\{a\}\\left\(X^\{\\ell\+\\frac\{1\}\{2\}\}\\right\),Uaℓ\\displaystyle U\_\{a\}^\{\\ell\}=FFNa\(X¯aℓ\+12\)\.\\displaystyle=\\mathrm\{FFN\}\_\{a\}\\left\(\\bar\{X\}^\{\\ell\+\\frac\{1\}\{2\}\}\_\{a\}\\right\)\.
The three axis\-wise FFN outputs are first combined using learnable fusion weights:
𝜷ℓ=softmax\(𝜽ffnℓ\)∈ℝ3,\\boldsymbol\{\\beta\}^\{\\ell\}=\\mathrm\{softmax\}\\left\(\\boldsymbol\{\\theta\}\_\{\\mathrm\{ffn\}\}^\{\\ell\}\\right\)\\in\\mathbb\{R\}^\{3\},which denotes the relative contributions of the three axes\. DropPath is applied to the fused output\. With per\-channel learnable LayerScale parametersγffnℓ∈ℝC\\gamma\_\{\\mathrm\{ffn\}\}^\{\\ell\}\\in\\mathbb\{R\}^\{C\}and DropPath ratepffnℓp\_\{\\mathrm\{ffn\}\}^\{\\ell\}, the block output is
Xℓ\+1=Xℓ\+12\+γffnℓ⊙DropPath\(∑a∈𝒜βaℓUaℓ,pffnℓ\)\.X^\{\\ell\+1\}=X^\{\\ell\+\\frac\{1\}\{2\}\}\+\\gamma\_\{\\mathrm\{ffn\}\}^\{\\ell\}\\odot\\mathrm\{DropPath\}\\left\(\\sum\_\{a\\in\\mathcal\{A\}\}\\beta\_\{a\}^\{\\ell\}U\_\{a\}^\{\\ell\},p\_\{\\mathrm\{ffn\}\}^\{\\ell\}\\right\)\.
### Multi\-level Tri\-axis Readout
Representations from different encoder depths may contain complementary information\. We therefore attach a tri\-axis attention pooling head to every layer outputXℓX^\{\\ell\}, whereℓ∈\{1,…,N\}\\ell\\in\\\{1,\\ldots,N\\\}\.
ForXℓ∈ℝB×C×S×LX^\{\\ell\}\\in\\mathbb\{R\}^\{B\\times C\\times S\\times L\}, the pooling head constructs one axis\-specific sequence for each axis\. The selected axis indexes the sequence positions, while the other two axes are flattened into the feature dimension:
Zcℓ\\displaystyle Z\_\{c\}^\{\\ell\}∈ℝB×C×SL,\\displaystyle\\in\\mathbb\{R\}^\{B\\times C\\times SL\},Zsℓ\\displaystyle Z\_\{s\}^\{\\ell\}∈ℝB×S×CL,\\displaystyle\\in\\mathbb\{R\}^\{B\\times S\\times CL\},Zlℓ\\displaystyle Z\_\{l\}^\{\\ell\}∈ℝB×L×CS,\\displaystyle\\in\\mathbb\{R\}^\{B\\times L\\times CS\},whereSL=S⋅LSL=S\\cdot L,CL=C⋅LCL=C\\cdot L, andCS=C⋅SCS=C\\cdot S\.
An axis\-specific linear projectionWaℓW\_\{a\}^\{\\ell\}maps each sequence to a shared embedding dimensionEE, after which a learnable positional embeddingPEaℓ\\mathrm\{PE\}\_\{a\}^\{\\ell\}is added:
Haℓ=ZaℓWaℓ\+PEaℓ∈ℝB×\|a\|×E,H\_\{a\}^\{\\ell\}=Z\_\{a\}^\{\\ell\}W\_\{a\}^\{\\ell\}\+\\mathrm\{PE\}\_\{a\}^\{\\ell\}\\in\\mathbb\{R\}^\{B\\times\|a\|\\times E\},where\|a\|∈\{C,S,L\}\|a\|\\in\\\{C,S,L\\\}is the sequence length along axisaa\.
A lightweight attention pooling operatorAttnPoolaℓ\\mathrm\{AttnPool\}\_\{a\}^\{\\ell\}assigns normalized weights to the sequence positions and aggregates them into oneEE\-dimensional vector\. The three axis\-specific vectors are combined to form the representation of layerℓ\\ell:
hℓ=∑a∈𝒜ωaℓAttnPoolaℓ\(Haℓ\)∈ℝB×E,h^\{\\ell\}=\\sum\_\{a\\in\\mathcal\{A\}\}\\omega\_\{a\}^\{\\ell\}\\,\\mathrm\{AttnPool\}\_\{a\}^\{\\ell\}\\left\(H\_\{a\}^\{\\ell\}\\right\)\\in\\mathbb\{R\}^\{B\\times E\},where
𝝎ℓ=softmax\(𝝍ℓ\)∈ℝ3\\boldsymbol\{\\omega\}^\{\\ell\}=\\mathrm\{softmax\}\\left\(\\boldsymbol\{\\psi\}^\{\\ell\}\\right\)\\in\\mathbb\{R\}^\{3\}are learnable fusion weights\.
Finally, learnable depth\-fusion weights combine the representations from allNNencoder layers:
𝝅=softmax\(𝝆\)∈ℝN,z=∑ℓ=1Nπℓhℓ∈ℝB×E\.\\boldsymbol\{\\pi\}=\\mathrm\{softmax\}\\left\(\\boldsymbol\{\\rho\}\\right\)\\in\\mathbb\{R\}^\{N\},\\qquad z=\\sum\_\{\\ell=1\}^\{N\}\\pi\_\{\\ell\}h^\{\\ell\}\\in\\mathbb\{R\}^\{B\\times E\}\.The fused representationzzis passed to a classifier head consisting of LayerNorm, dropout, and a linear classification layer\.
### Experimental Settings
Datasets and evaluation protocols\.We evaluate TriDimEEG on eight EEG benchmarks covering four application scenarios: i\)Neurological disorder diagnosis: AD65\([Miltiadous et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib24)\), \(ii\)Sleep staging: SleepEDF\([Kemp et al\., 2000](https://arxiv.org/html/2609.19842#bib.bib25);[Goldberger et al\., 2000](https://arxiv.org/html/2609.19842#bib.bib26)\), \(iii\)Motor imagery: BCI\-IV\-2A\([Tangermann et al\., 2012](https://arxiv.org/html/2609.19842#bib.bib27)\), SHU\-MI\([Ma et al\., 2022](https://arxiv.org/html/2609.19842#bib.bib28)\), PhysioNet\-MI\([Schalk et al\., 2004](https://arxiv.org/html/2609.19842#bib.bib29);[Goldberger et al\., 2000](https://arxiv.org/html/2609.19842#bib.bib26)\), and \(iv\)Emotion recognition: FACED\([Chen et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib30)\), SEED\([Zheng and Lu, 2015](https://arxiv.org/html/2609.19842#bib.bib31)\), and SEED\-V\([Liu et al\., 2022](https://arxiv.org/html/2609.19842#bib.bib32)\)\. Table[1](https://arxiv.org/html/2609.19842#Sx3.T1)summarizes the processed datasets\.
We adopt a strict subject\-independent evaluation protocol and divide subjects into training, validation, and test sets with an approximately 8:1:1 ratio\. All recordings, sessions, and segments from the same subject are assigned to the same subset to prevent subject leakage\. The subject\-level partitioning procedure is repeated three times using different random seeds, yielding three predefined split manifests shared by all methods\.
Table 1:Statistics of the processed EEG datasets\. BCI\-2A and PNet\-MI stand for BCI\-IV\-2A and PhysioNet\-MI, respectively\. Subj\. and Ch\. denote subjects and channels\.Preprocessing\.All recordings are resampled to 200 Hz, band\-pass filtered from 0\.3 to 75 Hz, and notch filtered at the power\-line frequency corresponding to each recording source\. AD65, FACED, SEED, and SEED\-V are divided into 10\-s windows; BCI\-IV\-2A, SHU\-MI, and PhysioNet\-MI use 4\-s windows; and SleepEDF uses standard 30\-s sleep epochs following OmniEEG\-bench\([Lu et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib1)\)\.
Baselines\.We compare TriDimEEG with four supervised EEG architectures: EEGNet\([Lawhern et al\., 2018](https://arxiv.org/html/2609.19842#bib.bib5)\), EEGConformer\([Song et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib8)\), EEGDeformer\([Ding et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib10)\), and TeCh\([Yu et al\., 2026](https://arxiv.org/html/2609.19842#bib.bib33)\)\. We also evaluate ten pretrained EEG models: BENDR\([Kostas et al\., 2021](https://arxiv.org/html/2609.19842#bib.bib34)\), BIOT\([Yang et al\., 2023](https://arxiv.org/html/2609.19842#bib.bib12)\), LaBraM\([Jiang et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib13)\), CBraMod\([Wang et al\., 2025a](https://arxiv.org/html/2609.19842#bib.bib15)\), NeuroGPT\([Cui et al\., 2024](https://arxiv.org/html/2609.19842#bib.bib35)\), EEGMamba\([Wang et al\., 2025b](https://arxiv.org/html/2609.19842#bib.bib36)\), FEMBA\([Tegon et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib37)\), NeuroLM\([Jiang et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib38)\), BrainOmni\([Xiao et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib17)\), and REVE\([El Ouahidi et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib39)\)\. All methods use the same subject\-level split manifests\. For pretrained models, the complete pretrained encoder is fine\-tuned jointly with a task\-specific classification head on the training split\. Necessary input adaptations follow the official implementations, with further details provided in Appendix[E](https://arxiv.org/html/2609.19842#A5)\.
## Results
We evaluate TriDim through four analyses: cross\-subject benchmarking on eight EEG datasets, integration into three pretrained EEG encoders as replacing blocks, ablation studies, and interpretation analysis\. Results are reported as mean±\\pmsample standard deviation over three subject\-level splits, together with average rank across datasets\. Macro\-F1 results are provided in Appendix[C](https://arxiv.org/html/2609.19842#A3)\.
### Cross\-Subject EEG Decoding Performance
Table[2](https://arxiv.org/html/2609.19842#Sx4.T2)compares TriDimEEG with representative supervised and pretrained EEG models across clinical diagnosis, sleep staging, motor imagery, and emotion recognition tasks\.
Table 2:Cross\-subject classification accuracy \(%\) on eight EEG datasets\. BCI\-2A and PNet\-MI stand for BCI\-IV\-2A and PhysioNet\-MI, respectively\. The best and second\-best results in each column are highlighted in bold and underlined, respectively\. Lower average rank indicates better overall performance\. Results are reported as mean±\\pmstandard deviation\. Averages and ranks are computed across all eight datasets\.TriDimEEG achieves the highest average accuracy of57\.70%57\.70\\%and the best average rank of2\.752\.75, achieving a4\.3%4\.3\\%relative improvement in average accuracy over the second\-best model, EEGDeformer, and improving the average rank from4\.384\.38to2\.752\.75\. It ranks first on AD65, SleepEDF, BCI\-IV\-2A, and FACED, second on PhysioNet\-MI, and third on SHU\-MI\. EEGDeformer remains strongest on SHU\-MI and PhysioNet\-MI, whereas REVE and LaBraM lead on SEED and SEED\-V\. However, these models are less consistent across the full benchmark\. Although trained without large\-scale EEG pretraining, TriDimEEG exceeds every evaluated pretrained encoder in both average accuracy and average rank\.
### Replacing Transformer Blocks in Pretrained EEG Encoders
To evaluate TriDim as a reusable architectural block, we replace the original Transformer\-based blocks in REVE, CBraMod, and CSBrain\([Zhou et al\., 2025](https://arxiv.org/html/2609.19842#bib.bib18)\)with TriDim blocks\. For each encoder, the original and TriDim\-based variants are pretrained on the same2,0002\{,\}000TUH samples using identical pretraining and downstream settings\. Details are provided in Appendix[F](https://arxiv.org/html/2609.19842#A6)\.
Figure 3:Replacing Transformer blocks with TriDim blocks under the same TUH\-2K pretraining setting improves accuracy\. Arrow annotations indicate relative improvements over the corresponding original encoder\.Replacing Transformer\-based blocks with TriDim blocks yields relative improvements in average downstream accuracy of11\.91%11\.91\\%,3\.05%3\.05\\%, and7\.35%7\.35\\%over the original REVE, CBraMod, and CSBrain encoders, while reducing parameters by32\.0%32\.0\\%,17\.0%17\.0\\%, and47\.3%47\.3\\%, respectively\. The mean of these three relative improvements is7\.4%7\.4\\%, showing that TriDim is reusable across pretrained architectures while improving parameter efficiency\.
### Ablation Studies
We evaluate the contributions of two aspects of TriDimEEG: the three representation axes and the design choices within the TriDim backbone \(Table[3](https://arxiv.org/html/2609.19842#Sx4.T3)\)\. The first group removes or isolates the channel, short\-term temporal, and long\-term temporal axes\. The second group evaluates the multi\-level readout, cross\-axis attention, axis\-specific feed\-forward networks, attention sharing, and parallel axis integration\.
Removing the channel, short\-term temporal, or long\-term temporal axis decreases average accuracy by2\.3%2\.3\\%,0\.5%0\.5\\%, and1\.6%1\.6\\%, respectively, relative to the Full model\. The larger losses forCCandLLhighlight the importance of using channel or long\-term temporal information as the embedding dimensions, which were ignored in previous studies\. Retaining only one\-axis branch obtain55\.44%55\.44\\%,55\.04%55\.04\\%, and54\.46%54\.46\\%forCC,SS, andLL, respectively, all worse than the two\-axis model or full TriDimEEG\. Together, these results indicate three\-axis representations provide complementary information\.
Table 3:Ablation accuracy \(%, mean±\\pmstandard deviation\)\. BCI\-2A and PNet\-MI denote BCI\-IV\-2A and PhysioNet\-MI\. The best and second\-best results in each column are highlighted in bold and underlined, respectively\. Avg\. averages all eight datasets\.For model design choices, using only the final layer lowers average accuracy by0\.7%0\.7\\%relative to the Full model\. Removing cross\-axis attention and sharing one FFN across axes cause relative reductions of1\.0%1\.0\\%and1\.2%1\.2\\%, respectively, supporting the advantage of combining cross\-axis interaction and axis\-specific transformation\. Independent attention reaches a comparable performance of57\.69%57\.69\\%but uses twice as many parameters in attention modules, showing that parameter sharing preserves accuracy efficiently\. SequentialC→S→LC\\rightarrow S\\rightarrow Lprocessing lowers average accuracy by1\.0%1\.0\\%relative to the Full model, supporting the advantage of parallel integration over sequential processing\.
### Interpretation analysis
To examine whether the three TriDim axes capture physiologically meaningful and class\-specific EEG representations, we conduct perturbation analyses along the channel, within\-patch time, and across\-patch time dimensions\. These three perturbations characterize the spatial, local spectral, and temporal information used by TriDimEEG for motor imagery decoding, respectively\.
We analyze class\-conditional sensitivity on the patched representationXpatch∈ℝB×Cin×Sin×LinX^\{\\mathrm\{patch\}\}\\in\\mathbb\{R\}^\{B\\times C\_\{in\}\\times S\_\{in\}\\times L\_\{in\}\}\. Using 500\-ms patches with a 50\-ms stride, channel perturbation zeros one electrode across all patch positions, spectral perturbation removes one 2\-Hz component within every local patch, and temporal perturbation zeros one complete patch position\. We report the signed drop in true\-class probability over nine leave\-one\-subject\-out models, with 95% subject\-bootstrap confidence intervals in Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)\.
Figure 4:Class\-conditional C/S/L\-axis perturbations on BCI\-IV\-2A using channel, 2\-Hz spectral, and 500\-ms patch occlusion\. Shading denotes 95% confidence intervals across nine held\-out subjects\.Channel occlusionproduces class\-dependent scalp patterns \(Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)\(a\)\)\. The hand classes predominantly rely on contralateral centroparietal regions, whereas the feet class shows the strongest dependence near the central midline\.Spectral perturbationshows its largest class\-specific effects around 11–13 Hz \(Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)\(b\)\), overlapping the sensorimotorμ\\murhythm\. The effect is strongly positive for tongue, moderately positive for feet, and negative for right hand, indicating that the same spectral component provides different evidence across classes\. After aggregating the signed drops across classes within each subject, the overall response retains a clear 11–13 Hz peak \(1\.661\.66pp; 95% CI:0\.440\.44–3\.143\.14pp\), while other frequencies remain close to zero\.Temporal patch occlusionis strongest during the early trial period for the hand classes \(Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)\(c\)\)\. Feet exhibits a later peak around 1\.8–2\.0 s, whereas tongue shows an early negative response followed by weaker positive effects\. Together, these results show that TriDimEEG can capture task\-dependent spatial, local spectral, and temporal representations\.
## Conclusion
We introduced TriDim, a shape\-preserving tri\-axis block that jointly models channel organization, short\-term waveform structure, and long\-term temporal context through progressive axis\-specific refinement and cross\-axis interaction\. Under strict cross\-subject evaluation across eight EEG datasets, TriDimEEG achieved the highest average accuracy and best average rank among 15 models, with ablation and perturbation analyses confirming the complementary contributions of the three axes\. As a drop\-in replacement for Transformer blocks in three pretrained EEG encoders, TriDim achieved a mean relative improvement of7\.4%7\.4\\%in average downstream accuracy over the corresponding original encoders and reduced parameter counts\. These results demonstrate that TriDim is an effective and reusable backbone block for task\-specific and pretrained EEG models\.
## Acknowledgments
This work was supported by the National Natural Science Foundation of China \(62606214, 3254100307, and 62472206\), the National Science and Technology Major Project \(2021ZD0200500\), the National Key R&D Program of China \(2025YFC3410000\), the Guangdong Basic and Applied Basic Research Foundation \(2026A1515010121\), the Guangdong S&T Program \(2026B0101110003\), the Shenzhen Science and Technology Innovation Committee \(RCYX20231211090405003 and JCYJ20220818100213029\), the Guangdong Provincial Key Laboratory of Advanced Biomaterials \(2022B1212010003\), and the open research fund of the Guangdong Provincial Key Laboratory of Mathematical and Neural Dynamical Systems and the Center for Computational Science and Engineering at Southern University of Science and Technology\.
## References
- J\. Chen, X\. Wang, C\. Huang, X\. Hu, X\. Shen, and D\. ZhangA large finer\-grained affective computing EEG dataset\.Scientific Data10\(1\),pp\. 740\.External Links:[Document](https://dx.doi.org/10.1038/s41597-023-02650-w)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Craiket al\.\(2019\)A\. Craik, Y\. He, and J\. L\. Contreras\-VidalDeep learning for electroencephalogram \(EEG\) classification tasks: a review\.Journal of Neural Engineering16\(3\),pp\. 031001\.External Links:[Document](https://dx.doi.org/10.1088/1741-2552/ab0ab5)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1)\.
- Cuiet al\.\(2024\)W\. Cui, W\. Jeong, P\. Thölke, T\. Medani, K\. Jerbi, A\. A\. Joshi, and R\. M\. LeahyNeuro\-GPT: towards a foundation model for EEG\.In2024 IEEE International Symposium on Biomedical Imaging \(ISBI\),pp\. 1–5\.External Links:[Document](https://dx.doi.org/10.1109/ISBI56570.2024.10635453)Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Dinget al\.\(2025\)Y\. Ding, Y\. Li, H\. Sun, R\. Liu, C\. Tong, C\. Liu, X\. Zhou, and C\. GuanEEG\-Deformer: a dense convolutional transformer for brain–computer interfaces\.IEEE Journal of Biomedical and Health Informatics29\(3\),pp\. 1909–1918\.External Links:[Document](https://dx.doi.org/10.1109/JBHI.2024.3504604)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Dinget al\.\(2023\)Y\. Ding, N\. Robinson, S\. Zhang, Q\. Zeng, and C\. GuanTSception: capturing temporal dynamics and spatial asymmetry from EEG for emotion recognition\.IEEE Transactions on Affective Computing14\(3\),pp\. 2238–2250\.External Links:[Document](https://dx.doi.org/10.1109/TAFFC.2022.3169001)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1),[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1)\.
- Donget al\.\(2021\)Y\. Dong, J\. Cordonnier, and A\. LoukasAttention is not all you need: pure attention loses rank doubly exponentially with depth\.InInternational conference on machine learning,pp\. 2793–2803\.Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1)\.
- El Ouahidiet al\.\(2025\)Y\. El Ouahidi, J\. Lys, P\. Thölke, N\. Farrugia, B\. Pasdeloup, V\. Gripon, K\. Jerbi, and G\. LioiREVE: a foundation model for EEG—adapting to any setup with large\-scale pretraining on 25,000 subjects\.InAdvances in Neural Information Processing Systems,Vol\.38\.Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Gevaet al\.\(2020\)M\. Geva, R\. Schuster, J\. Berant, and O\. LevyTransformer feed\-forward layers are key\-value memories\.arXiv preprint arXiv:2012\.14913\.Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1)\.
- Goldbergeret al\.\(2000\)A\. L\. Goldberger, L\. A\. N\. Amaral, L\. Glass, J\. M\. Hausdorff, P\. Ch\. Ivanov, R\. G\. Mark, J\. E\. Mietus, G\. B\. Moody, C\. Peng, and H\. E\. StanleyPhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals\.Circulation101\(23\),pp\. e215–e220\.External Links:[Document](https://dx.doi.org/10.1161/01.CIR.101.23.e215)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Huanget al\.\(2016\)G\. Huang, Y\. Sun, Z\. Liu, D\. Sedra, and K\. Q\. WeinbergerDeep networks with stochastic depth\.InEuropean Conference on Computer Vision,pp\. 646–661\.Cited by:[TriDim Block](https://arxiv.org/html/2609.19842#Sx3.SSx2.p6.1)\.
- Jianget al\.\(2025\)W\. Jiang, Y\. Wang, B\. Lu, and D\. LiNeuroLM: a universal multi\-task foundation model for bridging the gap between language and EEG signals\.InInternational Conference on Learning Representations,Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Jianget al\.\(2024\)W\. Jiang, L\. Zhao, and B\. LuLarge brain model for learning generic representations with tremendous EEG data in BCI\.InInternational Conference on Learning Representations,Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Kempet al\.\(2000\)B\. Kemp, A\. H\. Zwinderman, B\. Tuk, H\. A\. C\. Kamphuisen, and J\. J\. L\. OberyeAnalysis of a sleep\-dependent neuronal feedback loop: the slow\-wave microcontinuity of the EEG\.IEEE Transactions on Biomedical Engineering47\(9\),pp\. 1185–1194\.External Links:[Document](https://dx.doi.org/10.1109/10.867928)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Kostaset al\.\(2021\)D\. Kostas, S\. Aroca\-Ouellette, and F\. RudziczBENDR: using transformers and a contrastive self\-supervised learning task to learn from massive amounts of EEG data\.Frontiers in Human Neuroscience15,pp\. 653659\.External Links:[Document](https://dx.doi.org/10.3389/fnhum.2021.653659)Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Lawhernet al\.\(2018\)V\. J\. Lawhern, A\. J\. Solon, N\. R\. Waytowich, S\. M\. Gordon, C\. P\. Hung, and B\. J\. LanceEEGNet: a compact convolutional neural network for EEG\-based brain–computer interfaces\.Journal of Neural Engineering15\(5\),pp\. 056013\.External Links:[Document](https://dx.doi.org/10.1088/1741-2552/aace8c)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1),[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Liuet al\.\(2022\)W\. Liu, J\. Qiu, W\. Zheng, and B\. LuComparing recognition performance and robustness of multimodal deep learning models for multimodal emotion recognition\.IEEE Transactions on Cognitive and Developmental Systems14\(2\),pp\. 715–729\.External Links:[Document](https://dx.doi.org/10.1109/TCDS.2021.3071170)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Luet al\.\(2026\)Z\. Lu, Z\. Li, X\. Shen, K\. Lou, Y\. Xin, X\. Chen, S\. Wang, X\. Chen, J\. Fan, C\. Huang,et al\.OmniEEG\-bench: a standardized evaluation benchmark for eeg foundation models\.arXiv preprint arXiv:2606\.00815\.Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p3.1)\.
- Maet al\.\(2022\)J\. Ma, B\. Yang, W\. Qiu, Y\. Li, S\. Gao, and X\. XiaA large EEG dataset for studying cross\-session variability in motor imagery brain\-computer interface\.Scientific Data9\(1\),pp\. 531\.External Links:[Document](https://dx.doi.org/10.1038/s41597-022-01647-1)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Maet al\.\(2026\)Z\. Ma, Z\. Li, Z\. Qiu, J\. Li, L\. Meng, X\. Zhang, Y\. Liu, X\. Shen, and S\. SongDSAINet: an efficient dual\-scale attentive interaction network for general EEG decoding\.External Links:2604\.18095Cited by:[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1)\.
- Miaoet al\.\(2023\)Z\. Miao, M\. Zhao, X\. Zhang, and D\. MingLMDA\-Net: a lightweight multi\-dimensional attention network for general EEG\-based brain–computer interfaces and interpretability\.NeuroImage276,pp\. 120209\.External Links:[Document](https://dx.doi.org/10.1016/j.neuroimage.2023.120209)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1)\.
- Miltiadouset al\.\(2023\)A\. Miltiadous, K\. D\. Tzimourta, T\. Afrantou, P\. Ioannidis, N\. Grigoriadis, D\. G\. Tsalikakis, P\. Angelidis, M\. G\. Tsipouras, E\. Glavas, N\. Giannakeas, and A\. T\. TzallasA dataset of EEG recordings from: Alzheimer’s disease, frontotemporal dementia and healthy subjects\.Data in Brief48,pp\. 109297\.External Links:[Document](https://dx.doi.org/10.1016/j.dib.2023.109297)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Royet al\.\(2019\)Y\. Roy, H\. Banville, I\. Albuquerque, A\. Gramfort, T\. H\. Falk, and J\. FaubertDeep learning\-based electroencephalography analysis: a systematic review\.Journal of Neural Engineering16\(5\),pp\. 051001\.External Links:[Document](https://dx.doi.org/10.1088/1741-2552/ab260c)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1)\.
- Schalket al\.\(2004\)G\. Schalk, D\. J\. McFarland, T\. Hinterberger, N\. Birbaumer, and J\. R\. WolpawBCI2000: a general\-purpose brain\-computer interface \(BCI\) system\.IEEE Transactions on Biomedical Engineering51\(6\),pp\. 1034–1043\.External Links:[Document](https://dx.doi.org/10.1109/TBME.2004.827072)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Songet al\.\(2023\)Y\. Song, Q\. Zheng, B\. Liu, and X\. GaoEEG Conformer: convolutional transformer for EEG decoding and visualization\.IEEE Transactions on Neural Systems and Rehabilitation Engineering31,pp\. 710–719\.External Links:[Document](https://dx.doi.org/10.1109/TNSRE.2022.3230250)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1),[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Tangermannet al\.\(2012\)M\. Tangermann, K\. Müller, A\. Aertsen, N\. Birbaumer, C\. Braun, C\. Brunner, R\. Leeb, C\. Mehring, K\. J\. Miller, G\. Mueller\-Putz, G\. Nolte, G\. Pfurtscheller, H\. Preissl, G\. Schalk, A\. Schlögl, C\. Vidaurre, S\. Waldert, and B\. BlankertzReview of the BCI competition IV\.Frontiers in Neuroscience6,pp\. 55\.External Links:[Document](https://dx.doi.org/10.3389/fnins.2012.00055)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Tegonet al\.\(2025\)A\. Tegon, T\. M\. Ingolfsson, X\. Wang, L\. Benini, and Y\. LiFEMBA: efficient and scalable EEG analysis with a bidirectional Mamba foundation model\.In2025 47th Annual International Conference of the IEEE Engineering in Medicine and Biology Society \(EMBC\),pp\. 1–7\.External Links:[Document](https://dx.doi.org/10.1109/EMBC58623.2025.11252697)Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Touvronet al\.\(2021\)H\. Touvron, M\. Cord, A\. Sablayrolles, G\. Synnaeve, and H\. JégouGoing deeper with image transformers\.InIEEE/CVF International Conference on Computer Vision \(ICCV\),pp\. 32–42\.Cited by:[TriDim Block](https://arxiv.org/html/2609.19842#Sx3.SSx2.p7.2)\.
- Wanet al\.\(2023\)Z\. Wan, M\. Li, S\. Liu, J\. Huang, H\. Tan, and W\. DuanEEGformer: a transformer\-based brain activity classification method using EEG signal\.Frontiers in Neuroscience17,pp\. 1148855\.External Links:[Document](https://dx.doi.org/10.3389/fnins.2023.1148855)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1)\.
- Wanget al\.\(2024\)G\. Wang, W\. Liu, Y\. He, C\. Xu, L\. Ma, and H\. LiEEGPT: pretrained transformer for universal and reliable representation of EEG signals\.InAdvances in Neural Information Processing Systems,Vol\.37\.Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1)\.
- Wanget al\.\(2025a\)J\. Wang, S\. Zhao, Z\. Luo, Y\. Zhou, H\. Jiang, S\. Li, T\. Li, and G\. PanCBraMod: a criss\-cross brain foundation model for EEG decoding\.InInternational Conference on Learning Representations,Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1),[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Structure\-Preserving EEG Architectures](https://arxiv.org/html/2609.19842#Sx2.SSx3.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Wanget al\.\(2025b\)J\. Wang, S\. Zhao, Z\. Luo, Y\. Zhou, S\. Li, and G\. PanEEGMamba: an EEG foundation model with Mamba\.Neural Networks192,pp\. 107816\.External Links:[Document](https://dx.doi.org/10.1016/j.neunet.2025.107816)Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Wanget al\.\(2026\)Y\. Wang, D\. Lee, and B\. X\. B\. YuBraSTORM: a dual\-branch self\-supervised framework for EEG representation learning via input\-level spatio\-temporal decomposition\.InProceedings of the AAAI Conference on Artificial Intelligence,Vol\.40,pp\. 17805–17813\.External Links:[Document](https://dx.doi.org/10.1609/aaai.v40i21.38838)Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p1.1),[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Structure\-Preserving EEG Architectures](https://arxiv.org/html/2609.19842#Sx2.SSx3.p1.1)\.
- Xiaoet al\.\(2025\)Q\. Xiao, Z\. Cui, C\. Zhang, S\. Chen, W\. Wu, A\. Thwaites, A\. Woolgar, B\. Zhou, and C\. ZhangBrainOmni: a brain foundation model for unified EEG and MEG signals\.InAdvances in Neural Information Processing Systems,Vol\.38\.Cited by:[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Xuet al\.\(2024\)Y\. Xu, C\. Li, D\. Li, X\. Sheng, F\. Jiang, L\. Tian, A\. Sirasao, and E\. BarsoumEnhancing vision transformer: amplifying non\-linearity in feedforward network module\.InForty\-first International Conference on Machine Learning,Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1)\.
- Yanget al\.\(2023\)C\. Yang, M\. B\. Westover, and J\. SunBIOT: biosignal transformer for cross\-data learning in the wild\.InAdvances in Neural Information Processing Systems,Vol\.36\.Cited by:[Introduction](https://arxiv.org/html/2609.19842#Sx1.p2.1),[Pretrained EEG Representation Models](https://arxiv.org/html/2609.19842#Sx2.SSx2.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Yuet al\.\(2026\)G\. Yu, J\. Wang, C\. Yang, J\. Qin, A\. I\. Aviles\-Rivero, and S\. WangDecentralized attention fails centralized signals: rethinking transformers for medical time series\.InInternational Conference on Learning Representations,Cited by:[Task\-Specific EEG Decoders](https://arxiv.org/html/2609.19842#Sx2.SSx1.p1.1),[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p4.1)\.
- Zhang and Sennrich \(2019\)B\. Zhang and R\. SennrichRoot mean square layer normalization\.InAdvances in Neural Information Processing Systems \(NeurIPS\),Vol\.32\.Cited by:[TriDim Block](https://arxiv.org/html/2609.19842#Sx3.SSx2.p2.1)\.
- Zheng and Lu \(2015\)W\. Zheng and B\. LuInvestigating critical frequency bands and channels for EEG\-based emotion recognition with deep neural networks\.IEEE Transactions on Autonomous Mental Development7\(3\),pp\. 162–175\.External Links:[Document](https://dx.doi.org/10.1109/TAMD.2015.2431497)Cited by:[Experimental Settings](https://arxiv.org/html/2609.19842#Sx3.SSx4.p1.1)\.
- Zhouet al\.\(2025\)Y\. Zhou, J\. Wu, Z\. Ren, Z\. Yao, W\. Lu, K\. Peng, Q\. Zheng, C\. Song, W\. Ouyang, and C\. GouCSBrain: a cross\-scale spatiotemporal brain foundation model for EEG decoding\.InAdvances in Neural Information Processing Systems,Vol\.38\.Cited by:[Structure\-Preserving EEG Architectures](https://arxiv.org/html/2609.19842#Sx2.SSx3.p1.1),[Replacing Transformer Blocks in Pretrained EEG Encoders](https://arxiv.org/html/2609.19842#Sx4.SSx2.p1.1)\.
Supplementary Material
## Appendix ADataset and Preprocessing Details
Table[1](https://arxiv.org/html/2609.19842#Sx3.T1)of the main text summarizes the eight processed datasets\. All datasets are publicly available for research purposes and are cited in the main text\. Raw recordings are preprocessed with a unified pipeline: resampling to 200 Hz, a 0\.3–75 Hz band\-pass filter, and a notch filter at the local power\-line frequency\. AD65, FACED, SEED, and SEED\-V are segmented into 10\-s windows; BCI\-IV\-2A, SHU\-MI, and PhysioNet\-MI use 4\-s windows; SleepEDF uses standard 30\-s sleep epochs\. Per\-sample, per\-channel temporal normalization is performed by the InstanceTimeNorm layer at the model input, so no dataset\-level statistics are computed or leaked across subjects\.
For SEED\-V we use the official 10\-s benchmark segments; the label of each 10\-s segment is inherited from its source 1\-s annotation block\. For SleepEDF we follow the supplied session\-level protocol \(153 sessions split 122/15/16\) and retain at most 40 30\-s epochs per session–sleep\-stage pair in the training subset only \(330,374→\\rightarrow22,861 training epochs; the frozen manifests record the exact retained indices\), which both bounds the training\-set size and balances the five sleep stages \(4,880/4,787/3,450/4,880/4,864 epochs for seed 5\)\. Validation and test sessions are kept complete without truncation or downsampling \(41,150 and 43,437 epochs\), preserving the natural class distribution\. The same retained\-epoch manifests are used by every compared method on SleepEDF\.
## Appendix BEvaluation Protocol
All methods are evaluated under a strict subject\-independent protocol\. Subjects are partitioned into training, validation, and test sets with an approximate 8:1:1 ratio, stratified by class where applicable; every recording, session, and segment of a subject is assigned to exactly one subset\. Subject\-level split manifests are generated before training, frozen as immutable JSON files, and shared by all compared methods, so every model is trained and tested on identical subject partitions\. The frozen manifests for SEED\-V and SleepEDF are shipped with the released code, together with their SHA\-256 checksums; the remaining datasets use a seeded stratified subject splitter \(train ratio 0\.8, validation ratio 0\.1\) whose outputs are equally immutable for a given seed\.
Each configuration is run under three independent subject\-level splits generated by random seeds 5, 42, and 43\. Within each run, the checkpoint with the highest validation accuracy is selected for final testing\. We report the mean and sample standard deviation \(ddof = 1\) of test accuracy across the three splits\. As aggregate measures across the eight datasets we report \(i\) average accuracy and \(ii\) average rank, where each model is ranked per dataset by mean accuracy \(rank 1 = best\) and the ranks are averaged; a lower average rank indicates better overall performance\. Macro\-F1 scores of the Full model and all ablation variants are reported in Supplementary Table[1](https://arxiv.org/html/2609.19842#A3.T1); per\-run records of every metric are included in the released result files\.
Performance changes in the text are expressed relative to the stated reference:100\(Anew−Aref\)/Aref%100\(A\_\{\\mathrm\{new\}\}\-A\_\{\\mathrm\{ref\}\}\)/A\_\{\\mathrm\{ref\}\}\\%, whereAAis the accuracy or macro\-F1 for the comparison\. For cross\-dataset comparisons,AAis the mean across the eight datasets\. Ablation reductions use the Full model as the reference\. Paired statistical differences and perturbation\-induced probability changes retain their percentage\-point units\.
## Appendix CMacro\-F1 Result Records
Supplementary Table[1](https://arxiv.org/html/2609.19842#A3.T1)reports the macro\-F1 score of the Full model and all eleven ablation variants on each of the eight datasets, as the mean±\\pmsample standard deviation across the three subject\-level splits \(seeds 5/42/43\); the underlying runs are exactly those behind Tables[2](https://arxiv.org/html/2609.19842#Sx4.T2)and[3](https://arxiv.org/html/2609.19842#Sx4.T3)of the main text\. Per\-split accuracy and macro\-F1 records of every run, including the compared baselines, are included in the released result files\. Average macro\-F1 and average rank are computed across all eight datasets; a lower average rank indicates better overall cross\-dataset ranking\.
Table 1:Macro\-F1 \(%\) of Full TriDimEEG and all ablation variants on the eight datasets\. Each cell reports the mean±\\pmsample standard deviation \(ddof = 1\) across the three subject\-level splits \(seeds 5/42/43\); the underlying runs are exactly those behind Tables[2](https://arxiv.org/html/2609.19842#Sx4.T2)and[3](https://arxiv.org/html/2609.19842#Sx4.T3)of the main text\. The best and second\-best results in each column are highlighted in bold and underlined, respectively\.
## Appendix DLeave\-One\-Subject\-Out Evaluation
The 8:1:1 subject partition used in the main table is retained for consistency across all eight datasets, but on small\-cohort datasets it yields test sets of one to three subjects\. We therefore additionally evaluate the Full TriDimEEG model under leave\-one\-subject\-out \(LOSO\) cross\-validation on the four small\-cohort datasets: BCI\-IV\-2A \(9 subjects\), SHU\-MI \(25 subjects\), SEED \(15 subjects\), and SEED\-V \(16 subjects\)\. BCI\-IV\-2A is the most affected case and is also the dataset with the most established LOSO convention; we report its complete fold×\\timesseed grid below and summarize the remaining three datasets at the fold level\.
#### Protocol\.
For foldk∈\{1,…,9\}k\\in\\\{1,\\dots,9\\\}, subject A0kkis the test subject, subject A0\(\(kmod9\)\+1\)\(\(k\\bmod 9\)\+1\)is the validation subject, and the remaining seven subjects form the training set\. Every fold is trained under the same three seeds \(5, 42, 43\) and the identical per\-dataset configuration as the main\-table BCI\-IV\-2A runs \(Supplementary Tables[5](https://arxiv.org/html/2609.19842#A7.T5)and[6](https://arxiv.org/html/2609.19842#A7.T6)\), with validation\-based checkpoint selection\. The fold manifests are frozen JSON files with disjoint train/validation/test trial indices, shipped with the released code\. The LOSO study uses the fixed BCI\-IV\-2A trial asset of 5,088 trials \(subject A04 contributes 480 trials; all other subjects 576\); Table[1](https://arxiv.org/html/2609.19842#Sx3.T1)of the main text reports the raw segment count \(5,184 =9×5769\\times 576\), and the released trial asset excludes 96 trials of subject A04, yielding the fixed 5,088\-trial set used by all BCI\-IV\-2A experiments\. The perturbation analysis uses the same nine seed\-42 LOSO subject partitions, but retrains a separate set of models with 500\-ms patches and a 50\-ms stride\. These interpretation models are therefore distinct from the 200\-ms\-patch models reported in Supplementary Table[2](https://arxiv.org/html/2609.19842#A4.T2)\.
Table 2:LOSO cross\-subject test accuracy \(%\) of Full TriDimEEG on BCI\-IV\-2A\. Each cell is one trained model; mean±\\pmstd is computed across the three seeds per fold\. The overall row reports the mean±\\pmstd across all 27 fold×\\timesseed runs\.
#### Results\.
Supplementary Table[2](https://arxiv.org/html/2609.19842#A4.T2)reports the complete fold×\\timesseed grid\. TriDim obtains51\.1±7\.4951\.1\\pm 7\.49% under LOSO, below its 8:1:1 result \(56\.6±2\.7156\.6\\pm 2\.71%\); this gap is expected because LOSO evaluates every subject, including the hardest ones, whereas the 8:1:1 protocol tests on a single drawn subject per split\. Per\-fold means range from42\.442\.4% \(A02\) to61\.561\.5% \(A03\), reflecting the well\-documented subject difficulty spread of this benchmark\. We report this LOSO study as a protocol\-robustness check; the main\-table protocol is unchanged for cross\-dataset comparability\.
#### LOSO on SHU\-MI, SEED, and SEED\-V\.
We additionally ran LOSO cross\-validation of the Full model \(seed 42, identical per\-dataset configurations as the main table\) on the three remaining small\-cohort datasets, with every subject serving once as the test subject and no subject leakage between training and testing\. Supplementary Table[3](https://arxiv.org/html/2609.19842#A4.T3)summarizes the fold\-level results\. On all three datasets the LOSO accuracy is consistent with the 8:1:1 three\-split results of the main table \(60\.2±7\.1660\.2\\pm 7\.16on SHU\-MI,52\.9±2\.2252\.9\\pm 2\.22on SEED, and29\.0±4\.3429\.0\\pm 4\.34on SEED\-V\), indicating that the main conclusions are robust to the choice of subject\-partition protocol\.
Table 3:LOSO cross\-subject results of Full TriDimEEG \(seed 42\) on SHU\-MI, SEED, and SEED\-V\. Accuracy and macro\-F1 \(%\) are reported as mean±\\pmstd across folds \(one test subject per fold\)\. The 3\-split column recalls the main\-table accuracy for reference\.
## Appendix EBaseline Compatibility and Adaptation Details
All baselines are evaluated under the same frozen subject\-level manifests, the same validation\-based checkpoint selection, and the same early\-stopping protocol as TriDim\. Input\-shape adaptation follows each official implementation without altering the subject partition: recordings are resampled, filtered, and segmented by the unified pipeline \(see the*Dataset and Preprocessing Details*section\), and each baseline receives its inputs in the channel layout its official code defines\. When a dataset montage cannot supply the channels or derivations a baseline requires, the corresponding inputs are zero\-filled; no method\-specific subject re\-partitioning, re\-segmentation, or label remapping is applied\.
## Appendix FPretrained\-Encoder Integration: Protocol and Per\-Split Results
#### Integration protocol\.
For each host encoder \(REVE, CBraMod, CSBrain\), the original backbone blocks are replaced with TriDim blocks and the modified encoder is*re\-pretrained from scratch*with the host’s original pretraining objective \(masked reconstruction\)\. For a controlled comparison, both the original and the TriDim\-based backbones are pretrained on the same2,0002\{,\}000samples selected from the TUH corpus \(TUH\-2K\), with identical pretraining schedules and otherwise unchanged components \(input frontend, pretraining head, positional schemes\)\. Downstream evaluation then loads the re\-pretrained weights and fine\-tunes*all*parameters end\-to\-end with a newly initialized linear classification head—no module is frozen, and shape\-compatible host tensors are loaded with a non\-strict state\-dict mapping\. Because this reduced TUH\-2K pretraining set differs from the original released checkpoints used in Table[2](https://arxiv.org/html/2609.19842#Sx4.T2)of the main text, the absolute numbers of this section are not directly comparable with the main table; the comparison of interest is original versus TriDim\-based backbone within each encoder under identical settings\.
#### Tensor interface\.
TriDim operates on a\[B,C,S,L\]\[B,C,S,L\]tensor; each host supplies its own grid mapping\. For CBraMod, the backbone is natively four\-dimensional\[B,C,P,dmodel\]\[B,C,P,d\_\{\\mathrm\{model\}\}\]withdmodel=200d\_\{\\mathrm\{model\}\}=200; the channel, patch, anddmodeld\_\{\\mathrm\{model\}\}axes map directly onto theCC,SS, andLLaxes of TriDim, and the convolutional stem, FFT features, and ACPE positional encoding are retained\. For CSBrain, the backbone tensor\[B,D,C,P\]\[B,D,C,P\]is mapped with the feature dimensionDDas the channel axis, the brain\-region axis asSS, and the patch axis asLL\. For REVE, the flattened token sequence is rearranged into a two\-dimensional token grid \(b \(c h\) d→\\rightarrowb c h d\), with the host’s Fourier 3D\-coordinate and time\-index positional embeddings retained on the tokens\. In all three cases no information is discarded by the mapping; the token count and feature width of the host are preserved\.
#### Result records\.
Supplementary Table[4](https://arxiv.org/html/2609.19842#A6.T4)lists the per\-dataset mean test accuracy of the original and TriDim\-based encoders under the TUH\-2K setting\. Per\-split values are included in the released result files\.
Table 4:Per\-dataset mean test accuracy \(%\) of original and TriDim\-based encoders under the TUH\-2K pretraining setting, averaged over the three subject\-level splits \(seeds 5/42/43\)\.Under this controlled evaluation, TriDim\-based re\-pretraining yields relative improvements in average downstream accuracy of11\.91%11\.91\\%,3\.05%3\.05\\%, and7\.35%7\.35\\%over the original REVE, CBraMod, and CSBrain encoders, respectively\. For each encoder, the relative improvement is computed from its mean accuracy across the eight datasets; averaging the three encoder\-level relative improvements gives7\.4%7\.4\\%\. This is not an average of per\-dataset relative improvements\. Parameter counts are reduced by32\.0%32\.0\\%,17\.0%17\.0\\%, and47\.3%47\.3\\%—matching the values reported in the main text\. Gains are consistent across task categories for REVE and CSBrain; the CBraMod variant trades small decreases on FACED and SleepEDF for gains on the remaining six datasets\.
## Appendix GComplete Hyperparameter Configurations
Supplementary Tables[5](https://arxiv.org/html/2609.19842#A7.T5)and[6](https://arxiv.org/html/2609.19842#A7.T6)list the exact final hyperparameters used for the Full TriDimEEG model on every dataset, taken verbatim from the released per\-dataset YAML configuration files\. Dataset\-specific values were selected exclusively according to validation accuracy during development; all ablation variants reuse the same per\-dataset configuration with only the structural change described in the following section\. The multi\-level tri\-axis readout is enabled in all main experiments\.
Table 5:Architecture hyperparameters of Full TriDimEEG per dataset\.C/S/LC/S/L: latent axis sizes \(basis dims\);DD: patch embedding width;NN: number of TriDim blocks;pap\_\{a\}: maximum DropPath rates of the attention views and the FFN branch \(pc/ps/pl/pffnp\_\{c\}/p\_\{s\}/p\_\{l\}/p\_\{\\mathrm\{ffn\}\}\)\. The last column states how the DropPath rate is applied across theNNblocks:*constant*holds the rate atpap\_\{a\}in every block, while*linear0→pa0\{\\to\}p\_\{a\}*ramps it linearly from00topap\_\{a\}across blocks\.
Table 6:Training hyperparameters of Full TriDimEEG per dataset\. All runs use AdamW, cosine learning\-rate decay with a 5\-epoch linear warm\-up, gradient clipping at norm 1\.0, early stopping on validation accuracy, and checkpoint selection by validation accuracy\.
## Appendix HAblation Variant Definitions
All ablation variants share the Full model’s frontend \(InstanceTimeNorm, patch unfolding, three basis projections\), the multi\-level tri\-axis readout, the classifier head, and every training hyperparameter; only the indicated structural element differs\. Parameter counts below are measured under the SEED configuration \(Full = 3,251,600\); exact counts vary slightly with per\-dataset configurations\.
- •w/o channel axis \(CC\) / w/o short\-term axis \(SS\) / w/o long\-term axis \(LL\)\.The indicated axis is removed from the block: its cross\-axis attention branch, its axis\-specific FFN, and the associated normalizations and DropPath are dropped, and the learned softmax view fusion is renormalized over the two surviving axis views\.
- •Channel axis only \(CC\) / short\-term axis only \(SS\) / long\-term axis only \(LL\)\.Only the indicated axis is retained: the attention branch and the axis\-specific FFN of the kept axis are unchanged, the corresponding branches of the other two axes are removed, and the view fusion degenerates to the single surviving view\.
- •w/o multi\-level readout\.The multi\-level tri\-axis readout is disabled \(use\_multi\_level\_readout=False\); the per\-level readout projections and the layer\-fusion weights are removed, and the single tri\-axis attention\-pooling head is applied to the final block output only\.
- •w/o cross\-axis attention\.The entire cross\-axis attention sublayer is removed \(the three shared axis\-attention modules, per\-axis attention normalizations, attention view\-fusion weights, attention LayerScale, and the per\-branch DropPath\), leaving a purely feed\-forward tri\-axis mixer:x←x\+γffn⊙FFN\(x\)x\\leftarrow x\+\\gamma\_\{\\mathrm\{ffn\}\}\\odot\\mathrm\{FFN\}\(x\)\.−137,865\-137\{,\}865parameters\.
- •Shared FFN\.The three axis\-specific FFNs are replaced by one shared two\-layer MLP \(same expansion ratio 2\)\. Each axis view is transposed so the target axis becomes the last dimension, right\-zero\-padded up to the channel\-axis width, passed through the shared MLP, and sliced back; padding only ever widens theSS/LLviews, so no view is truncated\.−25,152\-25\{,\}152parameters\.
- •Independent attention\.The three parameter\-shared attention modules are replaced by six independent modules, one per \(embedding axis, sequence axis\) pair; every other component is unchanged\.\+137,088\+137\{,\}088parameters\.
- •SequentialC→S→LC\{\\rightarrow\}S\{\\rightarrow\}L\.The three axis views are applied sequentially in the fixed orderC→S→LC\\rightarrow S\\rightarrow Linstead of in parallel with learned softmax fusion; each sub\-update reads the tensor updated by the previous one\. The view\-fusion weights are removed \(−18\-18scalar parameters\)\. During training, one of the three attention sub\-updates is randomly skipped per batch with the configured probability and the surviving two are scaled by1\.51\.5, keeping the expected update identical to the parallel design\.
## Appendix IStatistical Analysis
#### Variation across splits\.
Every entry in Tables[2](https://arxiv.org/html/2609.19842#Sx4.T2)and[3](https://arxiv.org/html/2609.19842#Sx4.T3)of the main text is the mean±\\pmsample standard deviation over the three independent subject\-level splits \(seeds 5/42/43\), rather than a single run; the corresponding macro\-F1 scores are listed in Supplementary Table[1](https://arxiv.org/html/2609.19842#A3.T1), and the underlying per\-split values are included in the released result files\.
#### Cross\-model omnibus test\.
A Friedman test across the 15 compared models and the eight datasets \(mean accuracy per dataset as the observation unit\) rejects the null hypothesis of equal model performance \(χ2\(14\)=55\.84\\chi^\{2\}\(14\)=55\.84,p<0\.001p<0\.001\), confirming that the per\-dataset ranking differences are systematic rather than random\.
#### Cross\-dataset pairwise tests\.
We compare TriDim with the strongest baselines using a two\-sided Wilcoxon signed\-rank test paired by dataset \(n=8n=8, exact\)\. TriDim wins 5/8 datasets against the second\-best model EEGDeformer \(W=11W=11,p=0\.383p=0\.383\), 6/8 against REVE \(W=6W=6,p=0\.109p=0\.109\), and 7/8 against CBraMod \(W=5W=5,p=0\.078p=0\.078\)\. With only eight paired datasets the test has limited power \(the smallest attainable two\-sidedppis0\.00780\.0078, and only for an 8/8 win\), so we interpret the aggregate performance improvements \(4\.3%4\.3\\%relative improvement in average accuracy over EEGDeformer, together with the better average rank\) as the primary evidence, and report the test outcomes for completeness rather than as the basis of our claims\.
#### Split\-level tests for ablations\.
For the ablation study we exploit the fact that every variant is evaluated on the*same*frozen split manifests as the Full model: we pair per\-split test accuracies by \(dataset, split\), givingn=8×3=24n=8\\times 3=24paired observations per variant, and apply a two\-sided Wilcoxon signed\-rank test \(normal approximation with continuity correction\)\. Supplementary Table[7](https://arxiv.org/html/2609.19842#A9.T7)reports the mean paired difference \(Full−\-variant\) and the test outcome\. Removing any single axis does not significantly degrade performance \(p=0\.089p=0\.089forCC, a marginal trend;p=0\.70p=0\.70forSS;p=0\.62p=0\.62forLL\), whereas restricting the model to a single axis does: theSS\-only andLL\-only variants are significantly worse than Full \(p=0\.012p=0\.012andp=0\.015p=0\.015\), withCC\-only a marginal trend \(p=0\.10p=0\.10\)\. The structural variants \(multi\-level readout, cross\-axis attention, shared FFN, sequential processing\) each reduce average accuracy by approximately0\.70\.7–1\.2%1\.2\\%relative to Full without reaching significance, and independent \(unshared\) attention performs essentially on par with Full, indicating that parameter sharing is performance\-neutral\. We therefore describe the tri\-axis design as improving cross\-task robustness—the Full model is the most consistent variant across datasets rather than the per\-dataset winner—and avoid claiming per\-variant significant differences where the paired test does not support them\.
Table 7:Split\-level paired Wilcoxon signed\-rank tests, Full TriDimEEG versus each ablation variant, paired by \(dataset, split\) withn=24n=24\. MeanΔ\\Deltais the mean paired accuracy difference \(Full−\-variant\) in percentage points; positive values favor Full\. MeanΔ\\Deltais computed from unrounded per\-split values and may therefore differ by up to0\.010\.01pp from the difference of the rounded table entries\.
## Appendix JSensitivity Analysis Methodology
#### Perturbation analysis\.
The axis\-aligned perturbation study shown in Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)uses nine seed\-42 LOSO models trained specifically for the interpretation analysis with 500\-ms patches and a 50\-ms stride\. These models use the same LOSO subject partitions as the protocol analysis above, but differ from the 200\-ms\-patch BCI\-IV\-2A models used in the main benchmark and Supplementary Table[2](https://arxiv.org/html/2609.19842#A4.T2)\. Each subject is scored by a model that never saw that subject during training or checkpoint selection\. Perturbations are applied after patchification to the patched representation rather than directly to the raw EEG signal\. Three perturbation families are evaluated class\-conditionally: \(i\)*electrode occlusion*, zeroing one of the 22 channels across all patch positions; \(ii\)*frequency\-band occlusion*, removing one 2\-Hz spectral component within each 500\-ms local patch; and \(iii\)*temporal patch occlusion*, zeroing one complete 500\-ms patch while leaving the overlapping neighboring patches unchanged\. With a 50\-ms stride over each 4\-s trial, the temporal analysis contains 71 patch positions\. We record the signed drop in true\-class probability \(in percentage points\) relative to the unperturbed input\. For each condition, the subject\-level mean drops are pooled across the nine held\-out subjects, and 95% confidence intervals are obtained from a 10,000\-draw bootstrap over subjects \(n=9n=9\)\. The complete per\-condition summary tables are included in the released result files\.
#### Alignment with the main configuration\.
The interpretation analysis deliberately uses a 500\-ms patch length with a 50\-ms stride, whereas the main BCI\-IV\-2A benchmark uses 200\-ms patches with the same 50\-ms stride\. The 500\-ms configuration provides the patch representation used for all three perturbation analyses in Figure[4](https://arxiv.org/html/2609.19842#Sx4.F4)and yields 71 temporal occlusion positions over each 4\-s trial\.
#### Shuffled\-label controls\.
As a negative control, we retrain the same nine LOSO models with the*training*labels randomly permuted \(fixed permutation, seed 42; validation and test labels untouched\) under the identical configuration\. These models perform at chance on the held\-out subjects \(mean test accuracy26\.1%26\.1\\%over the nine folds, versus25%25\\%chance\), confirming that no task\-relevant structure is learned\. Their perturbation profile collapses: the mean absolute true\-class\-probability drop shrinks from5\.715\.71to0\.710\.71pp for electrode occlusion, from0\.410\.41to0\.140\.14pp for frequency\-band occlusion, and from0\.560\.56to0\.210\.21pp for patch occlusion, and the largest single\-condition effect falls from22\.522\.5pp to at most2\.52\.5pp\. The class\-conditional spatial, spectral, and temporal structure reported in the main text is therefore absent under shuffled labels and cannot be attributed to the perturbation procedure itself\.
## Appendix KComputing Infrastructure and Software
Experiments were conducted on two Linux GPU clusters\. The main\-table results were produced primarily on an internal cluster equipped with Tesla V100 \(32 GB\) GPUs \(seven of the eight datasets\), using Python 3\.10, PyTorch 2\.4\.1 \(CUDA 12\.1, cuDNN 9\.1\), NumPy 2\.1\.2, and Zarr 2\.18\.3\. The SleepEDF main\-table runs, all block\-ablation runs, and the release validation were executed on Volcano Engine ML Platform nodes equipped with 4×\\timesNVIDIA H20 \(96 GB\) GPUs per node, with one training process pinned to each GPU, using Python 3\.11, PyTorch 2\.8\.0 \(CUDA 12\.8\), and Zarr 2\.18\.7\. Per\-run logs record the exact environment string \(torch/cuda/zarrversions\), the resolved data root, the split manifest SHA\-256, and a configuration check line \(multi\_level\_readout=True, selection metric\) for every training run\.
## Appendix LIndependent Reproduction Check
Before submission, the released code and configurations were re\-executed from scratch on a different hardware and software stack than the one used for the primary results \(H20 nodes with PyTorch 2\.8\.0/CUDA 12\.8, versus V100 with PyTorch 2\.4\.1/CUDA 12\.1\)\. Three checks were performed\. First, the released Full model builds to exactly 3,251,600 parameters under the SEED configuration, identical to the original training code, and the Full model together with all structurally distinct variant classes defined in the*Ablation Variant Definitions*section pass a forward/backward smoke test with gradients flowing to every parameter\. Second, an end\-to\-end re\-run of BCIC2A and SHU \(three seeds each, full training protocol\) reproduces the main\-table mean accuracies within the original three\-seed standard\-deviation bands:54\.854\.8versus56\.6±2\.7156\.6\\pm 2\.71on BCIC2A, and61\.061\.0versus60\.2±7\.1660\.2\\pm 7\.16on SHU\. Third, every reported number is machine\-audited against the per\-run result records \(release\_audit/traceability\_audit\.py, deviations below10−610^\{\-6\}\)\. Finally, the same cleaned release was re\-executed on the*original*V100/PyTorch 2\.4\.1 stack for BCIC2A \(three seeds, full protocol\): the per\-split test accuracies \(53\.553\.5/58\.358\.3/58\.058\.0, mean56\.656\.6\) match the original result records to the reported precision, confirming that the release reproduces the primary results under the original environment and that the cross\-stack deviations described below are environmental\.
Training is not run in deterministic mode\. Across heterogeneous GPU architectures and library versions, per\-seed accuracies on small datasets can vary across runs, because validation\-based early stopping amplifies kernel\-level nondeterminism into different selected checkpoints; the three\-seed mean and its spread are stable\. This is why all results in this paper are reported as mean±\\pmsample standard deviation over three seeds rather than per\-seed point values\.
## Appendix MCode Availability
The code repository for TriDimEEG is[https://github\.com/ncclab\-sustech/TriDim\_model](https://github.com/ncclab-sustech/TriDim_model)\. The release artifacts described in this supplement comprise the source code, per\-dataset YAML configurations, frozen split manifests with SHA\-256 checksums, per\-seed result records and training logs, and a release\-validation script \(scripts/validate\_release\.py\)\. Repository contents and licensing terms should be consulted before reuse\. The validation script checks the dataset registry, configuration consistency, immutable manifests, and result\-file completeness of the release\.
## Appendix NAdditional Architectural Controls
Beyond the ablation variants reported in the main text, we evaluate two additional architectural controls that isolate the contribution of the tri\-axis factorization itself\. Both controls reuse the frozen subject\-level split manifests, the per\-dataset configurations of Supplementary Tables[5](https://arxiv.org/html/2609.19842#A7.T5)and[6](https://arxiv.org/html/2609.19842#A7.T6), and the identical training protocol \(seeds 5/42/43, validation\-based checkpoint selection\)\.
#### Criss\-cross control\.
This control fixes the short\-term axisSSas the embedding dimension and applies only two attention operators,Attnc\|s\\mathrm\{Attn\}\_\{c\|s\}over the channel axis andAttnl\|s\\mathrm\{Attn\}\_\{l\|s\}over the long\-term axis, both withSSas the embedding axis\. The two operators share parameters, and their outputs are combined by a fixed0\.50\.5average in a direct residual connection—there is no axis\-role rotation and no learned attention view fusion\. The FFN path \(three axis\-specific MLPs with learned softmax fusion\) is identical to the Full model\. The control uses3,121,8003\{,\}121\{,\}800parameters versus3,246,6093\{,\}246\{,\}609for the Full model under the same configuration \(−3\.84%\-3\.84\\%\), so the comparison is not confounded by capacity\.
Supplementary Table[8](https://arxiv.org/html/2609.19842#A14.T8)reports the per\-dataset results\. The criss\-cross control falls below the Full model in accuracy on five of the eight datasets; the exceptions are SHU\-MI, SleepEDF, and PhysioNet\-MI, with relative improvements of approximately7\.6%7\.6\\%,0\.3%0\.3\\%, and1\.0%1\.0\\%over Full, respectively\. Across all eight datasets, it reduces average accuracy by approximately3\.1%3\.1\\%and average macro\-F1 by4\.0%4\.0\\%relative to Full\. A split\-level paired Wilcoxon signed\-rank test \(n=8×3=24n=8\\times 3=24\) does not reach significance \(accuracyp=0\.143p=0\.143; macro\-F1p=0\.107p=0\.107\), so we do not claim a significant per\-pair degradation; the relevant observation is the consistency of the gap across datasets rather than its size on any single one\.
Table 8:Criss\-cross control versus Full TriDimEEG: test accuracy and macro\-F1 \(%\), reported as mean±\\pmsample standard deviation across the three subject\-level splits \(seeds 5/42/43\)\.
#### Flattened Transformer control\.
This control replaces the tri\-axis encoder with a standard Transformer operating on a flattened token sequence\. Given the projected representation𝐗∈ℝB×C×S×L\\mathbf\{X\}\\in\\mathbb\{R\}^\{B\\times C\\times S\\times L\}, the short\-term and long\-term axes are flattened into a single sequence dimension, yielding𝐗flat∈ℝB×\(S⋅L\)×C\\mathbf\{X\}\_\{\\mathrm\{flat\}\}\\in\\mathbb\{R\}^\{B\\times\(S\\cdot L\)\\times C\}\. Thus, each\(s,l\)\(s,l\)position forms one token, with theCC\-dimensional channel representation serving as its embedding\. The flattened sequence is processed by the same number of standard pre\-norm Transformer layers as TriDimEEG, each consisting of multi\-head self\-attention followed by a two\-layer FFN with expansion ratio 2\. The number of attention heads, dropout rate, and LayerScale initialization are matched to the Full model\.
For compatibility with the unchanged multi\-level tri\-axis readout, the output of each Transformer layer is reshaped back to\[B,C,S,L\]\[B,C,S,L\]before readout\. All components outside the encoder—including the frontend, tri\-axis projections, multi\-level tri\-axis readout, and classifier head—are kept identical to the Full model\. As shown in Supplementary Table[9](https://arxiv.org/html/2609.19842#A14.T9), replacing the structure\-preserving TriDim encoder with this flattened\-token Transformer reduces average accuracy by approximately1\.2%1\.2\\%and average macro\-F1 by3\.1%3\.1\\%relative to the Full model\.
Table 9:Flattened Transformer control versus Full TriDimEEG: test accuracy and macro\-F1 \(%\), reported as mean±\\pmsample standard deviation across the three subject\-level splits \(seeds 5/42/43\)\.
#### An ordered evidence chain\.
Taken together with the structural ablations of the main text, the architecture controls form an ordered evidence chain: Full TriDimEEG \(57\.757\.7\)\>\>sequentialC→S→LC\{\\rightarrow\}S\{\\rightarrow\}L\(57\.157\.1\)\>\>Flattened Transformer \(57\.057\.0\)\>\>criss\-cross \(55\.955\.9\) in average accuracy\. Removing the axis\-role rotation while keeping two attention operators \(criss\-cross\) costs more than removing the tri\-axis factorization while keeping full self\-attention \(Flattened Transformer\), and both fall below the Full tri\-axis block, supporting the view that the structured three\-axis decomposition—not merely the presence of attention—underlies the cross\-task robustness of the Full model\.Similar Articles
Interpretable EEG Microstate Discovery via Variational Deep Embedding: A Systematic Architecture Search with Multi-Quadrant Evaluation
This paper presents Conv-VaDE, a variational deep embedding model for interpretable EEG microstate discovery that jointly learns topographic reconstruction and probabilistic soft clustering. It includes a systematic architecture search evaluated on resting-state EEG data to determine optimal model configurations for stability and interpretability.
MEL: Coordinate-Preserving EEG Tokenization for fMRI Translation
This paper introduces MEL, a coordinate-preserving EEG tokenization framework for translating EEG to fMRI, addressing representation-interface mismatch and improving prediction over baselines through explicit modeling of hemodynamic latency and spectral-spatial dynamics.
Block-Based Double Decoders
Proposes block-based double decoders, a novel transformer architecture using doubly-causal block-based attention masks to combine decoder-only training efficiency with encoder-decoder inference efficiency, achieving strong scaling performance and reduced KV-cache memory.
Enabling Unsupervised Training of Deep EEG Denoisers With Intelligent Partitioning
This paper proposes Intelligent Partitioning for Self-supervised Denoising (iPSD), a method enabling unsupervised training of deep EEG denoisers by partitioning noisy segments without requiring clean reference data.
Mechanistic Interpretability of EEG Foundation Models via Sparse Autoencoders
This paper applies TopK Sparse Autoencoders to three EEG foundation models (SleepFM, REVE, LaBraM) to extract interpretable feature dictionaries and introduces a framework for concept steering, revealing representational failures and clinical entanglements.