arXiv:2409.15375cs.NEcs.AI2024-09被引 3

提出新型脉冲注意力机制,让神经形态视觉模型更高效准确

DS2TA: Denoising Spiking Transformer with Attenuated Spatiotemporal Attention

  • 设计脉冲衰减时空注意力,同步捕捉时间和空间的脉冲相关性
  • 4步时序下在CIFAR10达94.92%准确率,10步时在DVS-Gesture达94.44%
  • 无需额外参数,用哈希非线性去噪增强脉冲注意力鲁棒性

视觉变压器(ViT)是当前各类视觉应用的高性能首选模型。近年来,生物启发的脉冲变压器在神经形态硬件上实现了超低功耗运行,但尚未完全释放脉冲神经网络的潜力。本文提出DS2TA——一种针对视觉任务的去噪脉冲变压器,采用衰减时空注意力机制,同时考虑时间和空间上的输入脉冲相关性,从而充分挖掘脉冲神经元在变压器架构中的计算能力。重要的是,DS2TA实现了参数高效的时空注意力计算,且不引入额外权重。通过基于哈希的非线性脉冲注意力去噪器,显著提升了脉冲注意力图的鲁棒性和表达能力。DS2TA在多个广泛使用的静态图像和动态神经形态数据集上表现达到领先水平。在4个时间步长下,于CIFAR10上取得94.92%的top-1准确率,在CIFAR100上为77.47%;在10个时间步长下,于CIFAR10-DVS和DVS-Gesture上分别达到79.1%和94.44%的准确率。

原文摘要 · Abstract (English)

Vision Transformers (ViT) are current high-performance models of choice for various vision applications. Recent developments have given rise to biologically inspired spiking transformers that thrive in ultra-low power operations on neuromorphic hardware, however, without fully unlocking the potential of spiking neural networks. We introduce DS2TA, a Denoising Spiking transformer with attenuated SpatioTemporal Attention, designed specifically for vision applications. DS2TA introduces a new spiking attenuated spatiotemporal attention mechanism that considers input firing correlations occurring in both time and space, thereby fully harnessing the computational power of spiking neurons at the core of the transformer architecture. Importantly, DS2TA facilitates parameter-efficient spatiotemporal attention computation without introducing extra weights. DS2TA employs efficient hashmap-based nonlinear spiking attention denoisers to enhance the robustness and expressive power of spiking attention maps. DS2TA demonstrates state-of-the-art performances on several widely adopted static image and dynamic neuromorphic datasets. Operated over 4 time steps, DS2TA achieves 94.92% top-1 accuracy on CIFAR10 and 77.47% top-1 accuracy on CIFAR100, as well as 79.1% and 94.44% on CIFAR10-DVS and DVS-Gesture using 10 time steps.

脉冲神经网络视觉变压器神经形态计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。