AIJun 4

CaRE Compute-aware Remasking Evaluation Protocol for Masked Diffusion Language Models

Yash Shah, Abhijit Chakraborty, Vivek Gupta
arXiv:2607.24763h-index: 2
Originality Incremental advance
AI Analysis

For researchers evaluating MDLMs, CaRE provides a standardized protocol to prevent evaluation artifacts from conflating algorithmic improvements, addressing a critical reproducibility bottleneck.

Masked diffusion language models (MDLMs) are advancing but lack standardized evaluation, causing incomparable results. CaRE standardizes compute, metrics, and stochasticity, revealing that temperature explains most MAUVE variance and reverses several published rankings, with high-entropy remasking reducing MAUVE by 0.296 at 256 steps.

Masked diffusion language models (MDLMs) are advancing rapidly, yet the evaluation standards needed to reliably interpret their progress have not kept pace. Despite MDLMs becoming competitive with autoregressive language models, seven recent remasking papers evaluate under incompatible settings, varying nominal step counts, metrics, and sampling temperatures without jointly controlling these factors, rendering their strategy rankings largely incomparable and leaving open whether reported gains reflect algorithmic improvements or evaluation artifacts. We present CaRE, a compute-aware evaluation framework that audits MDLM remasking strategies by standardizing actual number of function evaluations (NFE), enforcing multi-metric reporting, and explicitly controlling stochasticity. Applied to 7 remasking strategies across LLaDA-8B-Base and Dream-7B-Base at 4 stochasticity levels and 3 step budgets on OpenWebText and LM1B, CaRE reveals that: (i) temperature explains the majority of MAUVE variance, (ii) compute-matched comparisons reverse several published strategy rankings, and (iii) informed remasking and stochastic unmasking are in tension, with high-entropy remasking reducing MAUVE by 0.296 at 256 steps at unmask_temp=0.25 (p=0.020). A CaRE leaderboard covering 12 open-weight MDLMs (150M to 8B parameters) shows that this interaction direction holds across architectures and scales. These findings demonstrate that current MDLM evaluations can systematically conflate algorithmic improvements with hidden choices of compute and stochasticity. We release the evaluation protocol, implementation, and leaderboard to ensure future remasking claims are reproducible and comparable.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes