SPMDM: Enhancing Masked Diffusion Models through Simplifying Sampling Path
–Neural Information Processing Systems
Masked diffusion models (MDMs) address these issues by enabling controllable, any-order, and parallel generation but encounter training difficulties as token prediction complexity varies with unmasked token positions. This work identifies two key characteristics of efficient MDM sampling paths: prioritizing tokens near unmasked ones and generating subsequence earlier in reasoning. We propose the Simple Path Masked Diffusion Model (SPMDM), which partitions sequences into fixed-length, non-overlapping subsequences and applies varying noise scales to learn token-level and cross-subsequence dependencies.
Neural Information Processing Systems
Jun-14-2026, 03:28:20 GMT
- Technology: