Chasing Better Deep Image Priors between Over- and Under-parameterization
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.
Authors: Qiming Wu, Xiaohan Chen, Yifan Jiang, Zhangyang Wang
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.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
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 representationLook 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.