Skip to content

Commit

Permalink
u
Browse files Browse the repository at this point in the history
  • Loading branch information
dongzhuoyao committed Nov 16, 2024
1 parent 9945e6e commit 59eac2c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions index.html
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,7 @@ <h1 class="title is-1 publication-title">[MASK] is All You Need</h1>
<div class="hero-body">
<h2 class="subtitle has-text-centered" style="font-weight: bold;color:gray; margin-top: 25px;">
In generative models, two paradigms have gained attraction in various applications: next-set prediction-based Masked Generative Models and next-noise prediction-based Non-Autoregressive Models, e.g., Diffusion Models. In this work, we propose using discrete-state models to connect them and explore their scalability in the vision domain. First, we conduct a step-by-step analysis in a unified design space across two types of models including timestep-independence, noise schedule, temperature, guidance strength, etc in a scalable manner. Second, we re-cast typical discriminative tasks, e.g., image segmentation, as an unmasking process from [MASK] tokens on a discrete-state model. This enables us to perform various sampling processes, including flexible conditional sampling by only training once to model the joint distribution. All aforementioned explorations lead to our framework named Discrete Interpolants, which enables us to achieve state-of-the-art or competitive performance compared to previous discrete-state based methods in various benchmarks, like ImageNet256, MS COCO, and video dataset FaceForensics. In summary, by leveraging [MASK] in discrete-state models, we can bridge Masked Generative and Non-autoregressive Diffusion models, as well as generative and discriminative tasks.

</h2>

</div>
Expand Down

0 comments on commit 59eac2c

Please sign in to comment.