BaRISTA, coords scale
The released BaRISTA encoder of Oganesian, Hashemi and Shanechi
(2025) for the coords spatial scale,
converted to
braindecode.models.BaRISTA.
Source: pretrained_models/chans_chans.ckpt at revision
83b27375eba60e9eba9da4e7dd8fb283baace376,
sha256 400eeacc0697004cb81c9ecf754859da184ffeea40afc8ee7b5930c3b997e1d0. Conversion renames tensors and fuses the released gated
projections into the single projection this port uses; convert_barista_weights.py
in this repository reproduces it. Encoder tokens match the released forward
equations to 5.6e-06 in float32 on CPU.
Usage
Supply the montage of each batch as spatial_indices, an (n_chans, 3) tensor of grid coordinates in [0, 200):
import torch
from braindecode.models import BaRISTA
model = BaRISTA.from_pretrained("braindecode/BaRISTA-coords", n_chans=64, n_outputs=2)
logits = model(torch.randn(8, 64, 6144), spatial_indices=torch.randint(0, 200, (64, 3)))
The saved geometry is 64 channels and 6144 samples at 2048 Hz, the
pretraining window. Mean pooling makes the encoder independent of both, so pass
your own n_chans, n_times and n_outputs when loading.
Limitations
These files hold the encoder only, as does the release. Braindecode
initializes the classification head on load, so it needs fine-tuning, as does
the learned read-out of the paper's protocol (pooling="learned"), which was
never released. The check
above covers float32 CPU encoder tokens, not downstream accuracy, GPU kernels or
mixed precision.
Citation
@inproceedings{oganesian2025barista,
title={BaRISTA: Brain Scale Informed Spatiotemporal Representation of Human Intracranial Neural Activity},
author={Oganesian, Lucine L. and Hashemi, Saba and Shanechi, Maryam M.},
booktitle={Advances in Neural Information Processing Systems},
year={2025}
}
License
Copyright (c) 2025 University of Southern California. Educational, research and
non-profit use only; commercial use requires an agreement with the USC Stevens
Center for Innovation. See NOTICE.txt and the original licence.