/ Модель

Характеристики

Лицензия
BSD 3-Clause
Библиотека
braindecode
Опубликована
16 июн 2026

eegdino-small-pretrained

ЭмбеддингиТекст → embedding3 профессии

EEG-DINO Small извлекает признаки из ЭЭГ для последующей адаптации к задачам классификации мозговых сигналов.

3 447скачиваний за 30 дней0лайковвеса обновлены17 июн 2026

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

EEG-DINO Small — предварительно обученный энкодер для извлечения признаков из электроэнцефалограмм, разработанный в рамках проекта OpenEEG-Bench и основанный на архитектуре braindecode.models.EEGDINO. Модель обучена методом иерархической self-distillation по подходу DINO-v2 на корпусе Temple University Hospital EEG Corpus: более 9000 часов записей с 19 общими каналами системы 10–20, приведёнными к частоте 200 Гц.

Модель предназначена для извлечения EEG-признаков и последующей адаптации к классификации, например для задач на наборах TUEV и TUAB. При загрузке классификационная голова инициализируется заново, поэтому перед классификацией требуется дообучение или применение линейного классификатора. Ожидаются 19 каналов и частота дискретизации 200 Гц; используются веса PyTorch в формате Safetensors. Лицензия репозитория — BSD-3-Clause.

Результаты могут ухудшаться при отличии расположения каналов от обучающего монтажа или при другой частоте дискретизации; входные данные рекомендуется привести к стандартной раскладке каналов и 200 Гц. Для производных весов README также указывает на возможные ограничения лицензии исходного корпуса данных, поэтому условия коммерческого использования следует проверять по лицензии соответствующего обучающего датасета.

Сценарии применения

  • извлечение признаков из многоканальной ЭЭГ
  • подготовка представлений ЭЭГ для классификации
  • дообучение модели для классификации патологических ЭЭГ
  • линейное зондирование признаков ЭЭГ на наборах TUEV или TUAB

Ограничения

Для корректной работы нужны ожидаемые 19 каналов и частота дискретизации 200 Гц; отличия монтажа каналов могут снизить качество. Для классификации требуется дообучить модель или использовать линейный классификатор; для производных весов следует проверить лицензию исходного обучающего корпуса.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

EEG-DINO Small — Self-Distillation EEG Foundation Model

EEG-DINO-Small encoder pretrained with DINO-v2 hierarchical self-distillation (Wang et al., MICCAI 2025).

This is the eegdino-small-pretrained checkpoint for braindecode.models.EEGDINO, curated and re-uploaded as part of the OpenEEG-Bench effort.

Quick start

pip install braindecode[hub]
from braindecode.models import EEGDINO

model = EEGDINO.from_pretrained(
    "braindecode/eegdino-small-pretrained",
    n_outputs=2,            # set to your downstream task
    n_chans=19,
    sfreq=200,
)

from_pretrained reads both the architecture configuration (config.json) and the weights (model.safetensors or pytorch_model.bin) and returns a ready-to-fine-tune nn.Module.

Model details

Architecturebraindecode.models.EEGDINO
Expected channels19
Expected sampling frequency200 Hz
Librarybraindecode ≥ 1.5
Loaded viahuggingface_hub.PyTorchModelHubMixin (free with braindecode[hub])

For the full architecture description, parameter table, and references, see the rendered docstring at https://braindecode.org/stable/generated/braindecode.models.EEGDINO.html or in the interactive Model Explorer Space.

Training data

Temple University Hospital EEG Corpus (TUEG), 19 common 10-20 channels resampled to 200 Hz (>9000 hours), following CBraMod's preprocessing. Pretrained by hierarchical self-distillation.

Intended use

EEG feature extraction or fine-tuning for downstream classification (e.g., TUEV, TUAB). The classification head is re-initialized on load; fine-tune or linear-probe before use.

Limitations

  • Channel layout matters. Performance degrades when the input montage differs from the pretraining montage. Use the Interpolated* variant (where available) or resample channels with MNE before fine-tuning.
  • Sampling rate matters. Resample your data to 200 Hz before inference; the positional / patch embeddings assume this rate.
  • Inherited license restrictions. Downstream weights derived from this checkpoint inherit the license of the original training corpus (some braindecode pretraining corpora are CC-BY-NC). Verify the upstream dataset licence before commercial use.

Citation

If you use this checkpoint, please cite both the original architecture paper and braindecode.

@inproceedings{wang2025eegdino,
  title     = {{EEG-DINO}: Learning {EEG} Foundation Models via Hierarchical
               Self-Distillation},
  author    = {Wang, Xujia and Liu, Xuhui and Liu, Xi and Si, Qian and Xu,
               Zhaoliang and Li, Yang and Zhen, Xiantong},
  booktitle = {Medical Image Computing and Computer Assisted Intervention (MICCAI)},
  year      = {2025},
}

@article{aristimunha2025braindecode,
  title   = {Braindecode: a deep learning library for raw electrophysiological data},
  author  = {Aristimunha, Bruno and others},
  journal = {Zenodo},
  year    = {2025},
  doi     = {10.5281/zenodo.17699192},
}

License

BSD-3-Clause for the model code (matching braindecode). The pretraining data may impose additional restrictions — see Limitations.

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · braindecode

Нельзя запустить

Архитектура ? неизвестна transformers в образе рантайма.

Ревизия весов
d2a879b

0 моделей