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

FSpiNN:一个高效节能的脉冲神经网络优化框架

FSpiNN: An Optimization Framework for Memory- and Energy-Efficient Spiking Neural Networks

Technische Universität Wien (TU Wien)(维也纳工业大学);New York University Abu Dhabi (NYU AD)(纽约大学阿布扎比分校)

Rachmad Vidya Wicaksana Putra, Muhammad Shafique

摘要:

脉冲神经网络(SNN)由于其事件驱动的处理而受到关注,该处理在硬件平台中潜在地消耗低功率/能量计算,同时由于脉冲定时依赖可塑性(STDP)规则而提供无监督学习能力。然而,最先进的SNN需要占用大量内存才能实现高精度,因此很难部署在嵌入式系统上,例如电池供电的移动设备和物联网边缘节点。为此,我们提出了FSpiNN,一个优化框架,用于获得内存和能源效率的SNN的训练和推理处理,具有无监督学习能力,同时保持准确性。它通过以下方式实现:(1)减少神经元和STDP操作的计算需求,(2)提高基于STDP的学习的准确性,(3)通过定点量化压缩SNN,以及(4)在优化过程中纳入内存和能量需求。FSpiNN通过减少神经元操作的数量、基于STDP的突触权重更新和STDP复杂度来降低计算需求。为了提高学习的准确性,FSpiNN采用基于时间步长的突触权重更新,并自适应地确定STDP增强因子和有效抑制强度。实验结果表明,与最先进的工作相比,FSpiNN实现了7.5倍的内存节省,并在MNIST和Fashion MNIST数据集上将训练的能效平均提高了3.5倍,推理的能效平均提高了1.8倍,对于具有4900个兴奋性神经元的网络没有准确性损失,从而为边缘设备/嵌入式系统提供了节能的SNN。

英文摘要:

Spiking Neural Networks (SNNs) are gaining interest due to their event-driven processing which potentially consumes low power/energy computations in hardware platforms, while offering unsupervised learning capability due to the spike-timing-dependent plasticity (STDP) rule. However, state-of-the-art SNNs require a large memory footprint to achieve high accuracy, thereby making them difficult to be deployed on embedded systems, for instance on battery-powered mobile devices and IoT Edge nodes. Towards this, we propose FSpiNN, an optimization framework for obtaining memory- and energy-efficient SNNs for training and inference processing, with unsupervised learning capability while maintaining accuracy. It is achieved by (1) reducing the computational requirements of neuronal and STDP operations, (2) improving the accuracy of STDP-based learning, (3) compressing the SNN through a fixed-point quantization, and (4) incorporating the memory and energy requirements in the optimization process. FSpiNN reduces the computational requirements by reducing the number of neuronal operations, the STDP-based synaptic weight updates, and the STDP complexity. To improve the accuracy of learning, FSpiNN employs timestep-based synaptic weight updates, and adaptively determines the STDP potentiation factor and the effective inhibition strength. The experimental results show that, as compared to the state-of-the-art work, FSpiNN achieves 7.5x memory saving, and improves the energy-efficiency by 3.5x on average for training and by 1.8x on average for inference, across MNIST and Fashion MNIST datasets, with no accuracy loss for a network with 4900 excitatory neurons, thereby enabling energy-efficient SNNs for edge devices/embedded systems.

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

↑