atlas

Research map / Architectures and efficiency / Attention and transformers

LUNA: Efficient and Topology-Agnostic Foundation Model for EEG Signal Analysis

NeurIPS 2025 · Architectures and efficiency › Attention and transformers · approach: Block-sparse attention kernels · 2 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.

Proposes LUNA, a self-supervised EEG foundation model that uses learned-query cross-attention to map variable electrode layouts into a fixed latent space, then runs temporal attention on that latent, achieving linear-in-channel cost.

Problem

EEG corpora use heterogeneous electrode montages, so models do not transfer across datasets; existing joint spatio-temporal attention scales quadratically with channel count and patch count.

Approach

Learned queries with cross-attention compress channel features at each patch into Q latent tokens (linear in channel count), then patch-wise temporal transformers operate on the compact latent sequence, with 3D sinusoidal electrode position encodings and a query specialization loss; pre-trained with masked-patch reconstruction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace learned queries with predefined region-based attention (ablation)LUNA-Base · TUABAUROC: 0.887 -> 0.883; -0.004No clear effect
replace learned queries with predefined region-based attention (ablation)LUNA-Base · TUARAUC-PR: 0.495 -> 0.509; ↑ 0.014Mixed
replace learned queries with predefined region-based attention (ablation)LUNA-Base · TUARAUROC: 0.902 -> 0.896; -0.006No clear effect
remove query specialization loss (ablation)LUNA-Base · TUABAUROC: 0.887 -> 0.884; -0.003No clear effect
remove query specialization loss (ablation)LUNA-Base · TUARAUROC: 0.902 -> 0.895; -0.007No clear effect
remove query specialization loss (ablation)LUNA-Base · TUARAUC-PR: 0.495 -> 0.498; ↑ 0.003Mixed
remove frequency features (ablation)LUNA-Base · TUABAUROC: 0.887 -> 0.876; -0.011Hurt
remove frequency features (ablation)LUNA-Base · TUARAUROC: 0.902 -> 0.893; -0.009Hurt
use 2 queries of size 128 instead of 4 queries of size 64 (ablation)LUNA-Base · TUABAUROC: 0.887 -> 0.885Hurt
use 2 queries of size 128 instead of 4 queries of size 64 (ablation)LUNA-Base · TUARAUROC: 0.902 -> 0.885Hurt
use 8 queries of size 32 instead of 4 queries of size 64 (ablation)LUNA-Base · TUABAUROC: 0.887 -> 0.884Hurt
use 8 queries of size 32 instead of 4 queries of size 64 (ablation)LUNA-Base · TUARAUROC: 0.902 -> 0.899Hurt

Surprising findings

Experimental setup

Datasets: TUEG Siena TUAB TUAR TUSL SEED-V

Benchmarks: TUAB TUAR TUSL SEED-V

Metrics: AUROC AUC-PR balanced accuracy Cohen's kappa weighted F1 GFLOPs memory

Baselines: LaBraM CBraMod EEGFormer BIOT FEMBA CEReBrO BrainBERT SPaRCNet

Builds on: Set Transformer Perceiver IO LaBraM CBraMod RoPE

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