正在加载视频...

视频加载失败

Diffusion models generate high-quality images but require hundreds of forward passes. MIT CSAIL and Adobe Research introduce Distribution Matching Distillation (DMD), a distillation approach that converts costly multi-step diffusion models into fast one-step generators. A thread 🧵

34,383 次观看 • 2 年前 •via X (Twitter)

9 条评论

MIT CSAIL 的头像
MIT CSAIL2 年前

DMD trains a one-step generator that maps random noise into realistic images, consisting of two key components. First up: it uses a regression loss to anchor the mapping process, ensuring a coarse organization of the image space, enhancing the stability of the training phase.

MIT CSAIL 的头像
MIT CSAIL2 年前

Additionally, it employs a distribution matching loss to guarantee that the likelihood of generating a specific image w/the student model aligns w/its actual frequency of occurrence in the real world.

MIT CSAIL 的头像
MIT CSAIL2 年前

The gradient of this loss is formulated as the difference between two diffusion models’ output, trained on real and fake samples respectively.

MIT CSAIL 的头像
MIT CSAIL2 年前

DMD achieves a strong 11.49 FID on zero-shot COCO-30K, comparable to Stable Diffusion v1.5 while being 30X faster. Compared to previous approaches, it notably balances image quality with sample diversity.

MIT CSAIL 的头像
MIT CSAIL2 年前

DMD paves the way for real-time visual generation. This same approach could improve diffusion-based generative models across various fields, from design, to scientific discovery and beyond, by significantly enhancing speed and effectiveness.

MIT CSAIL 的头像
MIT CSAIL2 年前

Paper: Authors: @TianweiY, @m_gharbi, @rzhang88, @elishechtman, @fredodurand, Bill Freeman, and Taesung Park. Project page: MIT News:

menguzat 的头像
menguzat2 年前

@AdobeResearch will you release the code / model for this?

Prashant 的头像
Prashant2 年前

@AdobeResearch Could this approach of distribution matching loss be applied to other generative AI tasks besides image generation? For example, text generation or music composition?

𝗦𝗼𝘂𝗹𝘀𝗳𝗲𝗻𝗴 𝗡𝗲𝘄 𝗬𝗼𝗿𝗸 的头像
𝗦𝗼𝘂𝗹𝘀𝗳𝗲𝗻𝗴 𝗡𝗲𝘄 𝗬𝗼𝗿𝗸2 年前

@rzhang88 @AdobeResearch Good for you my friend, we are try use your model colorization(which is 4 years ago) for sneakers now, lol, thank you for everything.

相关视频