FastDiSS: Few-step Match Many-step Diffusion Language Model on Sequence-to-Sequence Generation--Full Version
Organizations: FPT Software AI Center, FPT Corporation · Department of Computer Science, Aalborg University, Denmark · Quantum AI and Cyber Security Institute, FPT Corporation
Abstract
Self-conditioning has been central to the success of continuous diffusion language models, as it allows models to correct previous errors. Yet its ability degrades precisely in the regime where diffusion is most attractive for deployment: few-step sampling for fast inference. In this study, we show that when models only have a few denoising steps, inaccurate self-conditioning induces a substantial approximation gap; this mistake compounds across denoising steps and ultimately dominate the sample quality. To address this, we propose a novel training framework that handles these errors during learning by perturbing the self-conditioning signal to match inference noise, improving robustness to prior estimation errors. In addition, we introduce a token-level noise-awareness mechanism that prevents training from saturation, hence improving optimization. Extensive experiments across conditional generation benchmarks demonstrate that our framework surpasses standard continuous diffusion models while providing up to 400x faster inference speed, and remains competitive against other one-step diffusion frameworks.
Figures & tables
| Model | Number of denoising steps (NFEs) | ||||
| 5 | 20 | 50 | 100 | 1,000 | |
| Original | 27.85 | 29.83 | 29.97 | 30.10 | 30.12 |
| Correct | 29.70 | 30.21 | 30.34 | 30.20 | 30.23 |
| Method | MBR | NFE | IWSLT14 | WMT14 | WMT16 | |||
| DE EN | EN DE | DE EN | EN DE | RO EN | EN RO | |||
| Transformer | 5 | - | 33.61 | 28.30 | 30.55 | 26.85 | 33.08 | 32.86 |
| CMLM | 5 | - | 29.41 | 24.34 | 28.71 | 23.22 | 31.13 | 31.26 |
| DiffusionLM ‡ | 50 | 20 | 29.11 | 22.91 | 19.69 | 17.41 | 30.17 | 29.39 |
| Diffomer ‡ | 20 | 20 | 28.01 | 23.31 | 25.30 | 23.80 | 29.37 | 29.20 |
| DINOISER | 50 | 20 | 31.61 | 25.70 | 29.05 | 24.26 | 31.22 | 31.08 |
| Method | MBR | NFE | QQP | Wiki-Auto | ||||
| BLEU | Rouge-L | BertScore | BLEU | Rouge-L | BertScore | |||
| Transformer ‡ | - | - | 27.22 | 57.48 | 83.81 | 26.93 | 49.07 | 73.81 |
| GPT2-large FT ‡ | - | - | 20.59 | 54.15 | 83.63 | 26.93 | 51.11 | 78.82 |
| CMLM × | - | 10 | 21.78 | 56.12 | - | 35.26 | 58.46 | 81.83 |
| LevT ‡ | - | - | 22.68 | 57.95 | 83.44 | 20.52 | 44.02 | 72.54 |
| Difformer ∗ | 10 | 20 | 27.95 | 59.24 | 82.97 | 34.78 | 54.55 | 78.86 |
| SCP | MANS | NFE | ||
|---|---|---|---|---|
| 27.98 | 29.78 | 1.80 | ||
| 29.64 | 30.36 | 0.72 | ||
| 30.77 | 31.49 | 0.72 | ||
| 31.17 | 31.66 | 0.49 |
| Steps | 5 | 20 | 100 | Fixed |
|---|---|---|---|---|
| 0.60 - 0.85 | 0.90 - 0.95 | 0.98 - 0.99 | 0.85 | |
| 0.25 - 0.90 | 0.15 - 0.35 | 0.05 - 0.15 | 0.50 | |
| BLEU | 25.48 | 26.35 | 25.70 | 25.84 |
| ROUGE-L | 57.29 | 57.47 | 56.82 | 57.18 |
| Noise Schedule | NFE=2 | NFE=5 | NFE=20 | |
|---|---|---|---|---|
| linear | Orig. | 26.50 | 27.17 | 27.54 |
| Ours | 26.84 | 27.52 | 27.54 | |
| cosine | Orig. | 25.43 | 27.05 | 27.57 |
| Ours | 27.32 | 27.77 | 27.94 | |
| NFE | DoT ( Plaid ) | ( CoT ) | |||
|---|---|---|---|---|---|
| Normal | SCP | Normal | SCP | ||
| 8 | 35.4 0.5 | 30.9 0.6 | |||
| 16 | 39.0 0.5 | 36.1 0.7 | |||
| 32 | 39.9 0.3 | 37.2 0.7 | |||
| 64 | 41.3 0.2 | 36.6 0.6 | |||
| NFE | Gen Perplexity ( ) | ||
|---|---|---|---|
| None | SC | SCP | |
| 2 | 3178.74 | 3250.22 | 2873.67 |
| 5 | 1144.42 | 1433.79 | 1215.41 |
| 20 | 243.81 | 254.72 | 244.40 |
| 50 | 143.82 | 133.79 | 130.52 |
| 100 | 113.34 | 97.88 | 95.48 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| NFEs | 5 | 20 | 50 | 100 |
|---|---|---|---|---|
| Sup | 0.047 | 0.008 | 0.009 | 0.010 |
| Method | MBR | NFE | Rouge-1 | Rouge-2 | Rouge-L |
| LSTM | 5 | - | 34.2 | 16.0 | 31.8 |
| CMLM | - | - | 34.4 | 15.6 | 32.2 |
| NAG-BERT | - | - | 35.1 | 16.5 | 33.3 |
| Difformer * | 10 | 20 | 34.9 | 17.0 | 32.4 |
| DiffuSeq | 10 | 2000 | 31.2 | 12.2 | 29.2 |
| Param | Metric | Default | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Value | – | 0.85 | 0.86 | 0.87 | 0.88 | 0.89 | 0.90 | 0.91 | 0.92 | 0.93 | |
| BLEU | 26.94 | 28.67 | 28.58 | 28.73 | 28.65 | 28.68 | 28.67 | 28.34 | 28.11 | 28.44 | |
| Value | – | 0.91 | 0.92 | 0.93 | 0.94 | 0.95 | 0.96 | 0.97 | 0.98 | 0.99 | |
| BLEU | 26.94 | 28.54 | 28.45 | 28.68 | 28.53 | 28.60 | 28.59 | 28.45 | 28.66 | 28.59 | |
| Value | – | 0.11 | 0.12 | 0.13 | 0.14 | 0.15 | 0.16 | 0.17 | 0.18 | 0.19 | |
| BLEU | 26.94 | 28.60 | 28.37 | 28.00 | 28.46 | 28.53 | 28.48 | 28.40 | 28.79 | 28.52 |
| Configurations | WMT14 | WMT16 | IWSLT14 | Gigaword | QQP | Wiki-Auto |
|---|---|---|---|---|---|---|
| Split | ||||||
| Training | 4,500,966 | 608,319 | 160,215 | 3,803,957 | 144,715 | 677,751 |
| Validation | 3,000 | 1,999 | 7,282 | 189,651 | 2,048 | 2,048 |
| Test | 3,003 | 1,999 | 6,750 | 1,951 | 2,500 | 5,000 |
| Preprocess | ||||||
| BPE | 40,000 | 30,000 | 10,000 | 60,000 | 15,000 | 40,000 |
| Step | Example 1 | |
| Source | - | He also twice participated in the Summer Olympics, starting in 1996. |
| Target | - | He was also in the 1996 Summer Olympics. |
| Baseline | 2 | He also twice in the Summer Summer Olympics 1996 Summer 1996. |
| 1 | He also twice in the Summer Summer in the in 1996. | |
| FastDiSS | 2 | He also twice twice in in Summer Summer Olympics, starting in 1996. |
| 1 | He also twice twice in the Summer Olympics, starting in 1996. |
| Example 1 | |
|---|---|
| Source | How do you think Federer, Nadal, Djokovic, Murray, Wawrinka, Delpo & Cilic rank in terms of potential |
| when it comes to greatest spot in the tennis? | |
| Target | How would you rank Federer, Nadal, Djokovic, Murray, Wawrinka, Delpo and Cilic in terms of greatness |
| achieved and potential spot in the history of sport? | |
| Baseline | How do you think about the Federer, rank in a good of Mury Wawrinka , Delgreatest rank and Cilic rank in |
| terms of for tennis? |
| Example 1 | |
| Source | This easy chicken curry recipe is made with just a handful of ingredients |
| Plaid | This easy chicken curry recipe is made with just a handful of ingredients make provide add that helps easy add help give add that helps welcome give help offer help for how you learn give help help help for act give help help help and perform give help help help for for how pay for help help help and even add help help help for give help that add give help up for the add provide help provide for help help help and add for how pay for help give help provide help and add give help help help and add help provide help to create help give help help help for for how doit add this give help help help that add help help give help help for and that give help add help give help give help for the run in the middle of the day add help help provide help help that gives simply give help help give help help provide help help provide help help for pay help help give help help pay give help give help give help help pay help the member on your family make help shop it … |
| Plaid + SCP | This easy chicken curry recipe is made with just a handful of ingredients at your disposal including a bow for glowing red ropes for strutter. It alone will be the trick since making anything else would normally be a lengthy method. Once that completes the dungeon is start laid out again towards the end where you learn the way to make naughty but once you find this little hidden dungeon point you will also need the basics of the tools to help you find your way to the recipe for cider seeds. So many would say these simple caves and dungeons are like paster glue in a Oddband environment. It is not just the recipe that is perhaps also the highlight in that the have side rooms for dungeon for completion in small spells or just locked in a spooky cave somewhere. The lack of questline support will prevent really good and fun faciettes. The formula the ingredients are a lot worth then anything else but for an hour of stats capable of requiring three seemed some how effective when you have the basics … |
| Example 2 | |
| Source | Generative models of text are very versatile: they can be used |
| Plaid | Negative models of text are very versatile: they can be used in the workplace and in the secondary market provide availability provides published versions varies and competitive paid provided available for tender available as the selection of availability choices were very common on a high … fair to read the initial sales framework very often … provide selection for the except immediately in commonly based on how pricing control edited ask that everyone creates as a matter of how much published provide is actually apply control on that immediately on offered provide on exchange for them provide paid write line of commissioned provide access made applied to more creative offered works cover different parts of the market agree that paid sharing is much more what use sharing work on the ability to in understanding in support of a contribution of a meaningful piece on say you can provide featured provide paid paid assignment paid submissions management provided applied arrangements with mixed paid paid shared fit easily perform create commissioned work in private … why is not offered applied availability on the subject or backing and reuse … to … more successful provide writing for working provide featured work submission creative exclusively created collection of icons is a symbol medium always learn how to make use of owns offering portfolio offering total free commissioned formats provided published licensing paid |
| Example | |
|---|---|
| Source | Lexically Constraints: Donald |
| Plaid | Sample 1: … focused in comments on newly uncovered FBI collusion with President Donald Trump, Bill Nunes revelation that almost no trump prosecute matters, despite widespread evidence, making him the primary target of these very crimes … |
| Sample 2: … It also includes some mud slinging with Obama FGA Chairman John Masters attacking Donald Trump for loose familiarity with the banking cartel. It’s no secret that Trump was beaten to the drapes by President Barack Obama on the actual exam. | |
| Sample 3: … March in London Members of Parliament looking at paintings of Prime Minister Donald trump during EU festival day Getty Images Brexit protest … | |
| Plaid + SCP | Sample 1: … It is also an equal, far more debasive sellout than the academia; some may have reasonably attributed the whole spread to Donald Nhutbout not clicking into the comments I have written this time; it seems to be unfortunate because both the coverage and commentary involved that continued focus could just seem to call out bias and denial … |
| Sample 2: … But at least on campuses, the so called feminist war on radical conservatives may actually be having a negative effect on Loomer attendance. When it comes to that Donald Trump presidency, we might actually find men who don’t have the viciousness, sorry love, of the feminist left … |
| Example | |
|---|---|
| Source | Negation: Donald but not Trump |
| Plaid | … [profile] Frederic Blanchfield Britain’s earliest basalt monument in Little Choyside, Chepole, Dorset. Source: MORI.mauipomwork-8x17-1.jpg… [profile] Marco Bori Special Projects Marco Bastoni Participants in discussion with Russian crionic residents in Venice. Artist/21-ponoco-8x17-1.jpg… [profile] Marco Baba.alexnet-8x17-1.jpg… [profile] Giananna Boone Interior Photography Histature from Rome Client – Social Architecture – Vanguard.photo/46-photograph-164x72-r.jpg… [profile] Donald Buckingham Industrial Art Izomage ProKnakon Call to exercise Tradarpones in Real Materials-stone-piping-8x17-608px.jpg… [slide] artmagazine-20x26x24.jpg… [slide] … |
| Plaid + SCP | … It was dress for contempt. The dishonesty of this whole development was clear, an astonishing and obvious, necessary to humanity, but seen in the remark Il Quartilver made which showed an almost impervolent fit of irony, necessary to discern truth from record. ‘ Our exile and our exile have been secured. The ocean has risen. Hell, Donald haath been a fool, before he came to save me, Taracharvi, and finally … not only in these troubled times it was taking a giant it-own for their charisma, but sol gallur phir bariet die Pope Donald . If they couldn’t say ‘wrong’, where did they get it? The slanders at Timon knew to just make radios, they could have solved … |
Diffusion Language Model Inference
Diffusion language model inference studies how to generate, guide, and serve text efficiently through iterative denoising and parallel token updates.
See all 165 papers