Chasing Better Deep Image Priors between Over- and Under-parameterization

AI-generated keywords: Deep learning

AI-generated Key Points

  • Deep Neural Networks (DNNs) are recognized for serving as over-parameterized deep image priors (DIP) in deep learning.
  • Compact, under-parameterized image priors like deep decoders show impressive performance in image restoration tasks with a compromise in accuracy.
  • The concept of "lottery image prior" (LIP) is proposed to find intermediate parameterized image priors that balance performance, efficiency, and transferability.
  • LIP leverages sparsity within DNNs to identify sparse subnetworks within over-parameterized DIPs that match the original DNN's performance in various image inverse problems.
  • Identified LIP subnetworks outperform deep decoders while maintaining comparable model sizes and demonstrate high transferability across different images and restoration tasks.
  • LIP concept extends to compressive sensing image reconstruction using a pre-trained GAN generator as the prior, confirming its validity in diverse settings.
  • This research demonstrates the relevance of the lottery ticket hypothesis (LTH) in inverse problems or image priors, offering new possibilities for optimizing deep learning models for enhanced efficiency and performance in image restoration applications.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Qiming Wu, Xiaohan Chen, Yifan Jiang, Zhangyang Wang

Transactions on Machine Learning Research (2023), 2835-8856
Codes are available at https://github.com/VITA-Group/Chasing-Better-DIPs
License: CC BY 4.0

Abstract: Deep Neural Networks (DNNs) are well-known to act as over-parameterized deep image priors (DIP) that regularize various image inverse problems. Meanwhile, researchers also proposed extremely compact, under-parameterized image priors (e.g., deep decoder) that are strikingly competent for image restoration too, despite a loss of accuracy. These two extremes push us to think whether there exists a better solution in the middle: between over- and under-parameterized image priors, can one identify "intermediate" parameterized image priors that achieve better trade-offs between performance, efficiency, and even preserving strong transferability? Drawing inspirations from the lottery ticket hypothesis (LTH), we conjecture and study a novel "lottery image prior" (LIP) by exploiting DNN inherent sparsity, stated as: given an over-parameterized DNN-based image prior, it will contain a sparse subnetwork that can be trained in isolation, to match the original DNN's performance when being applied as a prior to various image inverse problems. Our results validate the superiority of LIPs: we can successfully locate the LIP subnetworks from over-parameterized DIPs at substantial sparsity ranges. Those LIP subnetworks significantly outperform deep decoders under comparably compact model sizes (by often fully preserving the effectiveness of their over-parameterized counterparts), and they also possess high transferability across different images as well as restoration task types. Besides, we also extend LIP to compressive sensing image reconstruction, where a pre-trained GAN generator is used as the prior (in contrast to untrained DIP or deep decoder), and confirm its validity in this setting too. To our best knowledge, this is the first time that LTH is demonstrated to be relevant in the context of inverse problems or image priors.

Submitted to arXiv on 31 Oct. 2024

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2410.24187v1

In the realm of deep learning, <kw> Deep Neural Networks (DNNs) </kw> have been widely recognized for their ability to serve as over-parameterized deep image priors (DIP), effectively regularizing various image inverse problems. On the other end of the spectrum, researchers have also introduced remarkably compact, under-parameterized image priors such as deep decoders, which demonstrate impressive performance in image restoration tasks despite a slight compromise in accuracy. This dichotomy has prompted a critical question: is there an optimal solution that lies somewhere between these two extremes? Can we identify "intermediate" parameterized image priors that strike a balance between performance, efficiency, and maintaining strong transferability? Drawing inspiration from the <kw> lottery ticket hypothesis (LTH) </kw>, a novel concept known as "lottery image prior" (LIP) is proposed and investigated. The LIP leverages the inherent sparsity within DNNs and posits that within an over-parameterized DNN-based image prior, there exists a sparse subnetwork that can be isolated and trained to match the performance of the original DNN when applied as a prior in various image inverse problems. The results of this study validate the effectiveness of LIPs, showcasing the successful identification of sparse subnetworks within over-parameterized DIPs across significant sparsity ranges. These identified LIP subnetworks exhibit superior performance compared to deep decoders while maintaining comparable model sizes. They not only preserve the efficacy of their over-parameterized counterparts but also demonstrate high transferability across different images and types of restoration tasks. Furthermore, the concept of LIP is extended to compressive sensing image reconstruction by utilizing a pre-trained GAN generator as the prior, further confirming its validity in diverse settings. This groundbreaking research marks the first demonstration of the relevance of LTH in the context of inverse problems or image priors. Published in Transactions on Machine Learning Research (08/2023), this study sheds light on new possibilities for optimizing deep learning models for enhanced efficiency and performance in image restoration applications.
Created on 29 Jul. 2026

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.