arXiv:2409.02529cs.LGcs.CV2024-09被引 5

用扩散模型提升图像压缩的清晰度和生成能力。

Sample what you cant compress

  • 将自编码器与扩散模型结合,实现连续编码解码联合学习。
  • 重建质量优于基于GAN的自编码器,且更易调参。
  • 适合需要高质量图像生成与压缩的科研与工程场景。

对于学习到的图像表示,基础自编码器常产生模糊结果。通过引入对抗损失(GAN)或感知损失可提升重建质量,但这些方法缺乏理论依据。与此同时,扩散模型在生成清晰、高质量图像方面表现卓越,并具有坚实的理论基础(如变分推断或直接作为Fisher散度研究)。本文首次将自编码器表示学习与扩散模型结合,联合学习连续编码器与解码器,采用基于扩散的损失函数,证明其可实现更高压缩率并生成更优结果。实验表明,该方法重建质量优于GAN-based自编码器,且更易调参。同时,所获得的潜在表示比当前最先进的GAN损失得到的表示更易用潜在扩散模型建模。由于解码器是随机的,能生成未编码于确定性潜在表示中的细节,因此将该方法命名为‘采样你无法压缩的内容’(Sample what you can't compress, SWYCC)。

原文摘要 · Abstract (English)

For learned image representations, basic autoencoders often produce blurry results. Reconstruction quality can be improved by incorporating additional penalties such as adversarial (GAN) and perceptual losses. Arguably, these approaches lack a principled interpretation. Concurrently, in generative settings diffusion has demonstrated a remarkable ability to create crisp, high quality results and has solid theoretical underpinnings (from variational inference to direct study as the Fisher Divergence). Our work combines autoencoder representation learning with diffusion and is, to our knowledge, the first to demonstrate jointly learning a continuous encoder and decoder under a diffusion-based loss and showing that it can lead to higher compression and better generation. We demonstrate that this approach yields better reconstruction quality as compared to GAN-based autoencoders while being easier to tune. We also show that the resulting representation is easier to model with a latent diffusion model as compared to the representation obtained from a state-of-the-art GAN-based loss. Since our decoder is stochastic, it can generate details not encoded in the otherwise deterministic latent representation; we therefore name our approach "Sample what you can't compress", or SWYCC for short.

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

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