Team Ai
Modelpublic

Catalyst-Neuromorphic/shd-snn-benchmark

sourceHugging Facemitupdated 6mo agoView on Hugging Face
0likes14downloads
Model Card

Catalyst SHD SNN Benchmark

Spiking Neural Network trained on the Spiking Heidelberg Digits (SHD) dataset using surrogate gradient BPTT. Achieves 91.0% on SHD with adaptive LIF neurons (90.8% quantised int16).

Model Description

  • —Architecture (N3): 700 → 1536 (recurrent adLIF) → 20
  • —Architecture (N2): 700 → 512 (recurrent adLIF) → 20
  • —Architecture (N1): 700 → 1024 (recurrent LIF) → 20
  • —Neuron model: Adaptive Leaky Integrate-and-Fire (adLIF) with learnable per-neuron thresholds
  • —Training: Surrogate gradient BPTT, fast-sigmoid surrogate (scale=25), cosine LR scheduling
  • —Hardware target: Catalyst N1/N2/N3 neuromorphic processors

Results

GenerationArchitectureFloat AccuracyParams
N3700→1536→20 (rec, adLIF)91.0%3.47M
N2700→512→20 (rec, adLIF)84.5%759K
N1700→1024→20 (rec, LIF)85.9%1.79M

Reproduce

bash
git clone https://github.com/catalyst-neuromorphic/catalyst-benchmarks.git
cd catalyst-benchmarks
pip install -e .

# N3
python shd/train.py --neuron adlif --hidden 1536 --epochs 200 --device cuda:0 --amp

# N2
python shd/train.py --neuron adlif --hidden 512 --epochs 200 --device cuda:0

# N1
python shd/train.py --neuron lif --hidden 1024 --epochs 200 --device cuda:0

Deploy to Catalyst Hardware

bash
python shd/deploy.py --checkpoint shd_model.pt --threshold-hw 1000

Links

Citation

bibtex
@misc{catalyst-benchmarks-2026,
  author = {Shulayev Barnes, Henry},
  title = {Catalyst Neuromorphic Benchmarks},
  year = {2026},
  url = {https://github.com/catalyst-neuromorphic/catalyst-benchmarks}
}