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

通过具有精确梯度反向传播的SNN优化下采样提升基于Transformer的脉冲神经网络性能

Enhancing the Performance of Transformer-based Spiking Neural Networks by SNN-optimized Downsampling with Precise Gradient Backpropagation

Peng Cheng Laboratory(鹏城实验室);Harbin Institute of Technology(哈尔滨工业大学);Peking University(北京大学)

Chenlin Zhou, Han Zhang, Zhaokun Zhou, Liutao Yu, Zhengyu Ma, Huihui Zhou, Xiaopeng Fan, Yonghong Tian

AI总结:

针对现有基于Transformer的SNN下采样模块导致的梯度反向传播不精确问题,提出SNN优化下采样方法CML,理论证明并实验验证其在多个数据集上显著提升了模型性能。

摘要:

深度脉冲神经网络(SNNs)因其低功耗、生物合理性和事件驱动特性近年来备受关注。然而,最先进的深度SNN(包括Spikformer和Spikingformer)面临着与梯度反向传播不精确相关的关键挑战。该问题源于这些网络中下采样模块的设计不当,极大地阻碍了整体模型性能。本文提出ConvBN-MaxPooling-LIF(CML),一种具有精确梯度反向传播的SNN优化下采样方法。我们从理论上证明CML能有效克服梯度反向传播的不精确性。此外,我们在ImageNet、CIFAR10、CIFAR100、CIFAR10-DVS、DVS128-Gesture数据集上评估CML,在所有数据集上均展现出最先进性能,且与Spikingformer相比性能显著提升。例如,模型在ImageNet上达77.64%,CIFAR10上96.04%,CIFAR10-DVS上81.4%,在ImageNet上提升+1.79%,CIFAR100上提升+1.16%。

英文摘要:

Deep spiking neural networks (SNNs) have drawn much attention in recent years because of their low power consumption, biological rationality and event-driven property. However, state-of-the-art deep SNNs (including Spikformer and Spikingformer) suffer from a critical challenge related to the imprecise gradient backpropagation. This problem arises from the improper design of downsampling modules in these networks, and greatly hampering the overall model performance. In this paper, we propose ConvBN-MaxPooling-LIF (CML), an SNN-optimized downsampling with precise gradient backpropagation. We prove that CML can effectively overcome the imprecision of gradient backpropagation from a theoretical perspective. In addition, we evaluate CML on ImageNet, CIFAR10, CIFAR100, CIFAR10-DVS, DVS128-Gesture datasets, and show state-of-the-art performance on all these datasets with significantly enhanced performances compared with Spikingformer. For instance, our model achieves 77.64 $\%$ on ImageNet, 96.04 $\%$ on CIFAR10, 81.4$\%$ on CIFAR10-DVS, with + 1.79$\%$ on ImageNet, +1.16$\%$ on CIFAR100 compared with Spikingformer.

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

↑