SketchSSM: approximate reads not the state, avoiding compounding compression errors

sehoonkim418 · x · 2026-10-08

The author's principle: keep the state exact and approximate only reads. Once per window, the full state is updated and a compact sketch refreshed in the same kernel; within the window every step reads only the sketch, so no compression error is ever written into the state. In contrast, shrinking the state via pruning or quantization compounds errors each step — even 2–4× less traffic sharply hurts long-reasoning accuracy.

Related event: SketchSSM: Approximate State Reading Speeds Linear Attention Decoding by 2.8x(8 posts)→

Original post →

More from Infra

Infra channel →