GRAFT: Restoring the Missing Sampling Interface in Continuous Diffusion Language Models
Organizations: University of Electronic Science and Technology of China
Abstract
Continuous diffusion language models generate a whole sequence in parallel, never drawing a single word at any intermediate step; only the final step turns embeddings into text. Models such as ELF now report autoregressive-level quality at a fraction of the sampling steps of discrete-token diffusion models, and this design principle -- no step-wise discretization -- is what their own ablations credit for the gain. We show that this same principle carries a hidden cost: a collapse of the lexical long tail. Measured against held-out human text, ELF's output matches the human rate on common words almost exactly, yet its rarest words appear at only 0.30x the human rate, a gap that does not close from 105M to 652M parameters and that no sampler setting these models expose can reach. A same-size GPT-2, in contrast, spans zero to above the human rate by its decoding rule alone. We trace the difference to what continuous diffusion models lack: with no step that ever draws a discrete word, every step is instead driven by an average over the candidates still in play, and averaging a rare candidate together with the common word competing for its slot lands on the common word. An autoregressive model avoids this because its decoding rule samples one candidate rather than averaging over all of them. We propose GRAFT, a training-free method that grafts this missing choice back onto a frozen model: while the model is still undecided, GRAFT commits the rare words the average would otherwise discard, then regenerates the sequence from scratch with those commitments held in place. Across three ELF scales and on LangFlow, an independent continuous dLLM, GRAFT restores the rare-word share of generated text from 0.42-0.77x the human rate to 0.82-1.10x, a relative gain of 34-109%, and moves the entire word-frequency profile closer to human on every model tested, by 2.7x on ELF-B and 12x on LangFlow.
Figures & tables
| Zipf bucket (% of word tokens) | ELF-B | ELF-M | ELF-L | Human | ELF-L / Human |
|---|---|---|---|---|---|
| 20.60 | 20.62 | 21.02 | 20.38 | ||
| 53.07 | 51.59 | 48.13 | 46.20 | ||
| 20.81 | 20.69 | 22.64 | 21.27 | ||
| 3.14 | 4.93 | 5.82 | 8.12 | ||
| 0.44 | 0.68 | 0.80 | 2.51 | ||
| 0.27 | 0.24 | 0.26 | 0.86 |
| lexical tail, and its control | distance to human profile | |||||
| tail % | human | non-word % | JSD | types | ||
| Human reference | 11.88 | 1.00 | 1.14 | — | — | 17 726 |
| ELF-B (105M) | ||||||
| Stock | 5.01 | 0.42 | 2.64 | .0134 | .182 | 11 795 |
| Stock, best-of-4 | 5.97 | 0.50 | 1.79 | .0101 | .175 | 11 482 |
| Argmax control | 5.31 | 0.45 | 2.42 | .0133 | .209 | 12 124 |
| tail % | non-word % | fidelity % | |
|---|---|---|---|
| stock | 4.54 | 2.35 | — |
| rare commitment, no rule | 7.92 | 4.14 | 86.7 |
| atomic commitment | 8.86 | 2.71 | 87.4 |
| splicing | 9.61 | 2.47 | 100.0 |
| word-boundary co-pinning | 9.10 | 1.78 | 100.0 |
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| Alternative explanation | Measurement that rules it out |
|---|---|
| The decoding head favors common words | Clean embeddings of human text are decoded at , with a retention of in every band ( positions). |
| Rare words have smaller decision cells, the regions of embedding space that decode to them | They do, but only under noise far larger than the denoising head’s own error. With isotropic noise of per-dimension , of the rarest words flip, that is, decode to another word, compared with of the commonest; at , which covers the denoising head’s own errors ( – ), at most of any band flips. On the segment from the competitor’s embedding ( ) to the true word’s ( ), the flip boundary is symmetric, at for rare words and for common ones, both far from . |
| The output is an average of two candidates | Projected onto this segment, the denoising head’s output sits at and would still decode to the true word, but the perpendicular residual is the segment’s length, so the output is not on the segment. |
| The self-conditioning feedback of Zhang et al. (2026) | Sweeping the self-conditioning strength over , , , , and leaves the tail flat ( , , , , and ), even at , where Zhang et al. (2026) report generative perplexity rising to (ELF-B, 64 steps). |
| A single error direction, removable by a one-dimensional fix | The frequency-axis direction, the normalized difference between the mean embeddings of common and rare positions, carries only – of the error norm, and removing it does not recover accuracy ( ). |
| The denoising head picks the commonest candidate | At contested positions, taken at where the decoding head is uncertain and its most common candidate is not its most probable, the denoising head chooses the most common candidate only of the time (the most probable one, ). Where its word is among the candidates ( of these positions), its mean frequency rank is , against for a random candidate: a real but weak bias. |
| Arm | accept % | tail@acc | tail (unfiltered) | non-word (unfiltered) |
|---|---|---|---|---|
| GPT-2 greedy | 0.0 | — | 0.00 | 0.00 |
| GPT-2 top- 50 | 90.3 | 5.28 | 5.56 | 0.24 |
| GPT-2 nucleus | 84.9 | 9.80 | 10.06 | 1.17 |
| GPT-2 sample | 51.7 | 13.36 | 14.60 | 3.01 |
| GPT-2 medium greedy | 0.0 | — | 0.26 | 0.00 |
| GPT-2 medium top- 50 | 92.6 | 5.92 | 6.14 | 0.25 |
| Arm | tail (unfiltered) | non-word (unfiltered) | top-5k tail |
|---|---|---|---|
| LangFlow stock | 8.59 | 2.44 | 11.05 |
| LangFlow champion | 15.09 | 3.27 | 13.70 |
| LangFlow champion co-pinning | 16.20 | 2.41 | 16.48 |
| LangFlow ladder (5) | 12.35 | 1.98 | 15.87 |
| ELF-B stock | 4.54 | 2.35 | 6.61 |
| ELF-B champion | 9.45 | 2.52 | 11.26 |
| Excess error (denoising / decoding) | Zipf | ||||
|---|---|---|---|---|---|
| ELF-B | |||||
| ELF-M | |||||
| ELF-L | |||||
| ELF-B error rates (%) |
| Model | champion tail@acc | champion co-pinning: accept % / tail@acc % / | ||
| ELF-S (43.7M) | (unfiltered, /sample) | — | ||
| ELF-B (105M) | to | 69.5 | 9.68 | |
| ELF-M (342M) | to | 64.0 | 11.81 | |
| ELF-L (652M) | to | 63.5 | 12.98 | |
| Sampler | accept % | tail@acc | tail@acc [95% CI] | 4-gram rep. | non-word % |
|---|---|---|---|---|---|
| Stock | 60.5 | 8.28 | — | 0.53 | 2.44 |
| MCB | 77.9 | 7.77 | [ ], | 0.00 | 1.87 |
| MCB , | 19.8 | 4.77 | [ ] | 20.10 | 0.74 |
| MCB , | — | off-manifold breakdown | 35.26 | 4.23 | |
| champion co-pinning | 52.0 | 13.61 | [ ] | 1.28 | 2.41 |
| Arm | tail (unfiltered) | 4-gram rep. | types/100k | accept (%) | tail@acc vs. argmax |
|---|---|---|---|---|---|
| argmax (interface shut) | 6.89 | 80.9 | 5,755 | 7.2 | (reference) |
| low-confidence, | 6.85 | 77.6 | 6,002 | 13.5 | , n.s. |
| low-confidence, | 7.15 | 63.8 | 7,919 | 17.0 | , n.s. |
| random-order, | 11.16 | 0.0 | 28,599 | 55.8 | , |
| Human (OpenWebText) | 11.49 | — | 5,497 | 90.5 |
| Configuration | tail (%) | gen. perplexity | 4-gram rep. (%) |
|---|---|---|---|
| ELF-B, 16 steps, | 7.05 | 44.51 | 0.63 |
| ELF-B, 32 steps, | 3.88 | 29.23 | 1.21 |
| ELF-B, 32 steps, (default) | 3.75 | 24.18 | 2.84 |
| ELF-B, 32 steps, | 4.02 | 23.40 | 2.87 |
| ELF-B, 64 steps, | 4.57 | 18.97 | 7.61 |
| ELF-B, final-step sampling | 4.72 | — | — |
| Model | Arm | tail@acc | foreign % | raw | discounted |
|---|---|---|---|---|---|
| human | — | 11.17 | 0.8 | — | — |
| ELF-B | stock | 4.30 | 1.6 | — | — |
| fixed | 9.67 | 13.8 | |||
| fixed, gated | 9.54 | 5.0 | |||
| tuned | 12.53 | 24.5 | |||
| tuned, gated | 11.65 | 2.1 |
| Sampler | accept % | tail@acc % | tail@acc |
| ELF-B | |||
| stock | 66.0 | 4.30 | — |
| stock best-of-4 | 98.0 | 5.64 | |
| argmax control | 74.5 | 4.92 | (n.s.) |
| Graft | 68.3 | 9.54 | |
| Graft (tuned) | 57.3 | 11.74 | |
| Model | Dose | selection | held-out 1 | held-out 2 |
|---|---|---|---|---|
| ELF-B | fixed | .0049 | .0047 | .0052 |
| tuned | .0012 | .0012 | .0012 | |
| ELF-M | fixed | .0082 | .0083 | .0072 |
| tuned | .0059 | .0079 | .0052 | |
| ELF-L | fixed | .0049 | .0048 | .0041 |
| tuned | .0047 | .0045 | .0033 |
| Arm | commits | JSD | accept % | tail@acc | disc. | |
|---|---|---|---|---|---|---|
| stock | — | .0025 | .023 | 60.5 | — | — |
| stock best-of-4 | — | .0015 | .002 | 99.0 | ||
| quota | 581 | .0001 | .040 | 51.3 | (n.s.) | |
| quota ungated | 781 | .0003 | .052 | 49.5 | (n.s.) | |
| quota ungated | 1407 | .0024 | .079 | 42.4 | ||
| quota | 1489 | .0044 | .091 | 39.2 |
| Sampler | accept % | tail@acc | |||||
|---|---|---|---|---|---|---|---|
| Human reference | 90.5 | 11.17 | — | 88.39 | 8.01 | 2.41 | 0.74 |
| Stock | 62.0 | 4.31 | — | 94.35 | 3.53 | 0.50 | 0.29 |
| champion co-pinning ( ) | 68.0 | 9.65 | 88.71 | 8.01 | 1.06 | 0.58 | |
| attestation gate | 61.5 | 9.67 | 88.68 | 8.15 | 1.01 | 0.51 | |
| Dose gate | 74.0 | 14.39 | 84.06 | 12.40 | 1.34 | 0.66 | |
| Band quota | 72.0 | 12.04 | 86.50 | 8.54 | 2.40 | 1.10 |
| ELF-B arm | non-word | JSD | ||||
|---|---|---|---|---|---|---|
| Human | 21.34 | 8.13 | 2.52 | 0.86 | 0.66 | — |
| Stock | 22.88 | 3.76 | 0.65 | 0.35 | 2.40 | 0.0124 |
| Champion co-pinning ( ) | 20.24 | 8.23 | 1.12 | 0.64 | 2.30 | 0.0032 |
| , | 20.39 | 8.52 | 0.98 | 0.57 | 2.24 | 0.0038 |
| 18.03 | 12.79 | 1.23 | 0.74 | 2.01 | 0.0067 | |
| Band quota | 19.22 | 8.78 | 2.55 | 1.19 | 2.00 | 0.0011 |
| Model | Arm | commits | unattested | tail@acc | with gate |
|---|---|---|---|---|---|
| ELF-B | champion co-pinning | 6,220 | |||
| ELF-B | band quota | 9,517 | |||
| ELF-B | dose | 12,800 | — | ||
| LangFlow | champion co-pinning | 2,429 | |||
| LangFlow | quota | 2,640 |
| LangFlow arm | accept % | tail@acc | tail@acc | ||
|---|---|---|---|---|---|
| Human | 90.5 | 11.17 | — | 2.41 | 0.74 |
| Stock | 60.5 | 8.28 | — | 1.11 | 0.35 |
| Quota | 35.7 | 13.24 | — | — | |
| Quota | 37.0 | 16.69 | 3.32 | 2.25 | |
| attestation gate | 38.0 | 15.05 | 4.10 | 0.87 | |
| Quota | 39.0 | 24.16 | — | — |
| Value | Rule | Used in |
|---|---|---|
| Tail | all-token, unfiltered, foreign discount on the numerator | Table 2 |
| Tail | alphabetic, unfiltered, no discount | Table 1 (three rare bands), Figure 1 b, Tables 5 , 6 , 10 , 11 |
| Tail ( ) | alphabetic, accepted set under the dual bar (accept ) | every tail@acc reading, e.g. Tables 8 , 12 , 13 , 16 , 19 |
| Tail | top-5k tail, accepted set | Table 6 (right) |
| Non-words / | all-token / alphabetic, unfiltered | Table 2 / Tables 1 , 5 , 17 |
| ; slope | all-token; regular-expression tokens | Table 2 ; § 4 |
| Model | Row | seed | tail % | human | non-word % | JSD | types | |
|---|---|---|---|---|---|---|---|---|
| ELF-B | Graft | 31337 | 9.78 | 0.82 | 2.61 | .0049 | .138 | 13,757 |
| 55555 | 9.91 | 0.83 | 2.55 | .0047 | .111 | 14,037 | ||
| 77777 | 9.58 | 0.81 | 2.54 | .0052 | .141 | 13,693 | ||
| ELF-M | Graft | 31337 | 12.18 | 1.03 | 3.13 | .0082 | .061 | 15,832 |
| 55555 | 12.14 | 1.02 | 2.96 | .0083 | .063 | 15,853 | ||
| 77777 | 11.75 | 0.99 | 3.22 | .0072 | .075 | 15,400 |
| Setting | ELF-B / M / L | LangFlow |
| Checkpoint, vocabulary | official OpenWebText (105M / 342M / 652M), T5 ( k) | official OpenWebText (652M), GPT-2 BPE |
| Sampler | SDE, steps, churn , self-conditioning guidance , logit-normal schedule a | deterministic Euler, steps, linear time grid |
| Length; samples, batch | ; , batch (ELF-B tuned: ) | ; , batch (stock, seed : ) |
| Seeds; read time | , , (best-of-4 adds ); | , , (best-of-4 adds ); |
| Selection | , ; candidates among the most likely words, rare words ; ; spacing ; atomic commitment, splicing and co-pinning on | |
| Attestation gate | on; attestation lexicon of documents – of the reference shard, minimum count | |
Diffusion Language Model Decoding
Diffusion language model decoding generates text through iterative parallel refinement, balancing fewer denoising steps with accuracy, coherence, and control.
See all 250 papers