Uni-NTFM: A Unified Foundation Model for EEG Signal Representation Learning
2509.24222v1
eess.SP, cs.AI, cs.LG
2025-10-01
Авторы:
Zhisheng Chen, Yingwei Zhang, Qizhen Lan, Tianyu Liu, Huacan Wang, Yi Ding, Ziyu Jia, Ronghao Chen, Kun Wang, Xinliang Zhou
Резюме на русском
#### Контекст
Исследования по фундаментальным моделям для представления сигналов электроэнцефалограммы (EEG) возникают в свету приближения к "зрелости" в области биомедицинских технологий. Тем не менее, существуют три главных проблемы в существующих моделях: 1) конфликт временных шкал временных рядов и ритмических частотных признаков, 2) невязыка структуры электродов, что приводит к потере структурных данных во время обучения, и 3) ограниченность дальнейшего расширения моделей за счет использования неэффективных архитектур. Эти ограничения мешают получению нейронных сигналов, которые требуются для эффективного использования моделей в медицинских и исследовательских задачах.
#### Метод
Uni-NTFM представляет собой инновационную модель, основанную на основных принципах нейроника. Она разделяет обработку данных на три основные пути: 1) временную структуру, 2) частотные признаки, и 3) непосредственно сигнал в прямом виде. Данные потоки обрабатываются параллельно и потом объединяются с использованием мощной модели Mixture-of-Experts neural Transformer. Модель также включает новую механику топологического вложения, которая использует принципы многоугольников и решеток для генерации унифицированных последовательностей для разных стандартов электродов, чтобы улучшить структурную информацию.
#### Результаты
Uni-NTFM была тренирована на более 28,000 часов EEG-данных с помощью масокнутых методов обучения (masked reconstruction objectives). Эта модель достигла заметного превосходства по сравнению с другими моделями в 9 различных задачах (таких как классификация состояния сознания, обнаружение заболеваний, и др.), применяя обоие методы пробного обучения и полного тренировка. Модель Uni-NTFM$_{large}$ достигла рекордных 1.9 млрд параметров и показала выдающиеся результаты в предсказании различных распределенных паттернов в ЭЭГ.
#### Значимость
Uni-NTFM имеет широкие приложения в области медицинских исследований, техники биологического мониторинга, и компьютерных технологий в области здоровья. Отличительными особенностями являются: 1) эффективность обучения на больших данных, 2) универсальность и интерпретируемость представлений, 3) гибкость в применении и расширении для различных задач. Эта модель может стать основополагающим элементом для дальнейших исследований в нейронауках и биомедицинских технологиях.
#### Выводы
Uni-NTFM достигла существенных улучшений в представлении EEG-сигналов, что демонстрирует её потенциал для решения ключевых проблем в этой области. Будущие исследования будут сфокусированы на улучшении модели для работы с реальными
Abstract
Foundation models pretrained on various and unlabeled data have demonstrated
significant success in natural language and vision, but their application to
electroencephalography (EEG) remains challenged due to the signal's unique
properties. Existing brain foundation models that inherit architectures
designed for text or images lead to three limitations in pre-training: 1)
conflating time-domain waveform patterns with frequency-domain rhythmic
features in a single processing stream, 2) ignoring the critical spatial
topology of electrodes with different standards, and 3) reliance on the
inflexible, dense network to process functionally distinct EEG patterns. To
address these challenges, we introduce the Unified Neural Topological
Foundation Model (Uni-NTFM), which is designed based on neuroscience principles
to produce universal and interpretable representations. Uni-NTFM integrates
three core innovations: 1) a decoupled architecture parallelly encodes time,
frequency, and raw signal representations before performing cross-domain
feature integration; 2) a topological embedding mechanism to unify electrodes
from different international standards and generate structured input sequences
for brain regions; and 3) a Mixture-of-Experts neural Transformer that
efficiently scales model capacity by routing signal patterns to specialized
subnetworks. The largest model, Uni-NTFM$_{large}$, has a record-breaking 1.9B
parameters and was pretrained on over 28,000 hours of diverse EEG data via a
dual-domain masked reconstruction objective. Uni-NTFM significantly outperforms
existing task-specific methods and foundation models across nine distinct
downstream tasks under both linear probing and fine-tuning settings,
demonstrating a superior ability to learn universal representations of brain
activity.
Ссылки и действия
Дополнительные ресурсы: