arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2206.09449cs.NEcs.CVcs.LG

SNN2ANN:一种快速且内存高效的脉冲神经网络训练框架

SNN2ANN: A Fast and Memory-Efficient Training Framework for Spiking Neural Networks

Sun Yat-sen University(中山大学)

Jianxiong Tang, Jianhuang Lai, Xiaohua Xie, Lingxiao Yang, Wei-Shi Zheng

AI总结:

提出SNN2ANN框架,通过权重共享和脉冲映射单元在ANN分支上训练SNN,并设计自适应阈值调整算法,实现快速、内存高效的SNN训练,在多个基准数据集上达到与基于脉冲BP相当的精度。

摘要:

脉冲神经网络是低功耗环境下高效的计算模型。基于脉冲的反向传播(BP)算法和ANN-to-SNN(ANN2SNN)转换是SNN训练的成功技术。然而,基于脉冲的BP训练速度慢且需要大量内存开销。尽管ANN2SNN提供了一种低成本的SNN训练方式,但它需要许多推理步骤来模仿训练良好的ANN以获得良好性能。在本文中,我们提出了一种SNN-to-ANN(SNN2ANN)框架,以快速且内存高效的方式训练SNN。SNN2ANN包含两个组件:a)ANN和SNN之间的权重共享架构,以及b)脉冲映射单元。首先,该架构在ANN分支上训练共享权重参数,从而实现SNN的快速训练和低内存成本。其次,脉冲映射单元确保ANN的激活值是脉冲特征。因此,SNN的分类误差可以通过训练ANN分支来优化。此外,我们设计了一种自适应阈值调整(ATA)算法来解决噪声脉冲问题。实验结果表明,我们基于SNN2ANN的模型在基准数据集(CIFAR10、CIFAR100和Tiny-ImageNet)上表现良好。此外,与基于脉冲的BP模型相比,SNN2ANN在0.625倍时间步长、0.377倍训练时间、0.27倍GPU内存成本和0.33倍脉冲活动下能够达到相当的精度。

英文摘要:

Spiking neural networks are efficient computation models for low-power environments. Spike-based BP algorithms and ANN-to-SNN (ANN2SNN) conversions are successful techniques for SNN training. Nevertheless, the spike-base BP training is slow and requires large memory costs. Though ANN2NN provides a low-cost way to train SNNs, it requires many inference steps to mimic the well-trained ANN for good performance. In this paper, we propose a SNN-to-ANN (SNN2ANN) framework to train the SNN in a fast and memory-efficient way. The SNN2ANN consists of 2 components: a) a weight sharing architecture between ANN and SNN and b) spiking mapping units. Firstly, the architecture trains the weight-sharing parameters on the ANN branch, resulting in fast training and low memory costs for SNN. Secondly, the spiking mapping units ensure that the activation values of the ANN are the spiking features. As a result, the classification error of the SNN can be optimized by training the ANN branch. Besides, we design an adaptive threshold adjustment (ATA) algorithm to address the noisy spike problem. Experiment results show that our SNN2ANN-based models perform well on the benchmark datasets (CIFAR10, CIFAR100, and Tiny-ImageNet). Moreover, the SNN2ANN can achieve comparable accuracy under 0.625x time steps, 0.377x training time, 0.27x GPU memory costs, and 0.33x spike activities of the Spike-based BP model.

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

↑