Controlling Rate, Distortion, and Realism: Towards a Single Comprehensive Neural Image Compression Model
About
In recent years, neural network-driven image compression (NIC) has gained significant attention. Some works adopt deep generative models such as GANs and diffusion models to enhance perceptual quality (realism). A critical obstacle of these generative NIC methods is that each model is optimized for a single bit rate. Consequently, multiple models are required to compress images to different bit rates, which is impractical for real-world applications. To tackle this issue, we propose a variable-rate generative NIC model. Specifically, we explore several discriminator designs tailored for the variable-rate approach and introduce a novel adversarial loss. Moreover, by incorporating the newly proposed multi-realism technique, our method allows the users to adjust the bit rate, distortion, and realism with a single model, achieving ultra-controllability. Unlike existing variable-rate generative NIC models, our method matches or surpasses the performance of state-of-the-art single-rate generative NIC models while covering a wide range of bit rates using just one model. Code will be available at https://github.com/iwa-shi/CRDR
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Compression | DIV2K 512 | BD-PSNR0.52 | 90 | |
| Image Compression | Kodak24 512 | PSNR30.07 | 76 | |
| Image Compression | CLIC2020 512x512 (test) | BD-PSNR6.83 | 66 | |
| Image Compression | Kodak24 512x512 (test) | BD-PSNR-0.22 | 13 |