Partition Generative Modeling: Masked Modeling Without Masks
Justin Deschenaux, Lan Tran, Caglar Gulcehre
Abstract
Masked generative models (MGMs) can generate tokens in parallel and in any order, unlike autoregressive models (ARMs), which decode one token at a time, left-to-right. However, MGMs process the full-length sequence at every sampling step, including tokens that carry no information. In contrast, ARMs process only the previously generated tokens. We introduce ``Partition Generative Models'' (PGMs), which replace masking with partitioning. Tokens are split into two groups that cannot attend to each other, and the model learns to predict each group conditioned on the other, eliminating mask tokens entirely. Because the groups do not interact, PGMs can process only the clean tokens during sampling, like ARMs, while retaining parallel, any-order generation, like MGMs. On OpenWebText, PGMs achieve higher throughput than MDLM while producing samples with lower Generative Perplexity. On ImageNet, PGMs reach comparable FID to MaskGIT with a throughput improvement. With twice as many steps, the FID improves to 4.56 while remaining faster than MGMs. Finally, PGMs remain compatible with existing MGM samplers and distillation methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1af29de0-7cbc-488f-9e94-0b8d031bad10Cited by top-tier papers3
- The Diffusion Duality, Chapter II: ψ-Samplers and Efficient CurriculumJustin Deschenaux, Caglar Gulcehre, Subham Sekhar SahooICLR 2026 · 14 citations
- Loopholing Discrete Diffusion: Deterministic Bypass of the Sampling WallMingyu Jo, Jaesik Yoon, Justin Deschenaux, Caglar Gulcehre et al.ICLR 2026 · 7 citations
- Efficient Autoregressive Inference for Transformer Probabilistic ModelsConor Hassan, Nasrulloh R. B. S. Loka, Cen-You Li, Daolang Huang et al.ICLR 2026 · 5 citations
Builds on28
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
- Tree of Thoughts: Deliberate Problem Solving with Large Language ModelsShunyu Yao, Dian Yu, Jeffrey Zhao, Izhak Shafran et al.NeurIPS 2023 · 5,068 citations
- Self-Refine: Iterative Refinement with Self-FeedbackAman Madaan, Niket Tandon, Prakhar Gupta, Skyler Hallinan et al.NeurIPS 2023 · 4,972 citations
- The Curious Case of Neural Text DegenerationAri Holtzman, Jan Buys, Li Du, Maxwell Forbes et al.ICLR 2020 · 4,112 citations
Related papers
- Parallel Sampling from Masked Diffusion Models via Conditional Independence TestingIskander Azangulov, Teodora Pandeva, Niranjani Prasad, Javier Zazo et al.ICLR 2026 · 13 citations
- Beyond Masked and Unmasked: Discrete Diffusion Models via Partial MaskingChen-Hao Chao, Wei-Fang Sun, Hanwen Liang, Chun-Yi Lee et al.NeurIPS 2025 · 21 citations
- Infinite Mask Diffusion for Few-Step DistillationJaehoon Yoo, Wonjung Kim, Chanhyuk Lee, Seunghoon HongICML 2026
- Gumbel Distillation for Parallel Text GenerationChi Zhang, Xixi Hu, Bo Liu, qiang liuICLR 2026 · 2 citations
- Plan for Speed: Dilated Scheduling for Masked Diffusion Language ModelsOmer Luxembourg, Haim Permuter, Eliya NachmaniICML 2026 · 30 citations
