Improving Vector-Quantized Image Modeling with Latent Consistency-Matching Diffusion
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866916669046128640 |
|---|---|
| author | Nguyen, Bac Lai, Chieh-Hsin Takida, Yuhta Murata, Naoki Uesaka, Toshimitsu Ermon, Stefano Mitsufuji, Yuki |
| author_facet | Nguyen, Bac Lai, Chieh-Hsin Takida, Yuhta Murata, Naoki Uesaka, Toshimitsu Ermon, Stefano Mitsufuji, Yuki |
| contents | By embedding discrete representations into a continuous latent space, we can leverage continuous-space latent diffusion models to handle generative modeling of discrete data. However, despite their initial success, most latent diffusion methods rely on fixed pretrained embeddings, limiting the benefits of joint training with the diffusion model. While jointly learning the embedding (via reconstruction loss) and the latent diffusion model (via score matching loss) could enhance performance, end-to-end training risks embedding collapse, degrading generation quality. To mitigate this issue, we introduce VQ-LCMD, a continuous-space latent diffusion framework within the embedding space that stabilizes training. VQ-LCMD uses a novel training objective combining the joint embedding-diffusion variational lower bound with a consistency-matching (CM) loss, alongside a shifted cosine noise schedule and random dropping strategy. Experiments on several benchmarks show that the proposed VQ-LCMD yields superior results on FFHQ, LSUN Churches, and LSUN Bedrooms compared to discrete-state latent diffusion models. In particular, VQ-LCMD achieves an FID of 6.81 for class-conditional image generation on ImageNet with 50 steps. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_14758 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving Vector-Quantized Image Modeling with Latent Consistency-Matching Diffusion Nguyen, Bac Lai, Chieh-Hsin Takida, Yuhta Murata, Naoki Uesaka, Toshimitsu Ermon, Stefano Mitsufuji, Yuki Machine Learning By embedding discrete representations into a continuous latent space, we can leverage continuous-space latent diffusion models to handle generative modeling of discrete data. However, despite their initial success, most latent diffusion methods rely on fixed pretrained embeddings, limiting the benefits of joint training with the diffusion model. While jointly learning the embedding (via reconstruction loss) and the latent diffusion model (via score matching loss) could enhance performance, end-to-end training risks embedding collapse, degrading generation quality. To mitigate this issue, we introduce VQ-LCMD, a continuous-space latent diffusion framework within the embedding space that stabilizes training. VQ-LCMD uses a novel training objective combining the joint embedding-diffusion variational lower bound with a consistency-matching (CM) loss, alongside a shifted cosine noise schedule and random dropping strategy. Experiments on several benchmarks show that the proposed VQ-LCMD yields superior results on FFHQ, LSUN Churches, and LSUN Bedrooms compared to discrete-state latent diffusion models. In particular, VQ-LCMD achieves an FID of 6.81 for class-conditional image generation on ImageNet with 50 steps. |
| title | Improving Vector-Quantized Image Modeling with Latent Consistency-Matching Diffusion |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2410.14758 |