Neural Dynamics Self-Attention for Spiking Transformers

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhang, Dehao, Guo, Fukai, Wang, Shuai, Wang, Jingya, Zhang, Jieyuan, Shan, Yimeng, Zhang, Malu, Yang, Yang, Li, Haizhou
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910060567855104
author Zhang, Dehao
Guo, Fukai
Wang, Shuai
Wang, Jingya
Zhang, Jieyuan
Shan, Yimeng
Zhang, Malu
Yang, Yang
Li, Haizhou
author_facet Zhang, Dehao
Guo, Fukai
Wang, Shuai
Wang, Jingya
Zhang, Jieyuan
Shan, Yimeng
Zhang, Malu
Yang, Yang
Li, Haizhou
contents Integrating Spiking Neural Networks (SNNs) with Transformer architectures offers a promising pathway to balance energy efficiency and performance, particularly for edge vision applications. However, existing Spiking Transformers face two critical challenges: (i) a substantial performance gap compared to their Artificial Neural Networks (ANNs) counterparts and (ii) high memory overhead during inference. Through theoretical analysis, we attribute both limitations to the Spiking Self-Attention (SSA) mechanism: the lack of locality bias and the need to store large attention matrices. Inspired by the localized receptive fields (LRF) and membrane-potential dynamics of biological visual neurons, we propose LRF-Dyn, which uses spiking neurons with localized receptive fields to compute attention while reducing memory requirements. Specifically, we introduce a LRF method into SSA to assign higher weights to neighboring regions, strengthening local modeling and improving performance. Building on this, we approximate the resulting attention computation via charge-fire-reset dynamics, eliminating explicit attention-matrix storage and reducing inference-time memory. Extensive experiments on visual tasks confirm that our method reduces memory overhead while delivering significant performance improvements. These results establish it as a key unit for achieving energy-efficient Spiking Transformers.
format Preprint
id arxiv_https___arxiv_org_abs_2603_19290
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Neural Dynamics Self-Attention for Spiking Transformers
Zhang, Dehao
Guo, Fukai
Wang, Shuai
Wang, Jingya
Zhang, Jieyuan
Shan, Yimeng
Zhang, Malu
Yang, Yang
Li, Haizhou
Neural and Evolutionary Computing
Artificial Intelligence
Integrating Spiking Neural Networks (SNNs) with Transformer architectures offers a promising pathway to balance energy efficiency and performance, particularly for edge vision applications. However, existing Spiking Transformers face two critical challenges: (i) a substantial performance gap compared to their Artificial Neural Networks (ANNs) counterparts and (ii) high memory overhead during inference. Through theoretical analysis, we attribute both limitations to the Spiking Self-Attention (SSA) mechanism: the lack of locality bias and the need to store large attention matrices. Inspired by the localized receptive fields (LRF) and membrane-potential dynamics of biological visual neurons, we propose LRF-Dyn, which uses spiking neurons with localized receptive fields to compute attention while reducing memory requirements. Specifically, we introduce a LRF method into SSA to assign higher weights to neighboring regions, strengthening local modeling and improving performance. Building on this, we approximate the resulting attention computation via charge-fire-reset dynamics, eliminating explicit attention-matrix storage and reducing inference-time memory. Extensive experiments on visual tasks confirm that our method reduces memory overhead while delivering significant performance improvements. These results establish it as a key unit for achieving energy-efficient Spiking Transformers.
title Neural Dynamics Self-Attention for Spiking Transformers
topic Neural and Evolutionary Computing
Artificial Intelligence
url https://arxiv.org/abs/2603.19290