arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.09403cs.AIcs.CV

通过CRNN-SNN转换在序列学习中获得最优脉冲神经网络

Obtaining Optimal Spiking Neural Network in Sequence Learning via CRNN-SNN Conversion

Shandong Normal University(山东师范大学);School of Information Science and Engineering, Shandong Normal University(山东师范大学信息科学与工程学院);Shanghai Jiao Tong University(上海交通大学);School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University(上海交通大学电子信息与电气工程学院)

Jiahao Su, Kang You, Zekai Xu, Weizhi Xu, Zhezhi He

AI总结:

针对SNN在长序列学习中难以收敛的问题,提出通过量化CRNN直接映射参数的零误差无损转换框架,在多个序列任务上达到最优性能。

摘要:

脉冲神经网络(SNNs)因其丰富的神经动力学和节能神经形态芯片的实现,正成为传统人工神经网络(ANNs)的有前景的替代方案。然而,不可微分的二值通信机制使得SNN难以达到ANN级别的准确率。当SNN遇到序列学习时,由于对长程依赖建模的困难,情况变得更糟。为了克服这些困难,研究人员开发了LIF神经元的变体和不同的替代梯度,但当序列变长(例如,$>$500)时仍未能获得良好结果。与他们不同,我们通过直接从量化CRNN映射参数,在序列学习中获得了一个最优的SNN。我们设计了两个子流水线来支持神经网络中不同结构的端到端转换,分别称为CNN-Morph(CNN $\rightarrow$ QCNN $\rightarrow$ BIFSNN)和RNN-Morph(RNN $\rightarrow$ QRNN $\rightarrow$ RBIFSNN)。使用转换流水线和s-analog编码方法,我们框架的转换误差为零。此外,我们给出了无损CRNN-SNN转换的理论和实验证明。我们的结果表明,与最先进的基于学习和基于转换的方法相比,我们的方法在短和长时间尺度任务上具有有效性。我们在S-MNIST上达到99.16%(0.46 $\uparrow$)的最高准确率,在PS-MNIST(序列长度为784)上达到94.95%(3.95 $\uparrow$),并在碰撞避免数据集中8个时间步内达到0.057(0.013 $\downarrow$)的最低损失。

英文摘要:

Spiking neural networks (SNNs) are becoming a promising alternative to conventional artificial neural networks (ANNs) due to their rich neural dynamics and the implementation of energy-efficient neuromorphic chips. However, the non-differential binary communication mechanism makes SNN hard to converge to an ANN-level accuracy. When SNN encounters sequence learning, the situation becomes worse due to the difficulties in modeling long-range dependencies. To overcome these difficulties, researchers developed variants of LIF neurons and different surrogate gradients but still failed to obtain good results when the sequence became longer (e.g., $>$500). Unlike them, we obtain an optimal SNN in sequence learning by directly mapping parameters from a quantized CRNN. We design two sub-pipelines to support the end-to-end conversion of different structures in neural networks, which is called CNN-Morph (CNN $\rightarrow$ QCNN $\rightarrow$ BIFSNN) and RNN-Morph (RNN $\rightarrow$ QRNN $\rightarrow$ RBIFSNN). Using conversion pipelines and the s-analog encoding method, the conversion error of our framework is zero. Furthermore, we give the theoretical and experimental demonstration of the lossless CRNN-SNN conversion. Our results show the effectiveness of our method over short and long timescales tasks compared with the state-of-the-art learning- and conversion-based methods. We reach the highest accuracy of 99.16% (0.46 $\uparrow$) on S-MNIST, 94.95% (3.95 $\uparrow$) on PS-MNIST (sequence length of 784) respectively, and the lowest loss of 0.057 (0.013 $\downarrow$) within 8 time-steps in collision avoidance dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑