A Reparameterized Discrete Diffusion Model for Text Generation
About
This work studies discrete diffusion probabilistic models with applications to natural language generation. We derive an alternative yet equivalent formulation of the sampling from discrete diffusion processes and leverage this insight to develop a family of reparameterized discrete diffusion models. The derived generic framework is highly flexible, offers a fresh perspective of the generation process in discrete diffusion models, and features more effective training and decoding techniques. We conduct extensive experiments to evaluate the text generation capability of our model, demonstrating significant improvements over existing diffusion models.
Lin Zheng, Jianbo Yuan, Lei Yu, Lingpeng Kong• 2023
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instruction Following | IFEval (test) | IFEval Score52.67 | 92 | |
| Mathematical Reasoning | GSM8k 5-shot | -- | 82 | |
| Code Generation | MBPP 3-shot | Pass Rate40 | 43 | |
| Code Generation | MBPP 3-shot | Pass@1 Accuracy40 | 36 | |
| Machine Translation | IWSLT De-En 14 | BLEU Score32.14 | 35 | |
| Sudoku Solving | Sudoku (test) | Accuracy82 | 33 | |
| Text Simplification | WikiAuto | BLEU43.86 | 29 | |
| Paraphrasing | QQP | BLEU30.83 | 22 | |
| Machine Translation | WMT En-De '14 | SacreBLEU26.54 | 22 | |
| Seq2Seq | QQP | ROUGE-L59.5 | 18 |
Showing 10 of 15 rows