LLM2D
TS-SNN: Temporal Shift 模块用于脉冲神经网络
TS-SNN: Temporal Shift Module for Spiking Neural Networks
作者: Kairong Yu, Tianqing Zhang, Qi Xu, Gang Pan, Hongwei Wang
发布日期: 5/8/2025
arXiv ID: oai:arXiv.org:2505.04165v1

摘要

arXiv:2505.04165v1 宣告类型: cross 摘要: 突触神经网络(SNNs)因其生物可解释性和能源效率而日益受到认可,并被定位为神经形态计算应用中人工神经网络(ANNs)的强有力替代者。SNNs 通过利用尖峰的精确时序来自然处理时间信息,但如何在充分利用时间特征的同时实现低能耗仍然是一个挑战。在这项工作中,我们引入了适用于突触神经网络的时序移位模块(TS-SNN),该模块通过简单的移位操作引入了新颖的时序移位(TS)模块,以便在一个时间步内整合过去、现在和未来的尖峰特征。残差组合方法通过整合移位和原始特征来防止信息丢失。TS 模块非常轻量,仅需要一个可学习的参数,并且可以轻松地以极少的额外计算成本融入现有的架构中。TS-SNN 在基准测试如 CIFAR-10(96.72%)、CIFAR-100(80.28%)和 ImageNet(70.61%)中实现了最先进的性能,同时保持了较低的能耗。这项工作标志着在开发高效且准确的 SNN 架构方面迈出了一大步。