SpikingBrain: Spiking Brain-inspired Large Models
About
Mainstream Transformer-based large language models face major efficiency bottlenecks: training computation scales quadratically with sequence length, and inference memory grows linearly, limiting long-context processing. Building large models on non-NVIDIA platforms also poses challenges for stable and efficient training. To address this, we introduce SpikingBrain, a family of brain-inspired models designed for efficient long-context training and inference. SpikingBrain leverages the MetaX GPU cluster and focuses on three aspects: (1) Model Architecture: linear and hybrid-linear attention architectures with adaptive spiking neurons; (2) Algorithmic Optimizations: an efficient, conversion-based training pipeline and a dedicated spike coding framework; (3) System Engineering: customized training frameworks, operator libraries, and parallelism strategies tailored to MetaX hardware. Using these techniques, we develop two models: SpikingBrain-7B, a linear LLM, and SpikingBrain-76B, a hybrid-linear MoE LLM. These models demonstrate the feasibility of large-scale LLM development on non-NVIDIA platforms, and training remains stable for weeks on hundreds of MetaX GPUs with Model FLOPs Utilization at expected levels. SpikingBrain achieves performance comparable to open-source Transformer baselines while using only about 150B tokens for continual pre-training. Our models also significantly improve long-context efficiency and deliver inference with (partially) constant memory and event-driven spiking behavior. For example, SpikingBrain-7B attains over 100x speedup in Time to First Token for 4M-token sequences. Furthermore, the proposed spiking scheme achieves 69.15 percent sparsity, enabling low-power operation. Overall, this work demonstrates the potential of brain-inspired mechanisms to drive the next generation of efficient and scalable large model design.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instruction Following | IFEval | -- | 836 | |
| Reasoning | BBH | Accuracy53.3 | 726 | |
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy72.97 | 711 | |
| Question Answering | ARC Challenge | Accuracy (ARC)42 | 598 | |
| Commonsense Reasoning | WinoGrande | Accuracy73.48 | 453 | |
| Multi-task Language Understanding | MMLU | MMLU Accuracy73.71 | 442 | |
| Physical Interaction Question Answering | PIQA | Accuracy80.03 | 415 | |
| Mathematical Reasoning | GSM8K | Accuracy (Acc)66.87 | 337 | |
| Multitask Language Understanding | MMLU | Accuracy73.58 | 263 | |
| Question Answering | TriviaQA | Accuracy57.03 | 117 |