Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Models

Wen Wang, Bozhen Fang, Chenchen Jing, Yongliang Shen, Yangyi Shen +4 more
8/12/2025
cs.CLcs.AI

No official code found. Best implementation: aim-uofa/dLLM-MidTruth (66 stars, Apache-2.0, last commit 2026-03-05).

Abstract

Diffusion large language models (dLLMs) generate text through iterative denoising, yet current decoding strategies discard rich intermediate predictions in favor of the final output. Our work here reveals a critical phenomenon, temporal oscillation, where correct answers often emerge in the middle process, but are overwritten in later denoising steps. To address this issue, we introduce two complementary methods that exploit temporal consistency: 1) Temporal Self-Consistency Voting, a training-free, test-time decoding strategy that aggregates predictions across denoising steps to select the most consistent output; and 2) a post-training method termed Temporal Consistency Reinforcement, which uses Temporal Semantic Entropy (TSE), a measure of semantic stability across intermediate predictions, as a reward signal to encourage stable generations. Empirical results across multiple benchmarks demonstrate the effectiveness of our approach. Using the negative TSE reward alone, we observe a remarkable average improvement of 24.7% on the Countdown dataset over an existing dLLM. Combined with the accuracy reward, we achieve absolute gains of 2.0% on GSM8K, 4.3% on MATH500, 6.6% on SVAMP, and 25.3% on Countdown, respectively. Our findings underscore the untapped potential of temporal dynamics in dLLMs and offer two simple yet effective tools to harness them.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

[ICLR'26] Official PyTorch implementation of "Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Models".

663Shell, PythonAug 4, 20257 months agoApache-2.0

Similar Papers

Cite this paper

@article{wang2025time,
  title  = {Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Models},
  author = {Wen Wang and Bozhen Fang and Chenchen Jing and Yongliang Shen and Yangyi Shen and Qiuyu Wang and Hao Ouyang and Hao Chen and Chunhua Shen},
  year   = {2025},
  eprint = {2508.09138},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2508.09138v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2508.09138.svg)](https://codeofpaper.com/paper/2508.09138)

Discussion