Optimising Event-Driven Spiking Neural Network with Regularisation and Cutoff

TL;DR

This paper introduces a cutoff-based approach with regularisation techniques to optimize event-driven spiking neural networks, significantly reducing inference time while maintaining accuracy across various datasets.

Contribution

It proposes novel Top-K cutoff and regularisation methods to enable dynamic inference in SNNs, improving efficiency and reducing latency.

Findings

Achieved 1.76 to 2.76x fewer timesteps on CIFAR-10

Achieved 1.64 to 1.95x fewer timesteps on event-based datasets

Maintained near-zero accuracy loss with improved inference efficiency

Abstract

Spiking neural network (SNN), as the next generation of artificial neural network (ANN), offer a closer mimicry of natural neural networks and hold promise for significant improvements in computational efficiency. However, the current SNN is trained to infer over a fixed duration, overlooking the potential of dynamic inference in SNN. In this paper, we strengthen the marriage between SNN and event-driven processing with a proposal to consider a cutoff in SNN, which can terminate SNN anytime during inference to achieve efficient inference. Two novel optimisation techniques are presented to achieve inference efficient SNN: a Top-K cutoff and a regularisation.The proposed regularisation influences the training process, optimising SNN for the cutoff, while the Top-K cutoff technique optimises the inference phase. We conduct an extensive set of experiments on multiple benchmark frame-based datasets, such asCIFAR10/100, Tiny-ImageNet, and event-based datasets, including CIFAR10-DVS, N-Caltech101 and DVS128 Gesture. The experimental results demonstrate the effectiveness of our techniques in both ANN-to-SNN conversion and direct training, enabling SNNs to require 1.76 to 2.76x fewer timesteps for CIFAR-10, while achieving 1.64 to 1.95x fewer timesteps across all event-based datasets, with near-zero accuracy loss. These findings affirms the compatibility and potential benefits of our techniques in enhancing accuracy and reducing inference latency when integrated with existing methods. Code available: https://github.com/Dengyu-Wu/SNNCutoff