Rate-Resolved Text Evaluation with Masked Diffusion Language Models
Organizations: The University of Manchester · University of Pittsburgh · The University of Sheffield
Abstract
A single evaluation score can rank generation systems but rarely reveals what failed. We propose rate-resolved text evaluation, which turns the masking rate of masked diffusion language models into a diagnostic axis. By reconstructing text under different fractions of masked tokens, one diffusion score becomes a vector of band-conditioned scores that reflect reconstruction difficulty at varying levels of visible context. On SummEval, preserving this rate structure yields complementary signal: the best masking-rate band improves fusion with autoregressive likelihood over a collapsed diffusion scalar, survives a sample-shuffle control that breaks rate ordering, and adds residual signal to a fixed prompted judge. A second diffusion model reproduces the effect more weakly. On WMT MQM zh-en, roundtrip mask-and-reconstruct probing produces a tunable localization curve in which ranking-oriented AUROC and span-detection F1 peak at different masking rates. Factuality benchmarks bound the interpretation: band semantics are model- and task-dependent, so each application requires empirical validation.
Figures & tables
| Role | Model |
|---|---|
| Main diffusion | LLaDA-8B-Instruct |
| Replication diffusion | Dream-7B-Instruct |
| AR / judge | Qwen2.5-7B-Instruct |
| Static cloze control | XLM-R-XXL |
| Evaluator | SummEval relevance |
|---|---|
| Qwen judge | 0.3065 |
| Judge + AR | 0.3273 |
| Judge + diffusion | 0.3454 |
| Judge + AR + diffusion | 0.3636 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Med. rel. err. | IQR | Emp./theory CV | Rank | |
|---|---|---|---|---|
| 0.1 | 0.0246 | 0.0317 | 0.344 / 0.338 | 0.9969 |
| 0.3 | 0.0207 | 0.0133 | 0.170 / 0.172 | 0.9986 |
| 0.5 | 0.0167 | 0.0078 | 0.112 / 0.113 | 0.9996 |
| 0.7 | 0.0102 | 0.0053 | 0.073 / 0.074 | 0.9994 |
| 0.9 | 0.0030 | 0.0016 | 0.037 / 0.038 | 0.9997 |
| A. No-judge signals | ||
|---|---|---|
| Signal | Relevance | Gain |
| AR likelihood | 0.2106 | n/a |
| Diffusion best band | 0.2209 | n/a |
| AR + full diffusion scalar | 0.2610 | ref. |
| AR + best diffusion band | 0.2795 | +0.0185 |
| Item | Reproducibility detail |
|---|---|
| Code and scripts | The paper source package contains the LaTeX sources, rendered figures, and rendered tables. Experimental code and configuration files will be released after de-anonymization. |
| Core models | LLaDA uses GSAI-ML/LLaDA-8B-Instruct ; Dream uses Dream-org/Dream-v0-Instruct-7B ; AR and the fixed judge use Qwen/Qwen2.5-7B-Instruct ; the static cloze baseline uses facebook/xlm-roberta-xxl . |
| Pinned model refs | The local Qwen cache resolves main to a09a35458c702b33eeacc393d103063234e8bc28 ; diffusion checkpoints are recorded by release name in the current artifact and should be archived with commit hashes before de-anonymized release. |
| Environment | Python 3.11 with package lower bounds torch>=2.1 , transformers>=4.46 , accelerate>=0.34 , datasets>=2.18 , scipy>=1.11 , and pandas>=2.1 ; LLaDA and Dream configs record Transformers 4.46.3 and 4.46.2 respectively. |
| Benchmarks and licenses | SummEval is distributed with MIT licenses in the local artifact; WMT MQM is distributed under Apache-2.0. All are used through their released benchmark files and cited in the main text. |
| WMT preprocessing | The loader keeps hyp.* systems from newstest2021 zh-en, groups rows by system/document/segment, strips MQM markup, projects spans to character-level masks, derives severity slices from released segment-level counts, and aggregates rater penalties with the released severity weights. |
| Model | Dim. | Band | AR | Band | AR+band | Gain |
|---|---|---|---|---|---|---|
| LLaDA | Rel. | 8 | 0.2106 | 0.2209 | 0.2857 | +0.0648 |
| Cons. | 7 | 0.1022 | 0.1592 | 0.1849 | +0.0257 | |
| Flu. | 0 | 0.1699 | 0.1709 | 0.1887 | +0.0177 | |
| Coh. | 1 | 0.1695 | 0.0869 | 0.1695 | +0.0000 | |
| Dream | Rel. | 9 | 0.2106 | 0.2024 | 0.2638 | +0.0532 |
| Cons. | 8 | 0.1022 | 0.1239 | 0.1465 | +0.0226 |
| Model | Dim. | AR | AR+scalar | AR+band | ||
|---|---|---|---|---|---|---|
| LLaDA | Rel. | 0.2106 | 0.2610 | 0.2795 | +0.0185 | .0015 |
| Cons. | 0.1022 | 0.1428 | 0.1705 | +0.0276 | .0085 | |
| Flu. | 0.1699 | 0.1692 | 0.1852 | +0.0160 | .0895 | |
| Coh. | 0.1695 | 0.1590 | 0.1530 | -0.0059 | .9675 | |
| Dream | Rel. | 0.2106 | 0.2524 | 0.2517 | -0.0006 | .6577 |
| Cons. | 0.1022 | 0.1361 | 0.1423 | +0.0062 | .0055 |
| Budget | Runtime | Band | Fusion | Gain |
|---|---|---|---|---|
| 3h43m | 0.2263 | 0.2837 | 0.0574 | |
| 9h06m | 0.2204 | 0.2786 | 0.0582 | |
| 2d23h21m | 0.2209 | 0.2795 | 0.0586 |
| Bands | Band | Fusion | Gain | Selected region |
|---|---|---|---|---|
| 5 | 0.1668 | 0.2521 | 0.0414 | mostly band 4 |
| 10 | 0.2209 | 0.2795 | 0.0586 | band 8 in all folds |
| 20 | 0.1799 | 0.2778 | 0.0672 | bands 16 and 17 |
| Signal structure | AR fusion | Notes |
|---|---|---|
| Observed rate bands | 0.2795 | grouped-CV result |
| Full diffusion scalar | 0.2610 | scalarized LLaDA |
| Shuffled pseudo-bands | 0.2098 | mean over 100 shuffles |
| Best shuffled run | 0.2431 | maximum over 100 shuffles |
| Metric | Diffusion best | AR | Winner |
|---|---|---|---|
| Overall AUROC | 0.6705 | 0.6330 | Diff. |
| Pooled F1 | 0.2311 | 0.2636 | AR |
| Major F1 | 0.3464 | 0.2811 | Diff. |
| Minor F1 | 0.2852 | 0.2115 | Diff. |
| Slice | Diff. .3 | Diff. .5 | AR | XLM-R |
|---|---|---|---|---|
| Err. | 0.3251 | 0.3246 | 0.2987 | 0.2874 |
| Major | 0.3385 | 0.3381 | 0.3116 | 0.3018 |
| Minor | 0.2832 | 0.2838 | 0.2602 | 0.2443 |
| Pos. | Rec. | Diff. AUC | AR AUC | Diff. F1 | AR F1 |
|---|---|---|---|---|---|
| Beginning | 1,215 | 0.6587 | 0.6053 | 0.2763 | 0.2344 |
| Middle | 3,168 | 0.6365 | 0.5579 | 0.3582 | 0.2437 |
| End | 1,504 | 0.6478 | 0.5263 | 0.2809 | 0.1652 |
| Slice | Failure mode and excerpt |
|---|---|
| Judge + AR w/o Diff. rank err. J AR Diff. 73.5 29.0 1329.0 | Long-range ordering sensitivity Eight Iranian border guards have been killed in clashes with militants near the border with Pakistan. A militant group called Jaish al Adal claimed responsibility for the attack. |
| Judge + Diff. w/o AR rank err. J AR Diff. 15.0 1513.0 3.0 | Local fluency trap under repetition Liverpool beat Newcastle United 2 to 0. “I thought it was a penalty,” Allen said. “I thought it was a penalty. I thought it was a penalty.” |
| AR + Diff. w/o Judge rank err. J AR Diff. 1425.5 32.5 181.5 | Possible judge preference mismatch for terse factual summaries Wales’ crunch Euro 2016 qualifier with Belgium has been declared a 33,000 sell-out. The top two sides in Group B meet in Cardiff on June 12, with Wales in their best position to qualify for a major tournament since 1958. |
| Probe | |||
|---|---|---|---|
| Token surface F1 (linear) | 0.830 | 0.611 | 0.244 |
| Major-error AUROC (linear) | 0.659 | 0.652 | 0.523 |
| Major-error F1 (linear) | 0.567 | 0.546 | 0.479 |
| Token surface F1 (MLP) | 0.849 | 0.638 | 0.331 |
| Major-error AUROC (MLP) | 0.737 | 0.691 | 0.568 |
| Major-error F1 (MLP) | 0.685 | 0.592 | 0.570 |
| Model | Maj. AUC | Maj. F1 | Min. AUC | Min. F1 |
|---|---|---|---|---|
| LLaDA | 0.6774 @ .5 | 0.3464 @ .3 | 0.6506 @ .5 | 0.2852 @ .3 |
| Dream | 0.5871 @ .7 | 0.3134 @ .9 | 0.5418 @ .7 | 0.2597 @ .9 |
Language Model Generation Evaluation
Language model generation evaluation measures the quality, reliability, diversity, and task fit of outputs produced by language models.
See all 395 papers