Processing math: 0%
Mixed Degradation Image Restoration via Deep Image Prior Empowered by Deep Denoising Engine | IEEE Conference Publication | IEEE Xplore

Mixed Degradation Image Restoration via Deep Image Prior Empowered by Deep Denoising Engine


Abstract:

Deep Image Prior (DIP) is a powerful unsupervised learning image restoration technique. However, DIP struggles when handling complex degradation scenarios involving mixed...Show More

Abstract:

Deep Image Prior (DIP) is a powerful unsupervised learning image restoration technique. However, DIP struggles when handling complex degradation scenarios involving mixed image artifacts. To address this limitation, we propose a novel technique to enhance DIP’s performance in handling mixed image degradation. Our method leverages additional deep denoiser, which is deployed as a denoising engine in the regularization by denoising (RED) framework. A new objective function is constructed by combining DIP with RED, and solved by the alternating direction method of multiplier (ADMM) algorithm. Our method explicitly learns a more comprehensive representation of the underlying image structure and being robust to different types of degradation. Experimental results demonstrate the effectiveness of our method, showing effective improvements in restoring images corrupted by mixed degradation on several image restoration tasks, such as image inpainting, super-resolution and deblurring.
Date of Conference: 30 June 2024 - 05 July 2024
Date Added to IEEE Xplore: 09 September 2024
ISBN Information:

ISSN Information:

Conference Location: Yokohama, Japan

Funding Agency:

References is not available for this document.

I. Introduction

This paper begins with the general image degraded model: \begin{equation*}Y = HX + n,\tag{1}\end{equation*} here, Y, X indicate the observed degraded and latent clean images, respectively. n is additive Gaussian noise. H is the degradation matrix relating to degradation system. Depending on the specific form of H, Eq. (1) can be expressed as different restoration tasks. For example, Eq. (1) becomes the image inpainting when H is a masking matrix; Eq. (1) represents the image super-resolution when H is a sub-sampling operator; Eq. (1) becomes the image deblurring when H is convolution operator, and image denoising when H is an identity matrix.

Select All
1.
D. Ulyanov, A. Vedaldi and V. Lempitsky, "Deep image prior", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 9446-9454, 2018.
2.
R. Heckel and P. Hand, "Deep decoder: Concise image representations from untrained non-convolutional networks", International Conference on Learning Representations, 2019.
3.
Y.-C. Chen, C. Gao, E. Robb and J.-B. Huang, "Nas-dip: Learning deep image prior with neural architecture search", European Conference on Computer Vision, 2020.
4.
G. Mataev, P. Milanfar and M. Elad, "Deepred: Deep image prior powered by red", Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops, Oct 2019.
5.
N. Shabtay, E. Schwartz and R. Giryes, "PIP: Positional-encoding Image Prior", Nov. 2022.
6.
Z. Shi, M. Pascal and S. Subhransu, "On measuring and controlling the spectral bias of the deep image prior", International Journal of Computer Vision, vol. 130, no. 4, pp. 885-908, Apr. 2022.
7.
T. Li, H. Wang, Z. Zhuang and J. Sun, "Deep random projector: Accelerated deep image prior", Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 18 176-18 185, June 2023.
8.
A. Buades, B. Coll and J.-M. Morel, "A non-local algorithm for image denoising", IEEE Computer Society Conference on Computer Vision and Pattern Recognition, vol. 2, pp. 60-65, 2005.
9.
K. Dabov, A. Foi, V. Katkovnik and K. Egiazarian, "Image denoising by sparse 3-d transform-domain collaborative filtering", IEEE Transactions on Image Processing, vol. 16, no. 8, pp. 2080-2095, 2007.
10.
D. Liu, J. Wang, Q. Shan, D. Smyl, J. Deng and J. Du, "Deepeit: Deep image prior enabled electrical impedance tomography", IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 45, no. 8, pp. 9627-9638, 2023.
11.
D. Ulyanov, A. Vedaldi and V. S. Lempitsky, "Deep image prior", International Journal of Computer Vision, vol. 128, pp. 1867-1888, 2020.
12.
K. Gong, C. Catana, J. Qi and Q. Li, "Pet image reconstruction using deep image prior", IEEE Transactions on Medical Imaging, vol. 38, no. 7, pp. 1655-1665, 2019.
13.
J. Yoo, K. H. Jin, H. Gupta, J. Yerly, M. Stuber and M. Unser, "Time-dependent deep image prior for dynamic mri", IEEE Transactions on Medical Imaging, vol. 40, no. 12, pp. 3337-3348, 2021.
14.
D. O. Baguer, J. Leuschner and M. Schmidt, "Computed tomography reconstruction using deep image prior and learned reconstruction methods", Inverse Problems, vol. 36, no. 9, pp. 094004, 2020.
15.
Y. Liu, J.-P. Li, Y. Pang, D. Nie and P.-T. Yap, "The devil is in the upsampling: Architectural decisions made simpler for denoising with deep image prior", ArXiv, vol. abs/2304.11409, 2023.
16.
M. E. Arican, O. Kara, G. Bredell and E. Konukoglu, "Isnas-dip: Image-specific neural architecture search for deep image prior", 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1950-1958, 2021.
17.
Y. Liu, J. Li, Y. Pang, D. Nie and P.-T. Yap, "The devil is in the upsampling: Architectural decisions made simpler for denoising with deep image prior", Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), pp. 12 408-12 417, October 2023.
18.
Z. Sun, F. Latorre, T. Sanchez and V. Cevher, "A plug-and-play deep image prior", 2021 IEEE International Conference on Acoustics Speech and Signal Processing, pp. 8103-8107, 2021.
19.
Y. Romano, M. Elad and P. Milanfar, "The little engine that could: Regularization by denoising (red)", SIAM Journal on Imaging Sciences, vol. 10, no. 4, pp. 1804-1844, 2017.
20.
K. Zhang, W. Zuo and L. Zhang, "Ffdnet: Toward a fast and flexible solution for cnn-based image denoising", IEEE Transactions on Image Processing, vol. 27, no. 9, pp. 4608-4622, 2018.
21.
J. Liu, Y. Sun, X. Xu and U. S. Kamilov, "Image restoration using total variation regularized deep image prior", 2019 IEEE International Conference on Acoustics Speech and Signal Processing, pp. 7715-7719, 2019.
22.
A. Beck and M. Teboulle, "Fast gradient-based algorithms for constrained total variation image denoising and deblurring problems", IEEE Transactions on Image Processing, vol. 18, no. 11, pp. 2419-2434, 2009.
23.
R. Fermanian, M. Le Pendu and C. Guillemot, "Regularizing the deep image prior with a learned denoiser for linear inverse problems", 2021 IEEE 23rd International Workshop on Multimedia Signal Processing (MMSP)., pp. 1-6, 2021.
24.
S. V. Venkatakrishnan, C. A. Bouman and B. Wohlberg, "Plug-and-play priors for model based reconstruction", 2013 IEEE global conference on signal and information processing., pp. 945-948, 2013.
25.
S. Boyd, N. Parikh, E. Chu, B. Peleato and J. Eckstein, "Distributed optimization and statistical learning via the alternating direction method of multipliers", Found. Trends Mach. Learn., vol. 3, no. 1, pp. 1-122, 2011.
26.
O. Ronneberger, P. Fischer and T. Brox, "U-net: Convolutional networks for biomedical image segmentation", Medical Image Computing and Computer-Assisted Intervention, 2015.
27.
A. Krizhevsky, I. Sutskever and G. E. Hinton, "Imagenet classification with deep convolutional neural networks", Commun. ACM, vol. 60, no. 6, pp. 84-90, may 2017.
28.
S. Ioffe and C. Szegedy, "Batch normalization: Accelerating deep network training by reducing internal covariate shift", Proceedings of the 32nd International Conference on International Conference on Machine Learning - Volume 37, pp. 448-456, 2015.
29.
Z. Wang, A. Bovik, H. Sheikh and E. Simoncelli, "Image quality assessment: from error visibility to structural similarity", IEEE Transactions on Image Processing, vol. 13, no. 4, pp. 600-612, 2004.
30.
E. T. Reehorst and P. Schniter, "Regularization by denoising: Clarifications and new interpretations", IEEE Transactions on Computational Imaging, vol. 5, pp. 52-67, 2018.
Contact IEEE to Subscribe

References

References is not available for this document.