atlas

Research map / Architectures and efficiency / Attention and transformers

Neural decoding from stereotactic EEG: accounting for electrode variability across subjects

NeurIPS 2024 · Architectures and efficiency › Attention and transformers · approach: Block-sparse attention kernels · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces seegnificant, a transformer that tokenizes sEEG per electrode with 1D convolutions, applies time and electrode self-attention with MNI-coordinate RBF positional encodings, and trains a shared multi-subject model with subject-specific heads to decode trial-wise response time.

Problem

Cross-subject neural decoding from stereotactic EEG is hard because each subject has a variable number of electrodes at distinct, clinically determined brain locations, so there is no correspondence of recorded sites across individuals.

Approach

Tokenize each electrode independently with a shared temporal 1D convolution, inject 3D MNI location via RBF positional encodings, and alternate self-attention over time (within electrodes) and over electrodes (within timepoints) with subject-specific regression heads over a shared trunk.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train jointly on data from all subjects instead of training a separate model per subjectseegnificant · Reaction-time task, per-subject test setR²: Δ R^{2} = 0.09Helped
Finetune the multi-subject model for each subjectseegnificant · Reaction-time task, per-subject test setR²: Δ R^{2} = 0.11Helped
Pretrain on other subjects and finetune on the left-out subject rather than train from scratchseegnificant · Reaction-time task, per-subject test setR²: Δ R^{2} = 0.08Helped
Transfer the pretrained model to a new subject instead of using the multi-subject modelseegnificant · Reaction-time task, per-subject test setR²: Δ R^{2} = -0.01Hurt
Use seegnificant rather than single-subject baseline modelsseegnificant, single-subject models · Reaction-time task, per-subject test setR²: Δ R^{2} ≥ 0.03Helped
Use the multi-subject model rather than single-subject baseline modelsseegnificant, multi-subject model · Reaction-time task, per-subject test setR²: Δ R^{2} ≥ 0.12Helped
Finetune the multi-subject model for each subject rather than use baseline modelsseegnificant, finetuned multi-subject model · Reaction-time task, per-subject test setR²: Δ R^{2} ≥ 0.14Helped
Transfer the pretrained model to a new subject rather than use baseline modelsseegnificant, transferred model · Reaction-time task, per-subject test setR²: Δ R^{2} ≥ 0.11Helped
Add subject-specific regression heads (ablation)seegnificant multi-subject model · Reaction-time task, multi-subject test setR²: Δ R^{2} = 18Helped
Add attention in the space dimension (ablation)seegnificant multi-subject model · Reaction-time task, multi-subject test setR²: Δ R^{2} = 0.10Helped
Replace separate time and space attention with combined 2D attention and remove positional encoding (ablation)seegnificant · Reaction-time task, per-subject test setR²: Δ R^{2} = 0.06Hurt
Replace separate time and space attention with combined 2D attention and remove positional encoding (ablation)seegnificant · Reaction-time task, model trainingTrain time: ∼ 5.5 ×Hurt

Surprising findings

Limitations

Experimental setup

Datasets: in-house sEEG color-change detection task dataset (21 subjects)

Benchmarks: in-house sEEG response time decoding dataset

Metrics: R^2 RMSE

Baselines: Wiener Filter Ridge Regression Lasso Regression XGBoost MLP CNN+MLP 2D attention variant Fourier MNI positional encoding

Builds on: Neural Data Transformer standard transformer block AdamW

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy