NAS-DIP: Learning Deep Image Prior with Neural Architecture Search
Recent work has shown that the structure of deep convolutional neural networks can be used as a structured image prior for solving various inverse image restoration tasks. Instead of using hand-designed architectures, we propose to search for neural architectures that capture stronger image priors. Building upon a generic U-Net architecture, our core contribution lies in designing new search spaces for (1) an upsampling cell and (2) a pattern of cross-scale residual connections. We search for an improved network by leveraging an existing neural architecture search algorithm (using reinforcement learning with a recurrent neural network controller). We validate the effectiveness of our method via a wide variety of applications, including image restoration, dehazing, image-to-image translation, and matrix factorization. Extensive experimental results show that our algorithm performs favorably against state-of-the-art learning-free approaches and reaches competitive performance with existing learning-based methods in some cases.
Code (1)
Tasks
Image RestorationImage-to-Image TranslationNeural Architecture SearchTranslationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Neural Architecture Search for Deep Image Prior
We present a neural architecture search (NAS) technique to enhance the performance of unsupervised image de-noising, in-painting and super-resolution under the recently proposed Deep Image Prior (DIP). We show that evolu…
DecoderImage RestorationNeural Architecture SearchSuper-ResolutionMulti-Prior Learning via Neural Architecture Search for Blind Face Restoration
Blind Face Restoration (BFR) aims to recover high-quality face images from low-quality ones and usually resorts to facial priors for improving restoration performance. However, current methods still suffer from two major…
Blind Face RestorationNeural Architecture SearchISNAS-DIP: Image-Specific Neural Architecture Search for Deep Image Prior
Recent works show that convolutional neural network (CNN) architectures have a spectral bias towards lower frequencies, which has been leveraged for various image restoration tasks in the Deep Image Prior (DIP) framework…
DenoisingImage DenoisingImage RestorationInductive Bias+2Retinex-inspired Unrolling with Cooperative Prior Architecture Search for Low-light Image Enhancement
Low-light image enhancement plays very important roles in low-level vision field. Recent works have built a large variety of deep learning models to address this task. However, these approaches mostly rely on significant…
Image EnhancementLow-Light Image EnhancementRolling Shutter CorrectionGI-NAS: Boosting Gradient Inversion Attacks through Adaptive Neural Architecture Search
Gradient Inversion Attacks invert the transmitted gradients in Federated Learning (FL) systems to reconstruct the sensitive data of local clients and have raised considerable privacy concerns. A majority of gradient inve…
Federated LearningNeural Architecture Search