Q-Delta: Beyond Key-Value Associative State Evolution
About
Linear attention reformulates sequence modeling as recurrent state evolution, enabling efficient linear-time inference. Under the key-value associative paradigm, existing approaches restrict the role of the query to the readout operation, decoupling it from state evolution. We show that query-conditioned state readout induces a structured value prediction over accumulated memory that complements key-based retrieval. Based on this insight, we propose Q-Delta, a query-aware delta rule that integrates mixed key-query prediction errors into state evolution, enabling jointly corrective dynamics while preserving delta-rule efficiency. We establish stability guarantees for the resulting dynamics and derive a hardware-efficient chunkwise-parallel formulation with a custom Triton implementation. Empirical results demonstrate stable optimization, competitive throughput, and consistent improvements over strong baselines on language modeling and long-context retrieval tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Zero-shot Language Modeling | Standard Language Modeling Suite WikiText-103, LAMBADA | WikiText-103 Perplexity19.21 | 19 | |
| Commonsense Reasoning | lm-evaluation-harness Zero-shot Reasoning Suite | ARC-Easy Accuracy68.27 | 12 | |
| Synthetic Needle-In-A-Haystack retrieval | Ruler S-NIAH | Accuracy (S-NIAH-1, 1K Context)100 | 6 | |
| Real-world retrieval | Real-world recall-intensive tasks | SWDE31.6 | 5 |