High-speed, low-power, and configurable on-chip training acceleration platform for spiking neural networks

Xu, Yujie,Ye, Wujian,Cui, Youfeng,Zhang, Boning,Lin, Wenjie
DOI: https://doi.org/10.1007/s10489-024-05689-3
IF: 5.3
2024-07-20
Applied Intelligence
Abstract:Spike timing-dependent plasticity (STDP) is crucial for training neural networks (SNNs), offering a hardware-compatible and energy-efficient alternative to backpropagation. Current STDP hardware platforms encounter significant challenges, such as slowness, high energy consumption, and limited configurability. To overcome these issues, this paper presents a high-performance SNN training platform. A parallel multi-ring first-in-first-out structure with event-driven processing for spike handling is proposed, which enhances training efficiency, and the flexible pre- and post-synaptic parallelism enhances speed and flexibility. A dataflow strategy that considers spike sparsity unifies spike representations by updating weights only upon spike arrival, thereby promoting logical symmetry and enabling parallelization. Additionally, three encoding strategies, including a hybrid encoding, are implemented to address diverse scenarios. Leveraging Xilinx field-programmable gate array and Jetson Xavier NX, the proposed platform achieved remarkable performance gains. On the MNIST dataset, the platform demonstrated a 22.51 speedup, 2.13% accuracy boost, and 14.79 reduction in energy consumption. On the Fashion-MNIST dataset, it improved accuracy by 10.74% and the training speed by 1.89 . Thus, this platform can significantly advance the SNN training efficiency and performance.
computer science, artificial intelligence
What problem does this paper attempt to address?