Efficient Structure Slimming for Spiking Neural Networks

Yaxin Li,Xuanye Fang,Yuyuan Gao,Dongdong Zhou,Jiangrong Shen,Jian K. Liu,Gang Pan,Qi Xu
DOI: https://doi.org/10.1109/tai.2024.3352533
2024-01-01
IEEE Transactions on Artificial Intelligence
Abstract:Spiking neural networks (SNNs) are deeply inspired by biological neural information systems. Compared to convolutional neural networks (CNNs), SNNs are low power consumption because of their spike based information processing mechanism. However, most of the current structures of SNNs are fully-connected or converted from deep CNNs which poses redundancy connections. While the structure and topology in human brain systems are sparse and efficient. This paper aims at taking full advantage of sparse structure and low power consumption which lie in human brain and proposed efficient structure slimming methods. Inspired by the development of biological neural network structures, this paper designed types of structure slimming methods including neuron pruning and channel pruning. In addition to pruning, this paper also considers the growth and development of the nervous system. Through iterative application of the proposed neural pruning and rewiring algorithms, experimental evaluations on CIFAR-10, CIFAR-100, and DVS-Gesture datasets demonstrate the effectiveness of the structure slimming methods. When the parameter count is reduced to only about 10% of the original, the performance decreases by less than 1%.
What problem does this paper attempt to address?