首页 > AI前沿 > Commit Locally, Exit Globally: Coordinating Adaptive Sampling and Early Exit in Diffusion Language Models

Commit Locally, Exit Globally: Coordinating Adaptive Sampling and Early Exit in Diffusion Language Models

arXiv自然语言 2026-07-30 21:04 4 阅读 查看原文

Diffusion language models predict tokens in parallel but still require many iterative forward passes.

Adaptive sampling accelerates local token commitment, while recent global stopping methods seek to end generation early.

Our Trajectory Analysis

Our trajectory analysis shows that short answers can stabilize early in decoding, whereas reasoning outputs often require more steps to settle.

A fixed progress threshold therefore struggles to avoid both redundant computation on short-answer tasks and premature termination on reasoning tasks.

C$^4$ Framework

We introduce C$^4$, a training-free framework that combines local decoding acceleration with adaptive global early exit.

Confidence-Verified Early Exit (CVEE) determines when to end generation, skipping the remaining steps once its stopping criterion is met.

It combines confidence and temporal agreement to adapt stopping to the observed revision history.

Commit-Core-Then-Confirm (CCTC) accelerates token commitment while generation continues, reducing the decoding steps needed to complete each block.

It extends confidence-threshold parallel decoding with a boundary-anchored core and one-step confirmation of deferred predictions, constraining additional commitments to mitigate premature freezing under unresolved context.

Calibration and Performance

With CVEE calibrated once on LLaDA, the same numerical settings transfer across all 12 zero-shot tasks and to Dream without recalibration.

Under this configuration, C$^4$ reduces decoding steps by 64-95% and achieves measured decoding-loop speedups of 2.6x-18.6x, with task performance close to full decoding.

Component Analyses

Component analyses show complementary contributions: early exit supplies most savings on short-answer tasks, while local acceleration contributes more on reasoning and code generation.