Optimizing Supervised Learning of Deep Spiking Neural Network towards Memristor Crossbar Implementation.

Qi Chen,Dayou Zhang,Jiawei Fu,Yuhui He
DOI: https://doi.org/10.1109/ASICON58565.2023.10396019
2023-01-01
Abstract:The gradient-based direct training with spike timing coding known as one major approach to supervised learning of deep spiking neural network often encounters problems like failure of convergence and quite low training efficiency. Here we develop a novel training framework by incorporating several algorithm-hardware codesign strategies: first, we optimize the expression of synaptic weight update by rewriting it as a product of the forward signal and backward error, and thus the expression becomes as convenient as that of conventional backpropagation for hardware implementation; second, coerced firing is exerted on those silent neurons to circumvent the problem of gradient exploding; finally, a stochastic computing technique is employed to formulate the write pulse sequence so that the memristor-based synaptic array can be updated in a one-step manner. Our network level simulation results show >90% accuracy on MNIST benchmark while with significantly enhanced training efficiency.
What problem does this paper attempt to address?