Lossy Image Compression with Quantized Hierarchical VAEs
Not on the map. Why?
Abstract (arXiv)
Recent research has shown a strong theoretical connection between variational autoencoders (VAEs) and the rate-distortion theory. Motivated by this, we consider the problem of lossy image compression from the perspective of generative modeling. Starting with ResNet VAEs, which are originally designed for data (image) distribution modeling, we redesign their latent variable model using a quantization-aware posterior and prior, enabling easy quantization and entropy coding at test time. Along with improved neural network architecture, we present a powerful and efficient model that outperforms previous methods on natural image lossy compression. Our model compresses images in a coarse-to-fine fashion and supports parallel encoding and decoding, leading to fast execution on GPUs. Code is available at https://github.com/duanzhiihao/lossy-vae.
Key facts. Cited by 35 papers in the map (data through 14 Oct 2026). Builds on 43 of its 58 references that are in the map. First posted on arXiv 27 Aug 2022.
Builds on
43 of 58 references.
References · 43
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2017Neural Discrete Representation Learningcited by 4,712
- 2013Auto-Encoding Variational Bayescited by 4,053
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2022A ConvNet for the 2020scited by 3,579
- 2022RePaint: Inpainting using Denoising Diffusion Probabilistic Modelscited by 1,430
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,387
- 2016Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Networkcited by 1,204
- 2018Variational image compression with a scale hyperpriorcited by 935
- 2018Joint Autoregressive and Hierarchical Priors for Learned Image Compressioncited by 664
31 more
- 2016Improving Variational Inference with Inverse Autoregressive Flowcited by 640
- 2020NVAE: A Deep Hierarchical Variational Autoencodercited by 637
- 2020Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modulescited by 590
- 2016Ladder Variational Autoencoderscited by 318
- 2017Lossy Image Compression with Compressive Autoencoderscited by 294
- 2020CompressAI: a PyTorch library and evaluation platform for end-to-end compression researchcited by 276
- 2020Channel-wise Autoregressive Entropy Models for Learned Image Compressioncited by 264
- 2020Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Imagescited by 264
- 2021Checkerboard Context Model for Efficient Learned Image Compressioncited by 202
- 2022The Devil Is in the Details: Window-based Attention for Image Compressioncited by 147
- 2020Nonlinear Transform Codingcited by 136
- 2016Discrete Variational Autoencoderscited by 125
- 2021Enhanced Invertible Encoding for Learned Image Compressioncited by 105
- 2019Non-local Attention Optimized Deep Image Compressioncited by 92
- 2019Practical Lossless Compression with Latent Variables using Bits Back Codingcited by 86
- 2016End-to-end optimization of nonlinear transform codes for perceptual qualitycited by 73
- 2020Universally Quantized Neural Compressioncited by 64
- 2021Transformer-based Image Compressioncited by 62
- 2020Hierarchical Quantized Autoencoderscited by 57
- 2019Bit-Swap: Recursive Bits-Back Coding for Lossless Compression with Hierarchical Latent Variablescited by 52
- 2021Consistency Regularization for Variational Auto-Encoderscited by 51
- 2020Learning Accurate Entropy Model with Global Reference for Image Compressioncited by 45
- 2020Compressing Images by Encoding Their Latent Representations with Relative Entropy Codingcited by 39
- 2019HiLLoC: Lossless Image Compression with Hierarchical Latent Variable Modelscited by 37
- 2021Algorithms for the Communication of Samplescited by 34
- 2021On the Out-of-distribution Generalization of Probabilistic Image Modellingcited by 33
- 2018DVAE++: Discrete Variational Autoencoders with Overlapping Transformationscited by 32
- 2022Neural Data-Dependent Transform for Learned Image Compressioncited by 30
- 2018DVAE#: Discrete Variational Autoencoders with Relaxed Boltzmann Priorscited by 20
- 2021Towards Empirical Sandwich Bounds on the Rate-Distortion Functioncited by 14
- 2020Relaxed-Responsibility Hierarchical Discrete VAEscited by 4
Built on by
Citing papers per year: 2023: 6; 2024: 15; 2025: 11; 2026: 3.
35 papers.
Sort:
By area:
- 2023MLIC++: Linear Complexity Multi-Reference Entropy Modeling for Learned Image Compressioncited by 58
- 2023QARV: Quantization-Aware ResNet VAE for Lossy Image Compressioncited by 32
- 2024MambaVC: Learned Visual Compression with Selective State Spacescited by 28
- 2024Diffusion-Driven Semantic Communication for Generative Models with Bandwidth Constraintscited by 19
- 2024All-in-One Image Coding for Joint Human-Machine Vision with Multi-Path Aggregationcited by 14
- 2023Dynamic Low-Rank Instance Adaptation for Universal Neural Image Compressioncited by 10
- 2024Progressive Compression with Universally Quantized Diffusion Modelscited by 10
- 2025GIViC: Generative Implicit Video Compressioncited by 10
- 2024Generalized Nested Latent Variable Models for Lossy Coding applied to Wind Turbine Scenarioscited by 7
- 2024Another Way to the Top: Exploit Contextual Clustering in Learned Image Codingcited by 6
- 2024STanH : Parametric Quantization for Variable Rate Learned Image Compressioncited by 4
- 2024JPEG AI Image Compression Visual Artifacts: Detection Methods and Datasetcited by 4
23 more
- 2025Flexible Mixed Precision Quantization for Learned Image Compressioncited by 4
- 2024Exploring adversarial robustness of JPEG AI: methodology, comparison and new methodscited by 3
- 2024Learned Image Transmission with Hierarchical Variational Autoencodercited by 2
- 2024High-Efficiency Neural Video Compression via Hierarchical Predictive Learningcited by 2
- 2025Lightweight Embedded FPGA Deployment of Learned Image Compression with Knowledge Distillation and Hybrid Quantizationcited by 2
- 2026CodecSplat: Ultra-Compact Latent Coding for Feed-Forward 3D Gaussian Splattingcited by 2
- 2023Generative AI for Rapid Diffusion MRI with Improved Image Quality, Reliability and Generalizabilitycited by 1
- 2023CPIPS: Learning to Preserve Perceptual Distances in End-to-End Image Compressioncited by 1
- 2023Effortless Cross-Platform Video Codec: A Codebook-Based Methodcited by 1
- 2024Learned Image Compression with Text Quality Enhancementcited by 1
- 2025Gone With the Bits: Revealing Racial Bias in Low-Rate Neural Compression for Facial Imagescited by 1
- 2025Structured Pruning and Quantization for Learned Image Compressioncited by 1
- 2025NIC-RobustBench: A Comprehensive Open-Source Toolkit for Neural Image Compression and Robustness Analysiscited by 1
- 2025Minimal High-Resolution Patches Are Sufficient for Whole Slide Image Representation via Cascaded Dual-Scale Reconstructioncited by 1
- 2024CIC: Circular Image Compressioncited by 0
- 2024Beta-Sigma VAE: Separating beta and decoder variance in Gaussian variational autoencodercited by 0
- 2024Edge-based Denoising Image Compressioncited by 0
- 2025Interleaved Block-based Learned Image Compression with Feature Enhancement and Quantization Error Compensationcited by 0
- 2025Support is All You Need for Certified VAE Trainingcited by 0
- 2025TerraCodec: Compressing Optical Earth Observation Datacited by 0
- 2025Adaptive Dataset Quantization: A New Direction for Dataset Pruningcited by 0
- 2026ARCHE: Autoregressive Residual Compression with Hyperprior and Excitationcited by 0
- 2026Every Packet Counts: Dispersing Information for Loss-Resilient Learned Image Compressioncited by 0
- 2026Every Packet Counts: Dispersing Information for Loss-Resilient Learned Image Compression11 Aug 2026
- 2026CodecSplat: Ultra-Compact Latent Coding for Feed-Forward 3D Gaussian Splatting25 May 2026
- 2026ARCHE: Autoregressive Residual Compression with Hyperprior and Excitation10 Mar 2026
- 2025Adaptive Dataset Quantization: A New Direction for Dataset Pruning28 Nov 2025
- 2025TerraCodec: Compressing Optical Earth Observation Data14 Oct 2025
- 2025Minimal High-Resolution Patches Are Sufficient for Whole Slide Image Representation via Cascaded Dual-Scale Reconstruction3 Aug 2025
- 2025NIC-RobustBench: A Comprehensive Open-Source Toolkit for Neural Image Compression and Robustness Analysis23 Jun 2025
- 2025Structured Pruning and Quantization for Learned Image Compression2 Jun 2025
- 2025Flexible Mixed Precision Quantization for Learned Image Compression2 Jun 2025
- 2025Gone With the Bits: Revealing Racial Bias in Low-Rate Neural Compression for Facial Images5 May 2025
- 2025Support is All You Need for Certified VAE Training16 Apr 2025
- 2025GIViC: Generative Implicit Video Compression25 Mar 2025
23 more
- 2025Lightweight Embedded FPGA Deployment of Learned Image Compression with Knowledge Distillation and Hybrid Quantization5 Mar 2025
- 2025Interleaved Block-based Learned Image Compression with Feature Enhancement and Quantization Error Compensation21 Feb 2025
- 2024Progressive Compression with Universally Quantized Diffusion Models14 Dec 2024
- 2024Exploring adversarial robustness of JPEG AI: methodology, comparison and new methods18 Nov 2024
- 2024JPEG AI Image Compression Visual Artifacts: Detection Methods and Dataset11 Nov 2024
- 2024High-Efficiency Neural Video Compression via Hierarchical Predictive Learning3 Oct 2024
- 2024STanH : Parametric Quantization for Variable Rate Learned Image Compression1 Oct 2024
- 2024All-in-One Image Coding for Joint Human-Machine Vision with Multi-Path Aggregation29 Sep 2024
- 2024Edge-based Denoising Image Compression17 Sep 2024
- 2024Beta-Sigma VAE: Separating beta and decoder variance in Gaussian variational autoencoder14 Sep 2024
- 2024Learned Image Transmission with Hierarchical Variational Autoencoder29 Aug 2024
- 2024Diffusion-Driven Semantic Communication for Generative Models with Bandwidth Constraints26 Jul 2024
- 2024CIC: Circular Image Compression18 Jul 2024
- 2024Generalized Nested Latent Variable Models for Lossy Coding applied to Wind Turbine Scenarios10 Jun 2024
- 2024MambaVC: Learned Visual Compression with Selective State Spaces24 May 2024
- 2024Learned Image Compression with Text Quality Enhancement13 Feb 2024
- 2024Another Way to the Top: Exploit Contextual Clustering in Learned Image Coding21 Jan 2024
- 2023Effortless Cross-Platform Video Codec: A Codebook-Based Method16 Oct 2023
- 2023CPIPS: Learning to Preserve Perceptual Distances in End-to-End Image Compression1 Oct 2023
- 2023Dynamic Low-Rank Instance Adaptation for Universal Neural Image Compression15 Aug 2023
- 2023MLIC++: Linear Complexity Multi-Reference Entropy Modeling for Learned Image Compression28 Jul 2023
- 2023Generative AI for Rapid Diffusion MRI with Improved Image Quality, Reliability and Generalizability10 Mar 2023
- 2023QARV: Quantization-Aware ResNet VAE for Lossy Image Compression16 Feb 2023
- 2023MLIC++: Linear Complexity Multi-Reference Entropy Modeling for Learned Image Compressioncited by 58
- 2023QARV: Quantization-Aware ResNet VAE for Lossy Image Compressioncited by 32
- 2024All-in-One Image Coding for Joint Human-Machine Vision with Multi-Path Aggregationcited by 14
- 2023Dynamic Low-Rank Instance Adaptation for Universal Neural Image Compressioncited by 10
- 2025GIViC: Generative Implicit Video Compressioncited by 10
- 2024Generalized Nested Latent Variable Models for Lossy Coding applied to Wind Turbine Scenarioscited by 7
- 2024Another Way to the Top: Exploit Contextual Clustering in Learned Image Codingcited by 6
- 2024STanH : Parametric Quantization for Variable Rate Learned Image Compressioncited by 4
- 2024JPEG AI Image Compression Visual Artifacts: Detection Methods and Datasetcited by 4
- 2025Flexible Mixed Precision Quantization for Learned Image Compressioncited by 4
- 2024Exploring adversarial robustness of JPEG AI: methodology, comparison and new methodscited by 3
- 2024Learned Image Transmission with Hierarchical Variational Autoencodercited by 2
12 most cited of 31 All 31 (JSON)
- 2026Every Packet Counts: Dispersing Information for Loss-Resilient Learned Image Compression11 Aug 2026
- 2026CodecSplat: Ultra-Compact Latent Coding for Feed-Forward 3D Gaussian Splatting25 May 2026
- 2026ARCHE: Autoregressive Residual Compression with Hyperprior and Excitation10 Mar 2026
- 2025Adaptive Dataset Quantization: A New Direction for Dataset Pruning28 Nov 2025
- 2025TerraCodec: Compressing Optical Earth Observation Data14 Oct 2025
- 2025Minimal High-Resolution Patches Are Sufficient for Whole Slide Image Representation via Cascaded Dual-Scale Reconstruction3 Aug 2025
- 2025NIC-RobustBench: A Comprehensive Open-Source Toolkit for Neural Image Compression and Robustness Analysis23 Jun 2025
- 2025Structured Pruning and Quantization for Learned Image Compression2 Jun 2025
- 2025Flexible Mixed Precision Quantization for Learned Image Compression2 Jun 2025
- 2025Gone With the Bits: Revealing Racial Bias in Low-Rate Neural Compression for Facial Images5 May 2025
- 2025Support is All You Need for Certified VAE Training16 Apr 2025
- 2025GIViC: Generative Implicit Video Compression25 Mar 2025
12 newest of 31 All 31 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2022Lossy Image Compression with Conditional Diffusion Models
- 2026Latent-Compressed Variational Autoencoder for Video Diffusion Models
- 2025Hierarchical Vector-Quantized Latents for Perceptual Low-Resolution Video Compression
- 2023Extreme Image Compression using Fine-tuned VQGANs
- 2023Improving Statistical Fidelity for Neural Image Compression with Implicit Local Likelihood Models
- 2026Is Hierarchical Quantization Essential for Optimal Reconstruction?
- 2026Tree-VQ: Progressive Image Compression from Pretrained Vector Quantizers
- 2025Generative Latent Coding for Ultra-Low Bitrate Image Compression