arXiv:2410.04081cs.CVcs.AI2024-10ICML被引 17

用迭代去噪替代传统解码,提升图像压缩与生成质量

Epsilon-VAE: Denoising as Visual Decoding

  • 用扩散模型迭代去噪代替传统解码器,实现渐进式图像恢复
  • 在相同压缩率下生成质量提升22%,或实现2.3倍推理加速
  • 适合关注高效图像生成与自编码器改进的研究者

在生成建模中,分词将复杂数据简化为紧凑、结构化的表示,从而创建更高效、可学习的表示空间。对于高维视觉数据,分词能减少冗余并突出关键特征,以实现高质量生成。当前视觉分词方法依赖传统的自编码器框架:编码器将数据压缩为潜在表示,解码器重建原始输入。本文提出新视角:将去噪视为解码,从单步重建转向迭代精炼。具体地,我们用扩散过程替代解码器,通过迭代将噪声逐步精炼为原图,由编码器提供的潜在变量引导。我们在重建(rFID)和生成质量(FID)两个方面评估该方法,对比当前最优自编码方法。通过扩散驱动的迭代重建,我们的自编码器Epsilon-VAE实现了高重建质量,进而使下游生成质量在相同压缩率下提升22%,或通过提高压缩率实现2.3倍推理速度提升。我们希望本工作为整合迭代生成与自编码提供新思路。

原文摘要 · Abstract (English)

In generative modeling, tokenization simplifies complex data into compact, structured representations, creating a more efficient, learnable space. For high-dimensional visual data, it reduces redundancy and emphasizes key features for high-quality generation. Current visual tokenization methods rely on a traditional autoencoder framework, where the encoder compresses data into latent representations, and the decoder reconstructs the original input. In this work, we offer a new perspective by proposing denoising as decoding, shifting from single-step reconstruction to iterative refinement. Specifically, we replace the decoder with a diffusion process that iteratively refines noise to recover the original image, guided by the latents provided by the encoder. We evaluate our approach by assessing both reconstruction (rFID) and generation quality (FID), comparing it to state-of-the-art autoencoding approaches. By adopting iterative reconstruction through diffusion, our autoencoder, namely Epsilon-VAE, achieves high reconstruction quality, which in turn enhances downstream generation quality by 22% at the same compression rates or provides 2.3x inference speedup through increasing compression rates. We hope this work offers new insights into integrating iterative generation and autoencoding for improved compression and generation.

自编码器扩散模型图像生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。