mlsanityData to 7 Oct 2026

Collapse or Preserve: Data-Dependent Temporal Aggregation for Spiking Neural Network Acceleration

arXiv 2603.13810 ↗First posted 14 Mar 2026cs.LG cs.AI cs.NEcited by 0

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

  1. 2019Surrogate Gradient Learning in Spiking Neural Networkscited by 604
  2. 2017Spatio-Temporal Backpropagation for Training High-performance Spiking Neural Networkscited by 399
  3. 2020Incorporating Learnable Membrane Time Constant to Enhance Learning of Spiking Neural Networkscited by 340
  4. 2020Going Deeper With Directly-Trained Larger Spiking Neural Networkscited by 268
  5. 2021Training Spiking Neural Networks Using Lessons From Deep Learningcited by 251
  6. 2022Temporal Efficient Training of Spiking Neural Network via Gradient Re-weightingcited by 192
  7. 2023SpikingJelly: An open-source machine learning infrastructure platform for spike-based intelligencecited by 163
  8. 2020The Hardware Lotterycited by 160
  9. 2021Temporal-wise Attention Spiking Neural Networks for Event Streams Classificationcited by 106
  10. 2020Revisiting Batch Normalization for Training Low-latency Deep Spiking Neural Networks from Scratchcited by 76
  11. 2021Pruning of Deep Spiking Neural Networks through Gradient Rewiringcited by 41
  12. 2022TCJA-SNN: Temporal-Channel Joint Attention for Spiking Neural Networkscited by 33
1 more
  1. 2024Mamba-Spike: Enhancing the Mamba Architecture with a Spiking Front-End for Efficient Temporal Data Processingcited by 11

Built on by

Not cited yet.