/ Модель

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

Лицензия
MIT
Библиотека
braindecode
Опубликована
25 сен 2026

DIVER-1-0.1s-tiny

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

Компактный энкодер DIVER-1 извлекает признаки из внутричерепных и скальповых нейрофизиологических сигналов для последующего анализа и дообучения.

520скачиваний за 30 дней0лайковвеса обновлены29 сен 2026

Бенчмарки

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

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

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

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

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

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

DIVER-1-0.1s-tiny — это компактный энкодер для внутричерепных нейрофизиологических сигналов, опубликованный в библиотеке Braindecode на основе весов проекта DIVER-1 авторов Han и соавторов (2025). В модели 12,7 млн параметров; она разбивает сигнал на патчи длительностью 0,1 секунды и была предварительно обучена на внутричерепленных записях. Репозиторий содержит веса энкодера в формате PyTorch Safetensors, преобразованные для `braindecode.models.DIVER1`.

Модель извлекает признаки из сигналов iEEG, включая SEEG, ECoG и DBS; поддерживаются также каналы скальповой EEG. Для входа сигнал нужно привести к частоте 500 Гц и передать положения электродов в `chs_info` в метрах. Эмбеддинги можно использовать как представления сигналов для последующей классификации и других задач анализа нейрофизиологических записей.

Веса получены преобразованием официального чекпойнта: bfloat16-тензоры приведены к float32, имена параметров адаптированы, а токен маскирования и головы реконструкции удалены. В поставке нет классификационной головы, поэтому для прикладной классификации требуется дообучение. Проверено совпадение признаков энкодера с официальной моделью на CPU в float32; это не подтверждает качество downstream-задач, работу GPU или смешанную точность. Лицензия — MIT.

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

  • извлечение признаков из записей SEEG
  • получение эмбеддингов для ECoG
  • анализ сигналов DBS
  • подготовка представлений EEG для классификации
  • дообучение нейропроб для анализа нейрофизиологических записей

Ограничения

Веса содержат только энкодер без классификационной головы, поэтому для классификации требуется дообучение. Проверка совпадения признаков на CPU в float32 не подтверждает точность downstream-задач, работу на GPU или смешанную точность.

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

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

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

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

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

Открыть на Hugging Face

DIVER-1, 0.1 s Tiny (iEEG)

The released encoder of Han et al. (2025), the iEEG encoder DIVER-1-0.1s Tiny (patches of 0.1 s at 500 Hz, pretrained on intracranial recordings), 12.7M parameters, converted to braindecode.models.DIVER1.

Source: weights/ieeg_pretrained_weights.pt of DIVER-Project/DIVER-1, sha256 b812093779fb5cf1b76e18f8307df4136b67b008b6cd94c6be75fa286a87864c (the GitHub LFS object is no longer served; the same file is in the authors' Google Drive folder). Conversion casts the bfloat16 tensors to float32, renames them and drops the pretraining-only mask token and reconstruction heads; convert_diver1_weights.py in this repository reproduces it. Encoder features match the official model to 0.0e+00 in float32 on CPU, for intracranial and scalp channels, with muP attention scaling (mup_attention=True).

Usage

Resample to 500 Hz and pass the electrode positions in chs_info (metres, as in MNE). The channel kinds set the modality: SEEG, ECoG and DBS are intracranial, EEG is scalp.

import torch
from braindecode.models import DIVER1

model = DIVER1.from_pretrained(
    "braindecode/DIVER-1-0.1s-tiny", chs_info=raw.info["chs"], n_times=500, n_outputs=2
)
logits = model(torch.randn(8, len(raw.ch_names), 500))

n_times must be a multiple of 50. The saved geometry is 6 channels and 500 samples; the encoder does not depend on it, so pass your own montage, n_times and n_outputs when loading.

Limitations

These files hold the encoder only; the release has no classification head. Braindecode initializes the head on load, so it needs fine-tuning, as in the paper's protocol (scripts/finetune_neuroprobe.sh). The check above covers float32 CPU encoder features, not downstream accuracy, GPU kernels or mixed precision.

Citation

@article{han2025diver,
    title={DIVER-1: Scaling intracranial EEG foundation models for transferable representations},
    author={Han, Danny Dongyeop and Gwon, Yonghyeon and Lee, Ahhyun Lucy and Lee, Taeyang and Lee, Seong Jin and Choi, Jubin and Lee, Sebin and Bang, Jihyun and Lee, Seungju and Park, David Keetae and Yoo, Shinjae and Chung, Chun Kee and Cha, Jiook},
    journal={arXiv preprint arXiv:2512.19097},
    year={2025}
}

License

The weights are released by the DIVER Project under the MIT licence (LICENSE). Braindecode's code is BSD-3-Clause.

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

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

GPU в ailist

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

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

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

Ревизия весов
62ad63d

0 моделей