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

使用 Group Neurons 的最优 ANN-SNN 转换

Optimal ANN-SNN Conversion with Group Neurons

Liuzhenghao Lv, Wei Fang, Li Yuan, Yonghong Tian

AI总结:

针对 ANN-SNN 转换中的误差与高延迟问题,提出由多个 IF 神经元构成的 Group Neurons 并替换转换后 SNN 的神经元,在极短时间步内达到接近 ANN 的精度。

摘要:

脉冲神经网络(SNN)已成为颇具前景的第三代神经网络,具备二值输出、高稀疏性和生物合理性等独特特性。然而,缺乏有效的学习算法仍是 SNN 面临的挑战。例如,尽管将人工神经网络(ANN)转换为 SNN 可以避免直接训练 SNN,但这种方法会遇到转换误差和较高推理时间延迟的问题。为了在减少推理时间步长的同时降低甚至消除转换误差,我们提出了一种名为 Group Neurons(GN)的新型神经元。一个 GN 由多个 Integrate-and-Fire(IF)神经元作为成员组成,并且其神经动力学经过精心设计。基于 GN,我们优化了传统的 ANN-SNN 转换框架。具体而言,我们将传统转换框架所得 SNN 中的 IF 神经元替换为 GN。由此产生的、使用 GN 的 SNN 即使在极短的推理时间步长内,也能够达到与 ANN 相当的精度水平。在 CIFAR10、CIFAR100 和 ImageNet 数据集上的实验表明,所提方法在推理精度和延迟方面均具有优越性。代码可在 https://github.com/Lyu6PosHao/ANN2SNN_GN 获取。

英文摘要:

Spiking Neural Networks (SNNs) have emerged as a promising third generation of neural networks, offering unique characteristics such as binary outputs, high sparsity, and biological plausibility. However, the lack of effective learning algorithms remains a challenge for SNNs. For instance, while converting artificial neural networks (ANNs) to SNNs circumvents the need for direct training of SNNs, it encounters issues related to conversion errors and high inference time delays. In order to reduce or even eliminate conversion errors while decreasing inference time-steps, we have introduced a novel type of neuron called Group Neurons (GNs). One GN is composed of multiple Integrate-and-Fire (IF) neurons as members, and its neural dynamics are meticulously designed. Based on GNs, we have optimized the traditional ANN-SNN conversion framework. Specifically, we replace the IF neurons in the SNNs obtained by the traditional conversion framework with GNs. The resulting SNNs, which utilize GNs, are capable of achieving accuracy levels comparable to ANNs even within extremely short inference time-steps. The experiments on CIFAR10, CIFAR100, and ImageNet datasets demonstrate the superiority of the proposed methods in terms of both inference accuracy and latency. Code is available at https://github.com/Lyu6PosHao/ANN2SNN_GN.

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

↑