arrow
返回

ESTSformer: Efficient spatio-temporal spiking transformer

delete2025-07-02
delete0
PRE
AI
C
Chengzhuo Lu
H
Huilin Du
W
Wenjie Wei
Q
Qian Sun
Y
Yuchen Wang
D
Dingyi Zeng
W
Wenyu Chen
M
Malu Zhang
Y
Yang Yang
DOI:10.1016/j.neunet.2025.107786delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
仿生脉冲神经网络(SNNs)因其二值化、异步和事件驱动的计算特性而备受关注。这些特性使SNNs成为传统人工神经网络(ANNs)的有力替代方案。随着Transformer在AI领域的扩展,将其与SNNs相结合为高性能和高效模型提供了有前景的路径。该领域的一个关键挑战是如何有效利用SNNs固有的时空优势。然而,现有关于脉冲时空注意力机制的研究存在某些局限性。本文中,我们首先分析了标准时空自注意力(STSA)的缺点,特别是其计算和存储需求会随着时间步的增加而显著增加。为解决此问题,我们提出了高效的时空自注意力(ESTSA)机制。ESTSA将注意力头划分为不同的集合以分别提取时序和空间信息,这种简单而有效的方法显著降低了计算和存储开销。基于ESTSA,我们构建了一个高效的脉冲Transformer架构,称为ESTSformer,该架构修改了编码器模块中的残差连接,以确保网络中的纯粹脉冲驱动计算。在神经形态和静态数据集上的广泛实验表明,我们的方法在性能和效率方面优于现有先进工作。

期刊

Neural Networks 封面图
Neural Networks
IF:
6.3
论文数:
8.2K
被引数:
3.0W

机构

暂无机构信息
引用论文

引用论文

err分享
err收藏
First-Spike-Based Visual Categorization Using Reward-Modulated STDP使用奖励调制STDP的基于第一尖峰的视觉分类
err2018-12-01
err113
errOAAI
errMozafari, Milad; Kheradpisheh, Saeed Reza; Masquelier, Timothee; Nowzari-Dalini, Abbas; Ganjtabesh, Mohammad
err分享
err收藏
Conversion of Continuous-Valued Deep Networks to Efficient Event-Driven Networks for Image Classification
err2017-12-07
err771
errOAAI
errRueckauer, Bodo; Lungu, Iulia-Alexandra; Hu, Yuhuang; Pfeiffer, Michael; Liu, Shih-Chii
err分享
err收藏
Rethinking the performance comparison between SNNS and ANNS
err2020-01-01
err214
PREAI
errDeng, Lei; Wu, Yujie; Hu, Xing; Liang, Ling; Ding, Yufei; Li, Guoqi; Zhao, Guangshe; Li, Peng; Xie, Yuan
err分享
err收藏
Batch normalization-free weight-binarized SNN based on hardware-saving IF neuron
err2023-08-01
err3
PREAI
errQiao, G. C.; Ning, N.; Zuo, Y.; Zhou, P. J.; Sun, M. L.; Hu, S. G.; Yu, Q.; Liu, Y.
err分享
err收藏
err分享
err收藏
学者 查看更多内容