Denoising and Regularization via Exploiting the Structural Bias of Convolutional Generators
Reinhard Heckel, Mahdi Soltanolkotabi
Abstract
Convolutional Neural Networks (CNNs) have emerged as highly successful tools for image generation, recovery, and restoration. This success is often attributed to large amounts of training data. However, recent experimental findings challenge this view and instead suggest that a major contributing factor to this success is that convolutional networks impose strong prior assumptions about natural images. A surprising experiment that highlights this architectural bias towards natural images is that one can remove noise and corruptions from a natural image without using any training data, by simply fitting (via gradient descent) a randomly initialized, over-parameterized convolutional generator to the single corrupted image. While this over-parameterized network can fit the corrupted image perfectly, surprisingly after a few iterations of gradient descent one obtains the uncorrupted image. This intriguing phenomenon enables state-of-the-art CNN-based denoising and regularization of linear inverse problems such as compressive sensing. In this paper, we take a step towards demystifying this experimental phenomenon by attributing this effect to particular architectural choices of convolutional networks, namely convolutions with fixed interpolating filters. We then formally characterize the dynamics of fitting a two-layer convolutional generator to a noisy signal and prove that early-stopped gradient descent denoises/regularizes. This result relies on showing that convolutional generators fit the structured part of an image significantly faster than the corrupted portion.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b388f3ef-aafb-4df0-b863-772d242cecb9Cited by top-tier papers19
- Measuring Robustness in Deep Learning Based Compressive SensingMohammad Zalbagi Darestani, Akshay S. Chaudhari, Reinhard HeckelICML 2021 · 94 citations
- Compressive sensing with un-trained neural networks: Gradient descent finds a smooth approximationReinhard Heckel, Mahdi SoltanolkotabiICML 2020 · 91 citations
- Early Stopping in Deep Networks: Double Descent and How to Eliminate itReinhard Heckel, Fatih Furkan YilmazICLR 2021 · 55 citations
- Robust Recovery via Implicit Bias of Discrepant Learning Rates for Double Over-parameterizationChong You, Zhihui Zhu, Qing Qu, Yi MaNeurIPS 2020 · 47 citations
- Approximation and Learning with Deep Convolutional Models: a Kernel PerspectiveAlberto BiettiICLR 2022 · 33 citations
Related papers
- Stochastic Solutions for Linear Inverse Problems using the Prior Implicit in a DenoiserZahra Kadkhodaie, Eero P. SimoncelliNeurIPS 2021 · 202 citations
- Invertible generative models for inverse problems: mitigating representation error and dataset biasMuhammad Asim, Max Daniels, Oscar Leong, Ali Ahmed et al.ICML 2020 · 172 citations
- The Devil is in the Upsampling: Architectural Decisions Made Simpler for Denoising with Deep Image PriorYilin Liu, Jiang Li, Yunkui Pang, Dong Nie et al.ICCV 2023 · 20 citations
- Recovery Analysis for Plug-and-Play Priors using the Restricted Eigenvalue ConditionJiaming Liu, M. Salman Asif, Brendt Wohlberg, Ulugbek KamilovNeurIPS 2021 · 55 citations
- The Surprising Effectiveness of Noise Pretraining for Implicit Neural RepresentationsKushal Vyas, Alper Kayabasi, Daniel Kim, Vishwanath Saragadam et al.CVPR 2026 · 1 citation
