SAGE: Surrogate-gradient Adaptation via Attention-Guided Entropy for Spiking Transformers

arXiv:2608.13702 2026 Training 1 ideas extracted · analyzed Sep 1, 2026

What the math gives to ML

The paper provides a lightweight rule for adapting the training-only slope of a spiking neuron's surrogate derivative from transformer attention uncertainty. Its transferable asset is the use of normalized attention entropy as a block-local difficulty signal, combined with a dead zone that prevents noisy updates and a bounded monotone response through tanh. The most direct implementation is to compute one uncertainty statistic per transformer block, map it to a surrogate-gradient temperature or slope, and leave the forward spikes and inference graph unchanged. This is a moderate but concrete idea for improving optimization of spiking transformers, especially when different blocks exhibit substantially different attention concentration.

Ideas from this paper

Unverified 2026

Entropy-Gated Surrogate Slopes

Adapt the slope of each spiking neuron's surrogate derivative using the normalized entropy of its block's attention distribution. High centered entropy uncertainty increases the slope, while low uncertainty decreases it, and a dead zone holds the default slope fixed for ordinary fluctuations. The adaptation exists only in backpropagation, so the forward spike function, parameter count, and inference cost remain unchanged.

Useful5/10
Difficulty4/10
Novelty5/10
Paper: SAGE: Surrogate-gradient Adaptation via Attention-Guided Entropy for Spiking Transformers arXiv:2608.13702