Intermediate Layer Optimization for Inverse Problems using Deep Generative Models
Giannis Daras, Joseph Dean, Ajil Jalal, Alex Dimakis
Abstract
We propose Intermediate Layer Optimization (ILO), a novel optimization algorithm for solving inverse problems with deep generative models. Instead of optimizing only over the initial latent code, we progressively change the input layer obtaining successively more expressive generators. To explore the higher dimensional spaces, our method searches for latent codes that lie within a small ball around the manifold induced by the previous layer. Our theoretical analysis shows that by keeping the radius of the ball relatively small, we can improve the established error bound for compressed sensing with deep generative models. We empirically show that our approach outperforms state-of-the-art methods introduced in StyleGAN-2 and PULSE for a wide range of inverse problems including inpainting, denoising, super-resolution and compressed sensing.
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 fd1a0b82-8b48-46fb-b9f2-2467db91584fCited by top-tier papers33
- Denoising Diffusion Restoration ModelsBahjat Kawar, Michael Elad, Stefano Ermon, Jiaming SongNeurIPS 2022 · 1,439 citations
- Improving Diffusion Models for Inverse Problems using Manifold ConstraintsHyungjin Chung, Byeongsu Sim, Dohoon Ryu, Jong Chul YeNeurIPS 2022 · 738 citations
- Robust Compressed Sensing MRI with Deep Generative PriorsAjil Jalal, Marius Arvinte, Giannis Daras, Eric Price et al.NeurIPS 2021 · 483 citations
- D2C: Diffusion-Decoding Models for Few-Shot Conditional GenerationAbhishek Sinha, Jiaming Song, Chenlin Meng, Stefano ErmonNeurIPS 2021 · 149 citations
- Measuring Robustness in Deep Learning Based Compressive SensingMohammad Zalbagi Darestani, Akshay S. Chaudhari, Reinhard HeckelICML 2021 · 94 citations
Builds on8
- Free-Form Image Inpainting With Gated ConvolutionJiahui Yu, Zhe Lin, Jimei Yang, Xiaohui Shen et al.ICCV 2019 · 1,990 citations
- Image2StyleGAN: How to Embed Images Into the StyleGAN Latent Space?Rameen Abdal, Yipeng Qin, Peter WonkaICCV 2019 · 1,195 citations
- Coherent Semantic Attention for Image InpaintingHongyu Liu, Bin Jiang, Yi Xiao, Chao YangICCV 2019 · 395 citations
- GAN-Based Projector for Faster Recovery With Convergence Guarantees in Linear Inverse ProblemsAnkit Raj, Yuqi Li, Yoram BreslerICCV 2019 · 61 citations
- Your Local GAN: Designing Two Dimensional Local Attention Mechanisms for Generative ModelsGiannis Daras, Augustus Odena, Han Zhang, Alexandros G. DimakisCVPR 2020
Related papers
- Score-Guided Intermediate Level Optimization: Fast Langevin Mixing for Inverse ProblemsGiannis Daras, Yuval Dagan, Alex Dimakis, Constantinos DaskalakisICML 2022 · 15 citations
- Differentiable Gaussianization Layers for Inverse Problems Regularized by Deep Generative ModelsDongzhuo LiICLR 2023 · 1 citation
- Optimization for Amortized Inverse ProblemsTianci Liu, Tong Yang, Quan Zhang, Qi LeiICML 2023 · 7 citations
- Integrating Intermediate Layer Optimization and Projected Gradient Descent for Solving Inverse Problems with Diffusion ModelsYang Zheng, Wen Li, Zhaoqiang LiuICML 2025
- Image-Adaptive GAN Based ReconstructionShady Abu Hussein, Tom Tirer, Raja GiryesAAAI 2020 · 104 citations
