Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

RDEIC: Accelerating Diffusion-Based Extreme Image Compression with Relay Residual Diffusion

About

Diffusion-based extreme image compression methods have achieved impressive performance at extremely low bitrates. However, constrained by the iterative denoising process that starts from pure noise, these methods are limited in both fidelity and efficiency. To address these two issues, we present Relay Residual Diffusion Extreme Image Compression (RDEIC), which leverages compressed feature initialization and residual diffusion. Specifically, we first use the compressed latent features of the image with added noise, instead of pure noise, as the starting point to eliminate the unnecessary initial stages of the denoising process. Second, we directly derive a novel residual diffusion equation from Stable Diffusion's original diffusion equation that reconstructs the raw image by iteratively removing the added noise and the residual between the compressed and target latent features. In this way, we effectively combine the efficiency of residual diffusion with the powerful generative capability of Stable Diffusion. Third, we propose a fixed-step fine-tuning strategy to eliminate the discrepancy between the training and inference phases, thereby further improving the reconstruction quality. Extensive experiments demonstrate that the proposed RDEIC achieves state-of-the-art visual quality and outperforms existing diffusion-based extreme image compression methods in both fidelity and efficiency. The source code and pre-trained models are available at https://github.com/huai-chang/RDEIC.

Zhiyuan Li, Yanhui Zhou, Hao Wei, Chenyang Ge, Ajmal Mian• 2024

Related benchmarks

TaskDatasetResultRank
Image CompressionKodak (test)--
32
Image CompressionDIV2K (test)
BD-DISTS81.12
9
Image CompressionCLIC 2020 (test)
BD-DISTS78.86
9
Showing 3 of 3 rows

Other info

Follow for update