In plain words: It scores neurons in a brain-inspired spiking network by how much information they pass on, then cuts the weakest and regrows promising ones, copying the brain's balanced critical state. It beat the best existing pruning method, cutting pruning cost by up to 95.26%.
Abstract · Brain-Inspired Efficient Pruning: Exploiting Criticality in Spiking Neural Networks
Spiking Neural Networks (SNNs) have gained significant attention due to the energy-efficient and multiplication-free characteristics. Despite these advantages, deploying large-scale SNNs on edge hardware is challenging due to limited resource availability. Network pruning offers a viable approach to compress the network scale and reduce hardware resource requirements for model deployment. However, existing SNN pruning methods cause high pruning costs and performance loss because they lack efficiency in processing the sparse spike representation of SNNs. In this paper, inspired by the critical brain hypothesis in neuroscience and the high biological plausibility of SNNs, we explore and leverage criticality to facilitate efficient pruning in deep SNNs. We firstly explain criticality in SNNs from the perspective of maximizing feature information entropy. Second, We propose a low-cost metric for assess neuron criticality in feature transmission and design a pruning-regeneration method that incorporates this criticality into the pruning process. Experimental results demonstrate that our method achieves higher performance than the current state-of-the-art (SOTA) method with up to 95.26\% reduction of pruning cost. The criticality-based regeneration process efficiently selects potential structures and facilitates consistent feature representation.
Shuo Chen, Boxiao Liu, Zeshi Liu, Haihang You
arXiv:2311.16141 · cs.NE, cs.AI, cs.CV, cs.LG · submitted Nov 5, 2023 · updated Nov 21, 2024
abstract · pdf · html
From the paper:
> Spiking Neural Networks (SNNs) have been an attractive option for deployment on devices with limited computing resources and lower power consumption because of the event-driven computing characteristic.
But training them requires new techniques compared to continuous neural networks — SNNs aren't differentiable and therefore you can't back-propagate (as I understand it; please correct me if I'm off here).