Learning Image-Adaptive Codebooks for Class-Agnostic Image Restoration
Not on the map. Why?
Abstract (arXiv)
Recent work on discrete generative priors, in the form of codebooks, has shown exciting performance for image reconstruction and restoration, as the discrete prior space spanned by the codebooks increases the robustness against diverse image degradations. Nevertheless, these methods require separate training of codebooks for different image categories, which limits their use to specific image categories only (e.g. face, architecture, etc.), and fail to handle arbitrary natural images. In this paper, we propose AdaCode for learning image-adaptive codebooks for class-agnostic image restoration. Instead of learning a single codebook for each image category, we learn a set of basis codebooks. For a given input image, AdaCode learns a weight map with which we compute a weighted combination of these basis codebooks for adaptive image restoration. Intuitively, AdaCode is a more flexible and expressive discrete generative prior than previous work. Experimental results demonstrate that AdaCode achieves state-of-the-art performance on image reconstruction and restoration tasks, including image super-resolution and inpainting.
Key facts. Cited by 14 papers in the map (data through 14 Oct 2026). Builds on 49 of its 64 references that are in the map. First posted on arXiv 10 Jun 2023.
Builds on
49 of 64 references.
References · 49
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2017Neural Discrete Representation Learningcited by 4,712
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,988
- 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,940
- 2016Perceptual Losses for Real-Time Style Transfer and Super-Resolutioncited by 2,694
- 2021SwinIR: Image Restoration Using Swin Transformercited by 1,887
- 2017Enhanced Deep Residual Networks for Single Image Super-Resolutioncited by 1,552
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
37 more
- 2021Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Datacited by 970
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 774
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 658
- 2018Free-Form Image Inpainting with Gated Convolutioncited by 594
- 2021Designing a Practical Degradation Model for Deep Blind Image Super-Resolutioncited by 529
- 2018Image Inpainting for Irregular Holes Using Partial Convolutionscited by 505
- 2018Recovering Realistic Texture in Image Super-resolution by Deep Spatial Feature Transformcited by 371
- 2020PULSE: Self-Supervised Photo Upsampling via Latent Space Exploration of Generative Modelscited by 358
- 2020Component Divide-and-Conquer for Real-World Image Super-Resolutioncited by 246
- 2022Towards Robust Blind Face Restoration with Codebook Lookup Transformercited by 231
- 2020Exploiting Deep Generative Prior for Versatile Image Restoration and Manipulationcited by 225
- 2022MAT: Mask-Aware Transformer for Large Hole Image Inpaintingcited by 218
- 2021GAN Prior Embedded Network for Blind Face Restoration in the Wildcited by 197
- 2021Unsupervised Degradation Representation Learning for Blind Super-Resolutioncited by 177
- 2019EdgeConnect: Generative Image Inpainting with Adversarial Edge Learningcited by 163
- 2020Learning Image-adaptive 3D Lookup Tables for High Performance Photo Enhancement in Real-timecited by 163
- 2020GLEAN: Generative Latent Bank for Large-Factor Image Super-Resolutioncited by 158
- 2021High-Fidelity Pluralistic Image Completion with Transformerscited by 134
- 2019StructureFlow: Image Inpainting via Structure-aware Appearance Flowcited by 118
- 2019Neural Blind Deconvolution Using Deep Priorscited by 113
- 2020Rethinking Image Inpainting via a Mutual Encoder-Decoder with Feature Equalizationscited by 102
- 2022Real-World Blind Super-Resolution via Feature Matching with Implicit High-Resolution Priorscited by 100
- 2019Learning Pyramid-Context Encoder Network for High-Quality Image Inpaintingcited by 94
- 2019Foreground-aware Image Inpaintingcited by 90
- 2022RestoreFormer: High-Quality Blind Face Restoration from Undegraded Key-Value Pairscited by 85
- 2022Incremental Transformer Structure Enhanced Image Inpainting with Masking Positional Encodingcited by 83
- 2018Image Inpainting via Generative Multi-column Convolutional Neural Networkscited by 81
- 2021Flow-based Kernel Prior with Application to Blind Super-Resolutioncited by 74
- 2022AdaInt: Learning Adaptive Intervals for 3D Lookup Tables on Real-time Image Enhancementcited by 70
- 2021Image Inpainting via Conditional Texture and Structure Dual Generationcited by 66
- 2017UV-GAN: Adversarial Facial UV Map Completion for Pose-invariant Face Recognitioncited by 62
- 2018SPG-Net: Segmentation Prediction and Guidance Network for Image Inpaintingcited by 58
- 2022SepLUT: Separable Image-adaptive Lookup Tables for Real-time Image Enhancementcited by 50
- 2022Reduce Information Loss in Transformers for Pluralistic Image Inpaintingcited by 36
- 20224D LUT: Learnable Context-Aware 4D Lookup Table for Image Enhancementcited by 34
- 2018Learning Hierarchical Semantic Image Manipulation through Structured Representationscited by 27
- 2022KXNet: A Model-Driven Deep Neural Network for Blind Super-Resolutioncited by 18
Built on by
Citing papers per year: 2023: 2; 2024: 4; 2025: 5; 2026: 3.
14 papers.
Sort:
By area:
- 2023AutoDIR: Automatic All-in-One Image Restoration with Latent Diffusioncited by 100
- 2024Restoring Images in Adverse Weather Conditions via Histogram Transformercited by 73
- 2025Learned Image Compression with Dictionary-based Entropy Modelcited by 30
- 2024HybridFlow: Infusing Continuity into Masked Codebook for Extreme Low-Bitrate Image Compressioncited by 14
- 2025TVC: Tokenized Video Compression with Ultra-Low Bit Ratecited by 7
- 2023Neural Image Compression Using Masked Sparse Visual Representationcited by 5
- 2024Monocular Identity-Conditioned Facial Reflectance Reconstructioncited by 5
- 2025Towards Improved Text-Aligned Codebook Learning: Multi-Hierarchical Codebook-Text Alignment with Long Textcited by 4
- 2025Texture Vector-Quantization and Reconstruction Aware Prediction for Generative Super-Resolutioncited by 3
- 2024GameIR: A Large-Scale Synthesized Ground-Truth Dataset for Image Restoration over Gaming Contentcited by 2
- 2025mathttM3VIR: A Large-Scale Multi-Modality Multi-View Synthesized Benchmark Dataset for Image Restoration and Content Creationcited by 0
- 2026LooC: Effective Low-Dimensional Codebook for Compositional Vector Quantizationcited by 0
- 2026Adaptive Fused Prior Transfer for Controllable Generative Image Compression16 May 2026
- 2026Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learning11 Feb 2026
- 2026LooC: Effective Low-Dimensional Codebook for Compositional Vector Quantization1 Jan 2026
- 2025Texture Vector-Quantization and Reconstruction Aware Prediction for Generative Super-Resolution28 Sep 2025
- 2025mathttM3VIR: A Large-Scale Multi-Modality Multi-View Synthesized Benchmark Dataset for Image Restoration and Content Creation21 Sep 2025
- 2025TVC: Tokenized Video Compression with Ultra-Low Bit Rate22 Apr 2025
- 2025Learned Image Compression with Dictionary-based Entropy Model1 Apr 2025
- 2025Towards Improved Text-Aligned Codebook Learning: Multi-Hierarchical Codebook-Text Alignment with Long Text3 Mar 2025
- 2024GameIR: A Large-Scale Synthesized Ground-Truth Dataset for Image Restoration over Gaming Content29 Aug 2024
- 2024Restoring Images in Adverse Weather Conditions via Histogram Transformer14 Jul 2024
- 2024HybridFlow: Infusing Continuity into Masked Codebook for Extreme Low-Bitrate Image Compression20 Apr 2024
- 2024Monocular Identity-Conditioned Facial Reflectance Reconstruction30 Mar 2024
2 more
- 2024Restoring Images in Adverse Weather Conditions via Histogram Transformercited by 73
- 2025Learned Image Compression with Dictionary-based Entropy Modelcited by 30
- 2024HybridFlow: Infusing Continuity into Masked Codebook for Extreme Low-Bitrate Image Compressioncited by 14
- 2025TVC: Tokenized Video Compression with Ultra-Low Bit Ratecited by 7
- 2023Neural Image Compression Using Masked Sparse Visual Representationcited by 5
- 2024Monocular Identity-Conditioned Facial Reflectance Reconstructioncited by 5
- 2025Towards Improved Text-Aligned Codebook Learning: Multi-Hierarchical Codebook-Text Alignment with Long Textcited by 4
- 2025Texture Vector-Quantization and Reconstruction Aware Prediction for Generative Super-Resolutioncited by 3
- 2024GameIR: A Large-Scale Synthesized Ground-Truth Dataset for Image Restoration over Gaming Contentcited by 2
- 2025mathttM3VIR: A Large-Scale Multi-Modality Multi-View Synthesized Benchmark Dataset for Image Restoration and Content Creationcited by 0
- 2026LooC: Effective Low-Dimensional Codebook for Compositional Vector Quantizationcited by 0
- 2026Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learningcited by 0
12 most cited of 13 All 13 (JSON)
- 2026Adaptive Fused Prior Transfer for Controllable Generative Image Compression16 May 2026
- 2026Enhancing Underwater Images via Adaptive Semantic-aware Codebook Learning11 Feb 2026
- 2026LooC: Effective Low-Dimensional Codebook for Compositional Vector Quantization1 Jan 2026
- 2025Texture Vector-Quantization and Reconstruction Aware Prediction for Generative Super-Resolution28 Sep 2025
- 2025mathttM3VIR: A Large-Scale Multi-Modality Multi-View Synthesized Benchmark Dataset for Image Restoration and Content Creation21 Sep 2025
- 2025TVC: Tokenized Video Compression with Ultra-Low Bit Rate22 Apr 2025
- 2025Learned Image Compression with Dictionary-based Entropy Model1 Apr 2025
- 2025Towards Improved Text-Aligned Codebook Learning: Multi-Hierarchical Codebook-Text Alignment with Long Text3 Mar 2025
- 2024GameIR: A Large-Scale Synthesized Ground-Truth Dataset for Image Restoration over Gaming Content29 Aug 2024
- 2024Restoring Images in Adverse Weather Conditions via Histogram Transformer14 Jul 2024
- 2024HybridFlow: Infusing Continuity into Masked Codebook for Extreme Low-Bitrate Image Compression20 Apr 2024
- 2024Monocular Identity-Conditioned Facial Reflectance Reconstruction30 Mar 2024
12 newest of 13 All 13 (JSON)