braindecode/DIVER-1-1s-small

DIVER-1, 1 s Small (iEEG and EEG)

The released encoder of Han et al. (2025), the joint encoder DIVER-1-1s Small (patches of 1 s at 500 Hz, pretrained on intracranial and scalp recordings; described in versions 1 and 2 of the paper), 50.6M parameters, converted to braindecode.models.DIVER1.

Source: weights/i_eeg_pretrained_weights.pt of DIVER-Project/DIVER-1, sha256 dbfa48289989475a52719b1bcb868e62a82877120ac8272ca7dab772e407b891 (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).

Training data

The data that shaped these weights. The ids are the datasets: field of this card, which braindecode reads into model.card_data.datasets when you call from_pretrained (MOABB class names for MOABB datasets, OpenNeuro / NEMAR accessions, Hub ids otherwise). nm/on ids are datasets on NEMAR (on = NEMAR's copy of the OpenNeuro dataset with the same number). Rows marked private: are data the authors did not release; they are not in datasets:.

datasets: id	Dataset	Stage	Modality
nm000358	AJILE12: Annotated Joints in Long-term Electrocorticography (Peterson et al., 2022; 12 subjects, DANDI 000055)	pretraining	iEEG
private:DIVER-self-collected-iEEG	DIVER self-collected long-term ECoG/sEEG from 25 drug-resistant epilepsy patients (~7 days each, 2 kHz; IRB-approved, multi-site, not released)	pretraining	iEEG
TUEG	Temple University Hospital EEG Corpus	pretraining	EEG
HBN	Healthy Brain Network EEG (HBN-EEG, Shirazi et al., 2024)	pretraining	EEG
NCHSDB	Nationwide Children's Hospital Sleep DataBank (Lee et al., 2022); EEG channels only	pretraining	EEG
on004395	PEERS: Penn Electrophysiology of Encoding and Retrieval Study (Kahana et al., 2023), OpenNeuro ds004395	pretraining	EEG

• joint iEEG+EEG corpus (DIVER IE): 17,718 subjects, 59,613 h; self-collected iEEG with stricter QC (2,844 h); NCHSDB EEG channels only; 16 epochs
• TUEG (the TUH EEG Corpus) contains the recordings of its sub-corpora TUAB, TUEV, TUSZ, TUAR, TUSL and TUEP; an evaluation on one of those is not held out from these weights unless the authors removed it.
• HBN: the source does not say which Healthy Brain Network EEG releases were used. The public releases are on NEMAR as on005505 (R1) to on005512 (R8) and on005514 (R9) to on005516 (R11), plus nm000103 (non-commercial release).
• Sources: arXiv:2512.19097v2 Table 3 caption (p.19); arXiv:2512.19097v2 Table 3 (p.19); arXiv:2512.19097v2 Figure caption p.10; arXiv:2512.19097v2 references.

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-1s-small", chs_info=raw.info["chs"], n_times=1000, n_outputs=2
)
logits = model(torch.randn(8, len(raw.ch_names), 1000))

n_times must be a multiple of 500. The saved geometry is 6 channels and 1000 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_faced.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.

Cite Braindecode

If you use this braindecode port in scientific work, cite the software on Zenodo, doi:10.5281/zenodo.17699192, and the original model and datasets cited above.

@software{braindecode,
  author = {Aristimunha, Bruno and
            Guetschel, Pierre and
            Wimpff, Martin and
            Gemein, Lukas and
            Rommel, Cedric and
            Banville, Hubert and
            Sliwowski, Maciej and
            Wilson, Daniel and
            Brandt, Simon and
            Gnassounou, Théo and
            Paillard, Joseph and
            {Junqueira Lopes}, Bruna and
            Sedlar, Sara and
            Moreau, Thomas and
            Chevallier, Sylvain and
            Gramfort, Alexandre and
            Schirrmeister, Robin Tibor},
  title = {Braindecode: toolbox for decoding raw electrophysiological brain data
           with deep learning models},
  url = {https://github.com/braindecode/braindecode},
  doi = {10.5281/zenodo.17699192},
  publisher = {Zenodo},
  license = {BSD-3-Clause},
}