DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
We present DiffusionBERT, a new generative masked language model based on discrete diffusion models. Diffusion models and many pre-trained language models have a shared training objective, i.e., denoising, making it possible to combine the two powerful models and enjoy the best of both worlds. On the one hand, diffusion models offer a promising training strategy that helps improve the generation quality. On the other hand, pre-trained denoising language models (e.g., BERT) can be used as a good initialization that accelerates convergence. We explore training BERT to learn the reverse process of a discrete diffusion process with an absorbing state and elucidate several designs to improve it. First, we propose a new noise schedule for the forward diffusion process that controls the degree of noise added at each step based on the information of each token. Second, we investigate several designs of incorporating the time step into BERT. Experiments on unconditional text generation demonstrate that DiffusionBERT achieves significant improvement over existing diffusion models for text (e.g., D3PM and Diffusion-LM) and previous generative masked language models in terms of perplexity and BLEU score.
Key facts. Cited by 214 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 29 of its 42 references that are in the map. First posted on arXiv 28 Nov 2022.
Builds on
29 of 42 references.
Closest ancestors · 2
Same direction, newest first
- 2022DiffuSeq: Sequence to Sequence Text Generation with Diffusion Models17 Oct
- 2021Autoregressive Diffusion Models5 Oct
Other references · 27
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
15 more
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2020Pre-trained Models for Natural Language Processing: A Surveycited by 505
- 2018Texygen: A Benchmarking Platform for Text Generation Modelscited by 491
- 2017Non-Autoregressive Neural Machine Translationcited by 463
- 2013One Billion Word Benchmark for Measuring Progress in Statistical Language Modelingcited by 308
- 2022Black-Box Tuning for Language-Model-as-a-Servicecited by 256
- 2019Levenshtein Transformercited by 199
- 2019BERT has a Mouth, and It Must Speak: BERT as a Markov Random Field Language Modelcited by 185
- 2021Step-unrolled Denoising Autoencoders for Text Generationcited by 102
- 2018Language GANs Falling Shortcited by 97
- 2020Non-Autoregressive Machine Translation with Latent Alignmentscited by 96
- 2017Quasar: Datasets for Question Answering by Search and Readingcited by 50
Built on by
Citing papers per year: 2022: 3; 2023: 26; 2024: 35; 2025: 75; 2026: 75.
214 papers.
Sort:
By area:
- 2025Large Language Diffusion Modelscited by 912
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2025Dream 7B: Diffusion Large Language Modelscited by 544
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 381
- 2024Discrete Flow Matchingcited by 317
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Modelscited by 274
- 2023DIFUSCO: Graph-based Diffusion Solvers for Combinatorial Optimizationcited by 221
- 2024Scaling up Masked Diffusion Models on Textcited by 199
38 more
- 2023A Reparameterized Discrete Diffusion Model for Text Generationcited by 152
- 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 148
- 2025dKV-Cache: The Cache for Diffusion Language Modelscited by 143
- 2025The Diffusion Dualitycited by 135
- 2022Latent Diffusion for Language Generationcited by 115
- 2023DiffuRec: A Diffusion Model for Sequential Recommendationcited by 93
- 2024A Phase Transition in Diffusion Models Reveals the Hierarchical Nature of Datacited by 93
- 2025Discrete Diffusion VLA: Bringing Discrete Diffusion to Action Decoding in Vision-Language-Action Policiescited by 90
- 2025A Survey on Diffusion Language Modelscited by 78
- 2022SeqDiffuSeq: Text Diffusion with Encoder-Decoder Transformerscited by 64
- 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
- 2025LLaDA-MoE: A Sparse MoE Diffusion Language Modelcited by 57
- 2023DINOISER: Diffused Conditional Sequence Learning by Manipulating Noisescited by 50
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Frameworkcited by 48
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2025Theoretical Benefit and Limitation of Diffusion Language Modelcited by 47
- 2025FUDOKI: Discrete Flow-based Unified Understanding and Generation via Kinetic-Optimal Velocitiescited by 46
- 2023BoDiffusion: Diffusing Sparse Observations for Full-Body Human Motion Synthesiscited by 45
- 2026ELF: Embedded Language Flowscited by 45
- 2023Diffusion Models for Non-autoregressive Text Generation: A Surveycited by 44
- 2025DiffuSpec: Unlocking Diffusion Language Models for Speculative Decodingcited by 42
- 2025LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMscited by 40
- 2023Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuningcited by 39
- 2023Diffusion Augmentation for Sequential Recommendationcited by 38
- 2023DiffuSeq-v2: Bridging Discrete and Continuous Text Spaces for Accelerated Seq2Seq Diffusion Modelscited by 36
- 2026Latent Forcing: Reordering the Diffusion Trajectory for Pixel-Space Image Generationcited by 36
- 2026LangFlow: Continuous Diffusion Rivals Discrete in Language Modelingcited by 36
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMscited by 34
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modelingcited by 34
- 2025Efficient-DLM: From Autoregressive to Diffusion Language Models, and Beyond in Speedcited by 33
- 2025Attention Is All You Need for KV Cache in Diffusion LLMscited by 32
- 2024Text-Guided Molecule Generation with Diffusion Language Modelcited by 31
- 2025Continuous Diffusion Model for Language Modelingcited by 31
- 2025Self Speculative Decoding for Diffusion Large Language Modelscited by 30
- 2024Diff-RNTraj: A Structure-aware Diffusion Model for Road Network-constrained Trajectory Generationcited by 29
- 2025Jailbreaking Large Language Diffusion Models: Revealing Hidden Safety Flaws in Diffusion-Based Text Generationcited by 26
- 2026Nemotron-Labs-Diffusion: A Tri-Mode Language Model Unifying Autoregressive, Diffusion, and Self-Speculation Decodingcited by 24
- 2025Breaking AR's Sampling Bottleneck: Provable Acceleration via Diffusion Language Modelscited by 23
50 most cited of 214 All 214 (JSON)
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMs13 Oct 2026
- 2026Prefilling-dLLM v2: Learned Chunk Routing for Long-Context Diffusion Language Models13 Oct 2026
- 2026BiMoGen: Bidirectional Motion-Text Generation via Unified Masked Discrete Diffusion28 Sep 2026
- 2026Does Uniform Discrete Diffusion Need Time?25 Sep 2026
- 2026Why Jailbreaks Succeed in Diffusion Language Models: An Energy Landscape Analysis25 Sep 2026
- 2026Entropy Can Flow, or It Can Guide. Be Entropy. LEDFlow: Introducing Entropy-guided Generation Order into Uniform Discrete Flow20 Sep 2026
- 2026Efficient One-to-Many Translation with Joint Multi-Stream Diffusion14 Sep 2026
- 2026Representation-based Masked Diffusion Model11 Sep 2026
- 2026In-Place Instruction Following in Diffusion Language Models7 Sep 2026
- 2026Beyond Token Positions: Safety Alignment Across Denoising Steps in Diffusion Language Models31 Aug 2026
- 2026Accelerating Diffusion Language Models via Structured Suffix Modeling24 Aug 2026
- 2026Multi Codec Discrete Diffusion Model for Text Guided Speech Inpainting and Editing5 Aug 2026
38 more
- 2026MDLMPE: Distribution Aware Positional Encoding for Masked Diffusion Language Models4 Aug 2026
- 2026LLaDA MoE v2: Scaling Mixture-of-Experts Diffusion Language Models4 Aug 2026
- 2026DDSynth-RL: Audio Synthesizer Inversion via Discrete Diffusion with Reinforcement Learning4 Aug 2026
- 2026EchoChange: A Diffusion Language Model with Dual Pass Remasking for Factual Remote Sensing Disaster Change Captioning3 Aug 2026
- 2026Diffusion Language Model for Recommendation23 Jul 2026
- 2026Mean-to-Score Discrete Diffusion: Posterior-Mean Denoisers for Score Entropy23 Jul 2026
- 2026MotifRole-Diff: Risk-Optimal Role-Aware Corruption for Masked Molecular Graph Diffusion20 Jul 2026
- 2026Nemotron-Labs-Diffusion: A Tri-Mode Language Model Unifying Autoregressive, Diffusion, and Self-Speculation Decoding7 Jul 2026
- 2026DiLaServe: High SLO Attainment Serving for Diffusion Language Models27 Jun 2026
- 2026Bifocal Diffusion Language Models: Asymmetric Bidirectional Context for Parallel Generation26 Jun 2026
- 2026Understanding Parallel Samplers in Masked Diffusion via Random Walks on Graphs22 Jun 2026
- 2026When to Plan, When to Polish: Noise Level as a Granularity Axis for Diffusion Language Models19 Jun 2026
- 2026VoidPadding: Let [VOID] Handle Padding in Masked Diffusion Language Models so that [EOS] Can Focus on Semantic Termination16 Jun 2026
- 2026Follow the Latent Roadmap: Navigating Revocable Decoding for Diffusion LLMs with Anchor Tokens15 Jun 2026
- 2026Prefilling-dLLM: Predictive Prefilling for Long-Context Inference in Diffusion Language Models9 Jun 2026
- 2026Forward-Free Diffusion Language Models with BPTT-Free Looped Refinement6 Jun 2026
- 2026Fast-dLLM++: Fr'echet Profile Decoding for Faster Diffusion LLM Inference1 Jun 2026
- 2026SimSD: Simple Speculative Decoding in Diffusion Language Models1 Jun 2026
- 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplers1 Jun 2026
- 2026DFlare: Scaling Up Draft Capacity for Block Diffusion Speculative Decoding1 Jun 2026
- 2026DLM-SWAI: Steering Diffusion Language Models Before They Unmask28 May 2026
- 2026Looped Diffusion Language Models25 May 2026
- 2026Extracting Training Data from Diffusion Language Models via Infilling22 May 2026
- 2026Backdooring Masked Diffusion Language Models19 May 2026
- 2026Continuous Diffusion Scales Competitively with Discrete Diffusion for Language18 May 2026
- 2026When Latent Geometry Is Not Enough: Draft-Conditioned Latent Refinement for Non-Autoregressive Text Generation15 May 2026
- 2026BlockVLA: Accelerating Autoregressive VLA via Block Diffusion Finetuning13 May 2026
- 2026Adaptive Steering and Remasking for Safe Generation in Diffusion Language Models13 May 2026
- 2026Understanding and Accelerating the Training of Masked Diffusion Language Models13 May 2026
- 2026Mixing Times of Glauber Dynamics on Masked Language Models11 May 2026
- 2026ELF: Embedded Language Flows11 May 2026
- 2026Dystruct: Dynamically Structured Diffusion Language Model Decoding via Bayesian Inference10 May 2026
- 2026Trajectory as the Teacher: Few-Step Discrete Flow Matching via Energy-Navigated Distillation8 May 2026
- 2026Computer-Aided Design Generation by Cascaded Discrete Diffusion Model6 May 2026
- 2026Towards A Generative Protein Evolution Machine with DPLM-Evo30 Apr 2026
- 2026Measuring Temporal Linguistic Emergence in Diffusion Language Models25 Apr 2026
- 2026NI Sampling: Accelerating Discrete Diffusion Sampling by Token Order Optimization20 Apr 2026
- 2026LangFlow: Continuous Diffusion Rivals Discrete in Language Modeling13 Apr 2026
50 newest of 214 All 214 (JSON)
- 2025Large Language Diffusion Modelscited by 912
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2025Dream 7B: Diffusion Large Language Modelscited by 544
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 381
- 2024Discrete Flow Matchingcited by 317
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Modelscited by 274
- 2023DIFUSCO: Graph-based Diffusion Solvers for Combinatorial Optimizationcited by 221
- 2024Scaling up Masked Diffusion Models on Textcited by 199
- 2023A Reparameterized Discrete Diffusion Model for Text Generationcited by 152
12 most cited of 157 All 157 (JSON)
- 2026Prefilling-dLLM v2: Learned Chunk Routing for Long-Context Diffusion Language Models13 Oct 2026
- 2026BiMoGen: Bidirectional Motion-Text Generation via Unified Masked Discrete Diffusion28 Sep 2026
- 2026Does Uniform Discrete Diffusion Need Time?25 Sep 2026
- 2026Why Jailbreaks Succeed in Diffusion Language Models: An Energy Landscape Analysis25 Sep 2026
- 2026Representation-based Masked Diffusion Model11 Sep 2026
- 2026In-Place Instruction Following in Diffusion Language Models7 Sep 2026
- 2026Beyond Token Positions: Safety Alignment Across Denoising Steps in Diffusion Language Models31 Aug 2026
- 2026Accelerating Diffusion Language Models via Structured Suffix Modeling24 Aug 2026
- 2026Multi Codec Discrete Diffusion Model for Text Guided Speech Inpainting and Editing5 Aug 2026
- 2026MDLMPE: Distribution Aware Positional Encoding for Masked Diffusion Language Models4 Aug 2026
- 2026LLaDA MoE v2: Scaling Mixture-of-Experts Diffusion Language Models4 Aug 2026
- 2026DDSynth-RL: Audio Synthesizer Inversion via Discrete Diffusion with Reinforcement Learning4 Aug 2026
12 newest of 157 All 157 (JSON)
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
- 2025Attention Is All You Need for KV Cache in Diffusion LLMscited by 32
- 2025Self Speculative Decoding for Diffusion Large Language Modelscited by 30
- 2026DFlare: Scaling Up Draft Capacity for Block Diffusion Speculative Decodingcited by 9
- 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Modelscited by 5
- 2026Gumbel Distillation for Parallel Text Generationcited by 1
- 2026DFlare: Scaling Up Draft Capacity for Block Diffusion Speculative Decoding1 Jun 2026
- 2026Gumbel Distillation for Parallel Text Generation23 Mar 2026
- 2025Attention Is All You Need for KV Cache in Diffusion LLMs16 Oct 2025
- 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Models6 Oct 2025
- 2025Self Speculative Decoding for Diffusion Large Language Models5 Oct 2025
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding28 May 2025
- 2025Jailbreaking Large Language Diffusion Models: Revealing Hidden Safety Flaws in Diffusion-Based Text Generationcited by 26
- 2026CoDAR: Continuous Diffusion Language Models are More Powerful Than You Thinkcited by 14
- 2025RFG: Test-Time Scaling for Diffusion Large Language Model Reasoning with Reward-Free Guidancecited by 7
- 2025Beyond Surface Reasoning: Unveiling the True Long Chain-of-Thought Capacity of Diffusion Large Language Modelscited by 4
- 2026BlockVLA: Accelerating Autoregressive VLA via Block Diffusion Finetuningcited by 3
- 2025ExLM: Rethinking the Impact of [MASK] Tokens in Masked Language Modelscited by 2
- 2026BlockVLA: Accelerating Autoregressive VLA via Block Diffusion Finetuning13 May 2026
- 2026CoDAR: Continuous Diffusion Language Models are More Powerful Than You Think3 Mar 2026
- 2025Beyond Surface Reasoning: Unveiling the True Long Chain-of-Thought Capacity of Diffusion Large Language Models10 Oct 2025
- 2025RFG: Test-Time Scaling for Diffusion Large Language Model Reasoning with Reward-Free Guidance29 Sep 2025
- 2025Jailbreaking Large Language Diffusion Models: Revealing Hidden Safety Flaws in Diffusion-Based Text Generation25 Jul 2025
- 2025ExLM: Rethinking the Impact of [MASK] Tokens in Masked Language Models23 Jan 2025
- 2025From Masks to Worlds: A Hitchhiker's Guide to World Models23 Oct 2025
- 2023BoDiffusion: Diffusing Sparse Observations for Full-Body Human Motion Synthesiscited by 45
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMscited by 34
- 2024Diff-RNTraj: A Structure-aware Diffusion Model for Road Network-constrained Trajectory Generationcited by 29
- 2026Nemotron-Labs-Diffusion: A Tri-Mode Language Model Unifying Autoregressive, Diffusion, and Self-Speculation Decodingcited by 24
- 2025Quantization Meets dLLMs: A Systematic Study of Post-training Quantization for Diffusion LLMscited by 23
- 2025Unifying Autoregressive and Diffusion-Based Sequence Generationcited by 17
- 2023IPAD: Iterative, Parallel, and Diffusion-based Network for Scene Text Recognitioncited by 12
- 2025Quant-dLLM: Post-Training Extreme Low-Bit Quantization for Diffusion Large Language Modelscited by 12
- 2024Transformers Can Navigate Mazes With Multi-Step Predictioncited by 11
- 2025Generative or Discriminative? Revisiting Text Classification in the Era of Transformerscited by 9
- 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusingcited by 7
- 2026Noise Scheduling as Information-Guided Allocation in Diffusion Trainingcited by 7
12 most cited of 38 All 38 (JSON)
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMs13 Oct 2026
- 2026Entropy Can Flow, or It Can Guide. Be Entropy. LEDFlow: Introducing Entropy-guided Generation Order into Uniform Discrete Flow20 Sep 2026
- 2026Efficient One-to-Many Translation with Joint Multi-Stream Diffusion14 Sep 2026
- 2026Nemotron-Labs-Diffusion: A Tri-Mode Language Model Unifying Autoregressive, Diffusion, and Self-Speculation Decoding7 Jul 2026
- 2026Follow the Latent Roadmap: Navigating Revocable Decoding for Diffusion LLMs with Anchor Tokens15 Jun 2026
- 2026Fast-dLLM++: Fr'echet Profile Decoding for Faster Diffusion LLM Inference1 Jun 2026
- 2026When Latent Geometry Is Not Enough: Draft-Conditioned Latent Refinement for Non-Autoregressive Text Generation15 May 2026
- 2026Mixing Times of Glauber Dynamics on Masked Language Models11 May 2026
- 2026Kaiwu-PyTorch-Plugin: Bridging Deep Learning and Photonic Quantum Computing for Energy-Based Models and Active Sample Selection22 Feb 2026
- 2026Noise Scheduling as Information-Guided Allocation in Diffusion Training20 Feb 2026
- 2026Causality in Video Diffusers is Separable from Denoising10 Feb 2026
- 2026AR-MAP: Are Autoregressive Large Language Models Implicit Teachers for Diffusion Large Language Models?2 Feb 2026
12 newest of 38 All 38 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025Discrete Diffusion Models for Language Generation
- 2024Masked Diffusion Models are Secretly Time-Agnostic Masked Models and Exploit Inaccurate Categorical Sampling
- 2026Diffusion Language Models: An Experimental Analysis
- 2025Simple Denoising Diffusion Language Models
- 2025Improving Text Style Transfer using Masked Diffusion Language Models with Inference-time Scaling
- 2026How to Train Your Latent Diffusion Language Model Jointly With the Latent Space
- 2026Unifying Masked Diffusion Models with Various Generation Orders and Beyond
- 2022Empowering Diffusion Models on the Embedding Space for Text Generation