arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.04081cs.CVcs.AIeess.IV

Epsilon-VAE:将去噪作为视觉解码

Epsilon-VAE: Denoising as Visual Decoding

City University of Hong Kong(香港城市大学)

Long Zhao, Sanghyun Woo, Ziyu Wan, Yandong Li, Han Zhang, Boqing Gong, Hartwig Adam, Xuhui Jia, Ting Liu

AI总结:

提出Epsilon-VAE,将自编码器的解码器替换为扩散过程进行迭代去噪重建,在相同压缩率下提升下游生成质量22%,或通过提高压缩率实现2.3倍推理加速。

摘要:

在生成建模中,分词将复杂数据简化为紧凑的结构化表示,从而创建更高效、可学习的空间。对于高维视觉数据,它减少了冗余并强调了高质量生成的关键特征。当前视觉分词方法依赖传统的自编码器框架,编码器将数据压缩为潜在表示,解码器重建原始输入。在本研究中,我们提出将去噪作为解码的新视角,从单步重建转向迭代细化。具体而言,我们用扩散过程替换解码器,该过程在编码器提供的潜在表示引导下,迭代细化噪声以恢复原始图像。我们通过评估重建质量(rFID)和生成质量(FID),并与最先进的自编码方法进行比较来评估我们的方法。通过采用基于扩散的迭代重建,我们的自编码器即Epsilon-VAE实现了高重建质量,进而在相同压缩率下将下游生成质量提升22%,或通过提高压缩率实现2.3倍的推理加速。我们希望这项工作为整合迭代生成与自编码以改善压缩和生成提供新见解。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑