Revisiting the Generalization Problem of Low-level Vision Models Through the Lens of Image Deraining
Not on the map. Why?
Abstract (arXiv)
Generalization to unseen degradations remains a fundamental challenge for low-level vision models. This paper aims to investigate the underlying mechanism of this failure, using image deraining as a primary case study due to its well-defined and decoupled structure. Through systematic experiments, we reveal that generalization issues are not primarily caused by limited network capacity, but rather by a ``shortcut learning'' phenomenon driven by the relative complexity between image content and degradation patterns. We find that when background content is excessively complex, networks preferentially overfit the simpler degradation characteristics to minimize training loss, thereby failing to learn the underlying image distribution. To address this, we propose two principled strategies: (1) balancing the complexity of training data (backgrounds vs. degradations) to redirect the network's focus toward content reconstruction, and (2) leveraging strong content priors from pre-trained generative models to physically constrain the network onto a high-quality image manifold. Extensive experiments on image deraining, denoising, and deblurring validate our theoretical insights. Our work provides an interpretability-driven perspective and a principled methodology for improving the robustness and generalization of low-level vision models.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 70 of its 102 references that are in the map. First posted on arXiv 18 Feb 2025.
Builds on
70 of 102 references.
References · 70
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,348
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2016Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Networkcited by 1,898
- 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
- 2017Learning Important Features Through Propagating Activation Differencescited by 1,454
38 more
- 2021MUSIQ: Multi-scale Image Quality Transformercited by 1,058
- 2014Striving for Simplicity: The All Convolutional Netcited by 983
- 2020Pre-Trained Image Processing Transformercited by 891
- 2021Multi-Stage Progressive Image Restorationcited by 813
- 2022MANIQA: Multi-dimension Attention Network for No-Reference Image Quality Assessmentcited by 514
- 2023Exploiting Diffusion Prior for Real-World Image Super-Resolutioncited by 486
- 2016Deep Joint Rain Detection and Removal from a Single Imagecited by 351
- 2015Sketch-based Manga Retrieval using Manga109 Datasetcited by 326
- 2024Scaling Up to Excellence: Practicing Model Scaling for Photo-Realistic Image Restoration In the Wildcited by 281
- 2018Toward Convolutional Blind Denoising of Real Photographscited by 274
- 2019Progressive Image Deraining Networks: A Better and Simpler Baselinecited by 271
- 2020Multi-Scale Progressive Fusion Network for Single Image Derainingcited by 259
- 2023DiffBIR: Towards Blind Image Restoration with Generative Diffusion Priorcited by 255
- 2023Pixel-Aware Stable Diffusion for Realistic Image Super-resolution and Personalized Stylizationcited by 251
- 2018Density-aware Single Image De-raining using a Multi-stream Dense Networkcited by 248
- 2022Towards Robust Blind Face Restoration with Codebook Lookup Transformercited by 230
- 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 204
- 2019Blind Super-Resolution With Iterative Kernel Correctioncited by 203
- 2017Image De-raining Using a Conditional Generative Adversarial Networkcited by 197
- 2019Spatial Attentive Single-Image Deraining with a High Quality Real Rain Datasetcited by 192
- 2020PIPAL: a Large-Scale Image Quality Assessment Dataset for Perceptual Image Restorationcited by 177
- 2019Heavy Rain Image Restoration: Integrating Physics Model and Conditional Adversarial Learningcited by 157
- 2016Clearing the Skies: A deep network architecture for single-image rain removalcited by 136
- 2020Interpreting Super-Resolution Networks with Local Attribution Mapscited by 136
- 2020A Model-driven Deep Neural Network for Single Image Rain Removalcited by 125
- 2019Uncertainty Guided Multi-Scale Residual Learning-using a Cycle Spinning CNN for Single Image De-Rainingcited by 117
- 2023Depicting Beyond Scores: Advancing Image Quality Assessment through Multi-modal Language Modelscited by 104
- 2017Interpreting Deep Visual Representations via Network Dissectioncited by 103
- 2021Blind Image Super-Resolution: A Survey and Beyondcited by 93
- 2022Cross Aggregation Transformer for Image Restorationcited by 92
- 2022Accurate Image Restoration with Attention Retractable Transformercited by 85
- 2019Dual Residual Networks Leveraging the Potential of Paired Operations for Image Restorationcited by 73
- 2024Diffusion Models in Low-Level Vision: A Surveycited by 73
- 2022Rethinking Alignment in Video Super-Resolution Transformerscited by 71
- 2018Lightweight Pyramid Networks for Image Derainingcited by 64
- 2021Structure-Preserving Deraining with Residue Channel Prior Guidancecited by 63
- 2020Syn2Real Transfer Learning for Image Deraining using Gaussian Processescited by 62
- 2021Unpaired Deep Image Deraining Using Dual Contrastive Learningcited by 60
50 most cited of 70 All 70 (JSON)
Built on by
Citing papers per year: 2025: 2; 2026: 2.
4 papers.
- 2025Exploring Scalable Unified Modeling for General Low-Level Visioncited by 4
- 2026How far have we gone in Generative Image Restoration? A study on its capability, limitations and evaluation practicescited by 2
- 2025RAM++: Robust Representation Learning via Adaptive Mask for All-in-One Image Restorationcited by 1
- 2026Hidden-Shot: Towards One-Shot Task Generalization for Low-Level Vision Generalist Modelscited by 0
- 2026Hidden-Shot: Towards One-Shot Task Generalization for Low-Level Vision Generalist Models1 Jul 2026
- 2026How far have we gone in Generative Image Restoration? A study on its capability, limitations and evaluation practices5 Mar 2026
- 2025RAM++: Robust Representation Learning via Adaptive Mask for All-in-One Image Restoration15 Sep 2025
- 2025Exploring Scalable Unified Modeling for General Low-Level Vision20 Jul 2025