AI RESEARCH

Self-Pruned Key-Value Attention: Learning When to Write by Predicting Future Utility

arXiv CS.LG

ArXi:2605.14037v1 Announce Type: new Under modern test-time compute and agentic paradigms, language models process ever-longer sequences. Efficient text generation with transformer architectures is increasingly constrained by the Key-Value cache memory footprint and bandwidth. To address this limitation, we Rather than enforcing a fixed compression ratio, SP-KV performs dynamic sparsification: the mechanism adapts to the input and typically reduces the KV cache size by a factor of $3$ to $10\times$, longer sequences often being compressible.