CORA-Diff: Confidence-Oriented Residual Acceptance for Efficient Diffusion Language Model Inference
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Diffusion language models (DLMs) update many tokens in parallel, yet practical decoders often use a fixed denoising horizon. Many predictions stabilize early, but blockwise decoding continues until all positions are resolved, causing repeated dense forward passes. Existing accelerators often rely on learned filters, modified scores, dependency models, or cache-specific mechanisms. We ask whether native trajectory signals can identify residual positions likely to match the deterministic dense endpoint. We propose CORA-Diff, a training-free method that preserves the original transfer rule and applies confidence-and-persistence gating only to positions that rule leaves unresolved. Accepted tokens remain visible as context, and the block terminates once all positions are resolved. This requires no backbone change, learned acceptance model, or logit modification. Our theory explains why high-confidence, persistent predictions are more likely to match the fixed-horizon dense endpoint, and paired post-intervention trajectories provide direct empirical support. We select one operating point on a separate GSM8K calibration subset and freeze it for all evaluations. Under a matched Learn2PD-style LLaDA protocol, CORA-Diff has the lowest measured runtime in all eight task-length settings. Task scores match or exceed dense decoding in five settings, and the largest observed drop is 1.22 points. Its incremental speedups over EOS-aware dense decoding are 2.70x and 3.32x on GSM8K and HumanEval. It also reaches 13.14x under the fixed-horizon 1024/1024 mechanism-isolation protocol and transfers to Dream without retuning at 3.18x-3.53x. These results show that native confidence and persistence enable reliable residual acceptance, reducing repeated denoising computation while preserving task quality.
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 24 of its 33 references that are in the map. First posted on arXiv 31 Jul 2026.
Builds on
24 of 33 references.
Closest ancestors · 9
Same direction, newest first
- 2026Attention-Discounted Adaptive Sampler for Masked Diffusion Language Models9 Jun
- 2026DMax: Aggressive Parallel Decoding for dLLMs9 Apr
- 2025WeDLM: Reconciling Diffusion Language Models with Standard Causal Attention for Fast Inference28 Dec
- 2025KLASS: KL-Guided Fast Inference in Masked Diffusion Models7 Nov
- 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion27 May
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models23 Oct
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data6 Jun
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
Other references · 15
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 246
- 2025dParallel: Learnable Parallel Decoding for dLLMscited by 59
- 2025Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Principlescited by 43
- 2025AdaBlock-dLLM: Semantic-Aware Diffusion LLM Inference via Adaptive Block Sizecited by 30
- 2025Learning to Parallel: Accelerating Diffusion Large Language Models via Learnable Parallel Decodingcited by 26
- 2026DAPD: Dependency-Aware Parallel Decoding via Attention for Diffusion LLMscited by 13
- 2026DAWN: Dependency-Aware Fast Inference for Diffusion LLMscited by 12
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026CAI-DLLM: Convergence Aware Inference for Diffusion Language Models
- 2026CoDR: Training-Free Confidence-Drift Remasking for Diffusion Language Models
- 2025Beyond Confidence: Adaptive and Coherent Decoding for Diffusion Language Models
- 2026CD4LM: Consistency Distillation and aDaptive Decoding for Diffusion Language Models
- 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Models
- 2025Sequential Diffusion Language Models
- 2026AsyncLane: Decoupling Refinement from Advancement in Diffusion Language Model Decoding
- 2026Self-Distilled Trajectory-Aware Boltzmann Modeling: Bridging the Training-Inference Discrepancy in Diffusion Language Models