论文标题

通过深尖峰神经网络进行模式识别的逐步串联学习

Progressive Tandem Learning for Pattern Recognition with Deep Spiking Neural Networks

论文作者

Wu, Jibin, Xu, Chenglin, Zhou, Daquan, Li, Haizhou, Tan, Kay Chen

论文摘要

尖峰神经网络(SNN)表现出与传统人工神经网络(ANN)相比,由于其事件驱动的性质和稀疏的沟通,具有低潜伏期和高计算效率。但是,深入SNN的培训并不直接。在本文中,我们提出了一种新颖的ANN-SNN转换和层次的学习框架,以进行快速有效的模式识别,这被称为深入SNN的渐进式串联学习。通过研究离散表示空间中ANN和SNN之间的等效性,引入了一种原始网络转换方法,该方法充分利用了尖峰计数以近似模拟神经元的激活值。为了补偿原始网络转换引起的近似错误,我们进一步引入了一种使用自适应训练调度程序的层次学习方法,以微调网络权重。进行性串联学习框架还允许在培训期间逐步强加硬件限制,例如有限的重量精度和粉丝连接。因此,受过训练的SNN在大规模对象识别,图像重建和语音分离任务上表现出显着的分类和回归功能,同时比其他最先进的SNN实现至少需要减少推理时间和突触操作的数量级。因此,它为电力预算有限的普遍移动设备和嵌入式设备打开了无数的机会。

Spiking neural networks (SNNs) have shown clear advantages over traditional artificial neural networks (ANNs) for low latency and high computational efficiency, due to their event-driven nature and sparse communication. However, the training of deep SNNs is not straightforward. In this paper, we propose a novel ANN-to-SNN conversion and layer-wise learning framework for rapid and efficient pattern recognition, which is referred to as progressive tandem learning of deep SNNs. By studying the equivalence between ANNs and SNNs in the discrete representation space, a primitive network conversion method is introduced that takes full advantage of spike count to approximate the activation value of analog neurons. To compensate for the approximation errors arising from the primitive network conversion, we further introduce a layer-wise learning method with an adaptive training scheduler to fine-tune the network weights. The progressive tandem learning framework also allows hardware constraints, such as limited weight precision and fan-in connections, to be progressively imposed during training. The SNNs thus trained have demonstrated remarkable classification and regression capabilities on large-scale object recognition, image reconstruction, and speech separation tasks, while requiring at least an order of magnitude reduced inference time and synaptic operations than other state-of-the-art SNN implementations. It, therefore, opens up a myriad of opportunities for pervasive mobile and embedded devices with a limited power budget.

扫码加入交流群

加入微信交流群

微信交流群二维码

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