FSTA-SNN:Frequency-based Spatial-Temporal Attention Module for Spiking Neural Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yu, Kairong, Zhang, Tianqing, Wang, Hongwei, Xu, Qi
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917913213009920
author Yu, Kairong
Zhang, Tianqing
Wang, Hongwei
Xu, Qi
author_facet Yu, Kairong
Zhang, Tianqing
Wang, Hongwei
Xu, Qi
contents Spiking Neural Networks (SNNs) are emerging as a promising alternative to Artificial Neural Networks (ANNs) due to their inherent energy efficiency. Owing to the inherent sparsity in spike generation within SNNs, the in-depth analysis and optimization of intermediate output spikes are often neglected. This oversight significantly restricts the inherent energy efficiency of SNNs and diminishes their advantages in spatiotemporal feature extraction, resulting in a lack of accuracy and unnecessary energy expenditure. In this work, we analyze the inherent spiking characteristics of SNNs from both temporal and spatial perspectives. In terms of spatial analysis, we find that shallow layers tend to focus on learning vertical variations, while deeper layers gradually learn horizontal variations of features. Regarding temporal analysis, we observe that there is not a significant difference in feature learning across different time steps. This suggests that increasing the time steps has limited effect on feature learning. Based on the insights derived from these analyses, we propose a Frequency-based Spatial-Temporal Attention (FSTA) module to enhance feature learning in SNNs. This module aims to improve the feature learning capabilities by suppressing redundant spike features.The experimental results indicate that the introduction of the FSTA module significantly reduces the spike firing rate of SNNs, demonstrating superior performance compared to state-of-the-art baselines across multiple datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2501_14744
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FSTA-SNN:Frequency-based Spatial-Temporal Attention Module for Spiking Neural Networks
Yu, Kairong
Zhang, Tianqing
Wang, Hongwei
Xu, Qi
Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
Machine Learning
Spiking Neural Networks (SNNs) are emerging as a promising alternative to Artificial Neural Networks (ANNs) due to their inherent energy efficiency. Owing to the inherent sparsity in spike generation within SNNs, the in-depth analysis and optimization of intermediate output spikes are often neglected. This oversight significantly restricts the inherent energy efficiency of SNNs and diminishes their advantages in spatiotemporal feature extraction, resulting in a lack of accuracy and unnecessary energy expenditure. In this work, we analyze the inherent spiking characteristics of SNNs from both temporal and spatial perspectives. In terms of spatial analysis, we find that shallow layers tend to focus on learning vertical variations, while deeper layers gradually learn horizontal variations of features. Regarding temporal analysis, we observe that there is not a significant difference in feature learning across different time steps. This suggests that increasing the time steps has limited effect on feature learning. Based on the insights derived from these analyses, we propose a Frequency-based Spatial-Temporal Attention (FSTA) module to enhance feature learning in SNNs. This module aims to improve the feature learning capabilities by suppressing redundant spike features.The experimental results indicate that the introduction of the FSTA module significantly reduces the spike firing rate of SNNs, demonstrating superior performance compared to state-of-the-art baselines across multiple datasets.
title FSTA-SNN:Frequency-based Spatial-Temporal Attention Module for Spiking Neural Networks
topic Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2501.14744