Spiking ST-former: enhancing spatio-temporal modeling in spiking transformers via integrated self-attention mechanisms
摘要
Spiking Neural Networks (SNNs) offer a low-power alternative to traditional Artificial Neural Networks (ANNs). However, their performance often lags behind due to challenges in training and the discrete nature of spike-based signal transmission. In this paper, we propose the Spiking ST-Former, a novel spiking transformer architecture that addresses these limitations by incorporating a Spatio-Temporal Spiking Self-Attention (STSA) mechanism. STSA enables efficient joint modeling of spatial and temporal information through a Temporal Processing Module (TPM), without increasing the number of model parameters. Experimental results on both static and neuromorphic datasets, including CIFAR-10/100, ImageNet, CIFAR10-DVS, and DVS128, demonstrate that STSA consistently outperforms spatial-only attention mechanisms and achieves competitive performance. These findings highlight the potential of STSA in enhancing SNN performance and advancing the development of spiking transformers. Code will be made publicly available at https://github.com/Bennie123-byte/Spiking-ST-former.