Search papers, labs, and topics across Lattice.
This paper introduces SpikON, a novel algorithm-hardware co-design framework that enhances online supervised learning for spiking neural networks (SNNs) by implementing learnable thresholds and scaled weight centralization techniques. The framework achieves significant improvements in training latency and energy consumption, demonstrating reductions of 32.2% and 35.0% respectively, while maintaining accuracy. Additionally, SpikON's dedicated accelerator design yields up to 7.2x training throughput and 26.8x energy efficiency compared to existing edge accelerators, showcasing its potential for scalable deployment in energy-constrained environments.
SpikON slashes training latency and energy use for spiking neural networks while boosting throughput by over 7x compared to conventional edge accelerators.
Spiking neural networks (SNNs) have emerged as a promising paradigm for energy-efficient brain-inspired computing. However, existing online unsupervised SNN learning suffers from low training accuracy and poor scalability. Although current online supervised learning algorithms perform well on large-scale datasets and networks, the non-hardware-friendly operations hinder efficient edge deployment. In this work, we propose SpikON, the first algorithm-hardware co-design framework for efficient and scalable end-to-end online supervised SNN learning. We first propose the learnable threshold through time and scaled weight centralization through time techniques to address the inefficiency of traditional algorithms. Moreover, to reduce latency and energy consumption, we introduce the novel training dataflow and cascade computation reuse scheme for SNNs that allows concurrent forward-backward computation and temporal reuse across timesteps. We further design the dedicated SNN accelerator with a dual-parallel engine and customized SIMD-based SNN core for efficient end-to-end online learning. Experiments show that the SpikON algorithm achieves 32.2% and 35.0% reductions in training latency and energy consumption over the baseline, without sacrificing accuracy. Moreover, the SpikON co-design achieves 7.2x (11.5x) and 26.8x (15.8x) training throughput (energy efficiency) compared with the edge Apple M4 GPU and TPU-like accelerator, respectively. The code is available at https://github.com/peilin-chen/SpikON.
