English

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

Machine Learning 2026-03-17 v1 Artificial Intelligence Neural and Evolutionary Computing

Abstract

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 KK spike frames before a single convolution call, reducing TT calls to T/KT/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.

Keywords

Cite

@article{arxiv.2603.13810,
  title  = {Collapse or Preserve: Data-Dependent Temporal Aggregation for Spiking Neural Network Acceleration},
  author = {Jiahao Qin},
  journal= {arXiv preprint arXiv:2603.13810},
  year   = {2026}
}

Comments

15 pages, 8 figures, 7 tables. Code available at https://github.com/jhao/mlx-snn

R2 v1 2026-07-01T11:19:48.958Z