Partition Generative Modeling: Masked Modeling Without Masks
Justin Deschenaux, Lan Tran, Caglar Gulcehre
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper3
- The Diffusion Duality, Chapter II: ψ-Samplers and Efficient CurriculumJustin Deschenaux, Caglar Gulcehre, Subham Sekhar SahooICLR 2026 · 被引用 14 次
- Loopholing Discrete Diffusion: Deterministic Bypass of the Sampling WallMingyu Jo, Jaesik Yoon, Justin Deschenaux, Caglar Gulcehre 等ICLR 2026 · 被引用 7 次
- Efficient Autoregressive Inference for Transformer Probabilistic ModelsConor Hassan, Nasrulloh R. B. S. Loka, Cen-You Li, Daolang Huang 等ICLR 2026 · 被引用 5 次
它引用的顶会 Paper28
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 被引用 35,902 次
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 被引用 5,568 次
- Tree of Thoughts: Deliberate Problem Solving with Large Language ModelsShunyu Yao, Dian Yu, Jeffrey Zhao, Izhak Shafran 等NeurIPS 2023 · 被引用 5,068 次
- Self-Refine: Iterative Refinement with Self-FeedbackAman Madaan, Niket Tandon, Prakhar Gupta, Skyler Hallinan 等NeurIPS 2023 · 被引用 4,972 次
- The Curious Case of Neural Text DegenerationAri Holtzman, Jan Buys, Li Du, Maxwell Forbes 等ICLR 2020 · 被引用 4,112 次
相关 Paper
- Parallel Sampling from Masked Diffusion Models via Conditional Independence TestingIskander Azangulov, Teodora Pandeva, Niranjani Prasad, Javier Zazo 等ICLR 2026 · 被引用 13 次
- Beyond Masked and Unmasked: Discrete Diffusion Models via Partial MaskingChen-Hao Chao, Wei-Fang Sun, Hanwen Liang, Chun-Yi Lee 等NeurIPS 2025 · 被引用 21 次
- 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 次
- Plan for Speed: Dilated Scheduling for Masked Diffusion Language ModelsOmer Luxembourg, Haim Permuter, Eliya NachmaniICML 2026 · 被引用 30 次
