论文标题

真实的尖峰:学习尖峰神经网络的实用值的尖峰

Real Spike: Learning Real-valued Spikes for Spiking Neural Networks

论文作者

Guo, Yufei, Zhang, Liwen, Chen, Yuanpei, Tong, Xinyi, Liu, Xiaode, Wang, YingLei, Huang, Xuhui, Ma, Zhe

论文摘要

由于其事件驱动和节能的特征,脑启发的尖峰神经网络(SNN)最近引起了越来越多的关注。在神经形态硬件上存储和计算范式的整合使SNN与深神经网络(DNN)大不相同。在本文中,我们认为SNN可能不会从体重分享机制中受益,这些机制可以有效地降低参数并提高某些硬件中DNN的推理效率,并假设具有未共享卷积内核的SNN可以表现更好。出于这种假设的促进,提出了一种称为真实尖峰的SNN的训练推导方法,它不仅在推理时都享有未共享的卷积内核和二进制尖峰,而且还保持了共享的卷积内核和训练期间的真实卷积内核。 SNN的这种去耦机制是通过重新参数化技术实现的。此外,根据训练推导的想法,提出了一系列在不同级别上实施真实尖峰的不同形式,这些形式也可以在推理中享受共同的卷积,并且对神经形态和非神经局和非神经粒性硬件平台都很友好。提供了理论上的证据,以澄清基于真正的尖峰的SNN网络优于其香草。实验结果表明,所有不同的实际峰值版本都可以始终如一地提高SNN性能。此外,所提出的方法在非加速静态和神经形态数据集上都优于最先进的模型。

Brain-inspired spiking neural networks (SNNs) have recently drawn more and more attention due to their event-driven and energy-efficient characteristics. The integration of storage and computation paradigm on neuromorphic hardwares makes SNNs much different from Deep Neural Networks (DNNs). In this paper, we argue that SNNs may not benefit from the weight-sharing mechanism, which can effectively reduce parameters and improve inference efficiency in DNNs, in some hardwares, and assume that an SNN with unshared convolution kernels could perform better. Motivated by this assumption, a training-inference decoupling method for SNNs named as Real Spike is proposed, which not only enjoys both unshared convolution kernels and binary spikes in inference-time but also maintains both shared convolution kernels and Real-valued Spikes during training. This decoupling mechanism of SNN is realized by a re-parameterization technique. Furthermore, based on the training-inference-decoupled idea, a series of different forms for implementing Real Spike on different levels are presented, which also enjoy shared convolutions in the inference and are friendly to both neuromorphic and non-neuromorphic hardware platforms. A theoretical proof is given to clarify that the Real Spike-based SNN network is superior to its vanilla counterpart. Experimental results show that all different Real Spike versions can consistently improve the SNN performance. Moreover, the proposed method outperforms the state-of-the-art models on both non-spiking static and neuromorphic datasets.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源