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.

Ссылки и действия