An analytic theory of convolutional neural network inverse problems solvers
Minh Hai Nguyen, Quoc Bao, Edouard Pauwels, Pierre Weiss
Abstract
Supervised convolutional neural networks (CNNs) are widely used to solve imaging inverse problems, achieving state-of-the-art performance in numerous applications. However, despite their empirical success, these methods are poorly understood from a theoretical perspective and often treated as black boxes. To bridge this gap, we analyze trained neural networks through the lens of the Minimum Mean Square Error (MMSE) estimator, incorporating functional constraints that capture two fundamental inductive biases of CNNs: translation equivariance and locality via finite receptive fields. Under the empirical training distribution, we derive an analytic, interpretable, and tractable formula for this constrained variant, termed Local-Equivariant MMSE (LE-MMSE). Through extensive numerical experiments across various inverse problems (denoising, inpainting, deconvolution, accelerated MRI), datasets (FFHQ, CIFAR-10, FashionMNIST, FastMRI), and architectures (U-Net, ResNet, PatchMLP), we demonstrate that our theory matches the neural networks outputs (PSNR ≳ 25dB). Furthermore, we provide insights into the differences between physics-aware and physics-agnostic estimators, the impact of high-density regions in the training (patch) distribution, and the influence of other factors (dataset size, patch size, etc. ).
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.
Builds on10
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- A Group-Theoretic Framework for Data AugmentationShuxiao Chen, Edgar Dobriban, Jane H. LeeNeurIPS 2020 · 254 citations
- Emergent and Predictable Memorization in Large Language ModelsStella Biderman, USVSN Sai Prashanth, Lintang Sutawika, Hailey Schoelkopf et al.NeurIPS 2023 · 205 citations
- Equivariant Imaging: Learning Beyond the Range SpaceDongdong Chen, Julián Tachella, Mike E. DaviesICCV 2021 · 139 citations
- On the Closed-Form of Flow Matching: Generalization Does Not Arise from Target StochasticityQuentin Bertrand, Anne Gagneux, Mathurin Massias, Rémi EmonetNeurIPS 2025 · 49 citations
Related papers
- Stochastic Solutions for Linear Inverse Problems using the Prior Implicit in a DenoiserZahra Kadkhodaie, Eero P. SimoncelliNeurIPS 2021 · 202 citations
- Equivariant Plug-and-Play Image ReconstructionMatthieu Terris, Thomas Moreau, Nelly Pustelnik, Julián TachellaCVPR 2024 · 25 citations
- It Has Potential: Gradient-Driven Denoisers for Convergent Solutions to Inverse ProblemsRegev Cohen, Yochai Blau, Daniel Freedman, Ehud RivlinNeurIPS 2021 · 84 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
- Convex Regularization behind Neural ReconstructionArda Sahiner, Morteza Mardani, Batu Ozturkler, Mert Pilanci et al.ICLR 2021 · 25 citations
