Collapse or Preserve: Data-Dependent Temporal Aggregation for Spiking Neural Network Acceleration
Not on the map. Why?
Abstract (arXiv)
Spike sparsity is widely believed to enable efficient spiking neural network (SNN) inference on GPU hardware. We demonstrate this is an illusion: five distinct sparse computation strategies on Apple M3 Max all fail to outperform dense convolution, because SIMD architectures cannot exploit the fine-grained, unstructured sparsity of i.i.d. binary spikes. Instead, we propose Temporal Aggregated Convolution (TAC), which exploits convolution linearity to pre-aggregate K spike frames before a single convolution call, reducing T calls to T/K. On rate-coded data, TAC achieves 13.8times speedup with +1.6% accuracy on MNIST and +5.4% on Fashion-MNIST -- a simultaneous improvement in both speed and accuracy. However, on event-based data where the temporal dimension carries genuine motion information, TAC's temporal collapse is harmful. We therefore introduce TAC-TP (Temporal Preservation), which shares each group's convolution output across K independent LIF steps, preserving full temporal resolution for downstream layers. On DVS128-Gesture, TAC-TP achieves 95.1% accuracy (vs. 96.3% baseline) with 50% fewer convolution calls, while standard TAC drops to 91.3%. Our key finding is that the optimal temporal aggregation strategy is data-dependent: collapse the temporal dimension for rate-coded data (noise reduction) but preserve it for event data (information retention). Speedup is hardware-agnostic: TAC achieves 11.0times on NVIDIA V100, confirming the mechanism transfers across GPU architectures. All operators in the mlx-snn library are open source.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 13 of its 24 references that are in the map. First posted on arXiv 14 Mar 2026.
Builds on
13 of 24 references.
References · 13
- 2019Surrogate Gradient Learning in Spiking Neural Networkscited by 604
- 2017Spatio-Temporal Backpropagation for Training High-performance Spiking Neural Networkscited by 399
- 2020Incorporating Learnable Membrane Time Constant to Enhance Learning of Spiking Neural Networkscited by 340
- 2020Going Deeper With Directly-Trained Larger Spiking Neural Networkscited by 268
- 2021Training Spiking Neural Networks Using Lessons From Deep Learningcited by 251
- 2022Temporal Efficient Training of Spiking Neural Network via Gradient Re-weightingcited by 192
- 2023SpikingJelly: An open-source machine learning infrastructure platform for spike-based intelligencecited by 163
- 2020The Hardware Lotterycited by 160
- 2021Temporal-wise Attention Spiking Neural Networks for Event Streams Classificationcited by 106
- 2020Revisiting Batch Normalization for Training Low-latency Deep Spiking Neural Networks from Scratchcited by 76
- 2021Pruning of Deep Spiking Neural Networks through Gradient Rewiringcited by 41
- 2022TCJA-SNN: Temporal-Channel Joint Attention for Spiking Neural Networkscited by 33
Built on by
Not cited yet.