mlsanityData to 7 Oct 2026

Archer: Adaptive Reuse of Cached Hidden States for Efficient Rollback in Diffusion Language Models

arXiv 2608.08086 ↗First posted 8 Aug 2026cs.CLcited by 0

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Diffusion language models (DLMs) iteratively refine a sequence, allowing earlier predictions to be revised as context evolves. This rollback capability distinguishes them from irreversible autoregressive generation, but makes inference costly. Every denoising update alters the global context, forcing both prompt and response states to be recomputed even though only response tokens are revisable. Key-value (KV) caching could reduce this cost, yet conventional caching assumes immutable historical states and is therefore difficult to reconcile with rollback. In this paper, we introduce Adaptive Reuse of Cached Hidden States for Efficient Rollback (Archer), a training-free KV caching method for rollback-capable DLMs. Archer asymmetrically keeps the mutable response synchronized with the current hypothesis while reusing prompt K/V within a bounded state neighborhood. Although prompt representations also change under bidirectional attention, their token identities remain fixed; bounded reuse therefore amortizes repeated prompt computation without caching mutable response states. It also delays feedback from tentative tokens, reducing premature reinforcement of transient high-confidence errors and giving rollback more opportunity to correct them. Our analysis characterizes prompt reuse as a reversibility-aligned cache boundary, bounds its state-dependent approximation error, and gives a decoder-margin condition for preserving full-refresh decisions. Existing DLM acceleration often trades quality for speed. Archer shifts this frontier, attaining the best mean performance of 33.63% together with a 2.57x mean speedup on the main suite. Across evaluated settings, it improves Pass@1 by up to 3.05 points and reaches up to 2.95x speedup. Controlled analyses connect the quality gain to delayed prompt feedback and validate state-aware refresh. Our code is available at https://github.com/Hxnng/Archer.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 17 of its 23 references that are in the map. First posted on arXiv 8 Aug 2026.

Builds on

17 of 23 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Models28 Sep
  2. 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation25 Jun
  3. 2025Accelerated Sampling from Masked Diffusion Models via Entropy Bounded Unmasking30 May
  4. 2025Remasking Discrete Diffusion Models with Inference-Time Scaling1 Mar
  5. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct

Other references · 12

  1. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  2. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  3. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  4. 2021Program Synthesis with Large Language Modelscited by 3,509
  5. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
  6. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
  7. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  8. 2021FinQA: A Dataset of Numerical Reasoning over Financial Datacited by 552
  9. 2022Efficiently Scaling Transformer Inferencecited by 529
  10. 2023Prompt Cache: Modular Attention Reuse for Low-Latency Inferencecited by 255
  11. 2023CodeScore: Evaluating Code Generation by Learning Code Executioncited by 66
  12. 2024Promises and Pitfalls of Generative Masked Language Modeling: Theoretical Framework and Practical Guidelinescited by 6

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025dKV-Cache: The Cache for Diffusion Language Models
  2. 2026SPA-Cache: Singular Proxies for Adaptive Caching in Diffusion Language Models
  3. 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion
  4. 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
  5. 2025A Survey on Diffusion Language Models
  6. 2025d2Cache: Accelerating Diffusion-Based LLMs via Dual Adaptive Caching
  7. 2026EntropyCache: Decoded Token Entropy Guided KV Caching for Diffusion Language Models
  8. 2026Fixed State, Long Reach: What a Constant-Size Cache Buys Block Diffusion at Scale