mlsanityData to 14 Oct 2026

Uncertainty-Aware Token Importance Estimation in Spiking Transformers

arXiv 2605.09276 ↗First posted 10 May 2026cs.LG cs.CVcited by 0

On the map: Efficient LLM inference and serving → Token pruning and merging for ViTs

Abstract (arXiv)

Spiking transformers have shown strong potential for neuromorphic vision, yet their token processing across multiple spiking steps still introduces substantial redundancy and inference cost. Existing token reduction methods mainly rely on response based cues, such as activation magnitude, firing statistics, or feature similarity. Although effective, these criteria do not explicitly characterize token importance from the perspective of temporally evolving class evidence. In spiking transformers, token representations are progressively formed across multiple spiking steps rather than determined at a single instant, suggesting that token importance should be evaluated not only by instantaneous responses but also by temporal uncertainty patterns. Our key observation is that tokens exhibit heterogeneous uncertainty trajectories over time, and that their temporally aggregated uncertainty statistics provide an effective cue for distinguishing informative tokens from redundant ones. Motivated by this, we propose Uncert, a training free and plug and play token importance estimation framework for spiking transformers. Specifically, Uncert models token wise class evidence with a Dirichlet distribution and summarizes each token temporal uncertainty using its mean and fluctuation across spiking steps, yielding an uncertainty aware importance score for token reduction during inference. Experiments on both static and neuromorphic benchmarks show that Uncert achieves favorable accuracy and efficiency tradeoffs, with the most consistent gains observed under token pruning. Further analysis reveals a clear empirical connection between temporal uncertainty patterns and token contribution, offering new insights into token dynamics in spiking transformers.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Token pruning and merging for ViTs, one of 12 directions in Efficient LLM inference and serving. Builds on 13 of its 35 references that are in the map. First posted on arXiv 10 May 2026.

Builds on

13 of 35 references.

Closest ancestors · 2

Same direction, newest first

  1. 2026TP-Spikformer: Token Pruned Spiking Transformer28 Feb
  2. 2022Token Merging: Your ViT But Faster17 Oct

Other references · 11

  1. 2021DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsificationcited by 864
  2. 2021FNet: Mixing Tokens with Fourier Transformscited by 347
  3. 2024Spike-driven Transformer V2: Meta Spiking Neural Network Architecture Inspiring the Design of Next-generation Neuromorphic Chipscited by 103
  4. 2022Not All Tokens Are Equal: Human-centric Visual Analysis via Token Clustering Transformercited by 101
  5. 2024Scaling Spike-driven Transformer with Efficient Spike Firing Approximation Trainingcited by 59
  6. 2023Speck: A Smart event-based Vision Sensor with a low latency 327K Neuron Convolutional Neuronal Network Processing Pipelinecited by 52
  7. 2024Spikformer V2: Join the High Accuracy Club on ImageNet with an SNN Ticketcited by 30
  8. 2025Towards High-performance Spiking Transformers from ANN to SNN Conversioncited by 11
  9. 2025Differential Coding for Training-Free ANN-to-SNN Conversioncited by 10
  10. 2023SparseSpikformer: A Co-Design Framework for Token and Weight Pruning in Spiking Transformercited by 7
  11. 2024Faster and Stronger: When ANN-SNN Conversion Meets Parallel Spiking Calculationcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025STEP: A Unified Spiking Transformer Evaluation Platform for Fair and Reproducible Benchmarking
  2. 2026Vision SmolMamba: Spike-Guided Token Pruning for Energy-Efficient Spiking State-Space Vision Models
  3. 2026Rethinking the Tradeoff Between Temporal Encoding and Nonlinear Computation in Spiking Language Models
  4. 2024Trimming Down Large Spiking Vision Transformers via Heterogeneous Quantization Search
  5. 2026Controllable Stochastic Quantization Encoding for Adversarially Robust Spiking Neural Networks
  6. 2026BASC : Behavior-Aligned Quantization and Pruning for Low-Bit Spiking Neural Networks
  7. 2026PTQ4SNN: Membrane-Aware Post-Training Quantization for Spiking Neural Networks
  8. 2026R'enyi Entropy: A New Token Pruning Metric for Vision Transformers