TY - JOUR
T1 - Deep image prior and weighted anisotropic-isotropic total variation regularization for solving linear inverse problems
AU - Xie, Yujia
AU - Chen, Wengu
AU - Ge, Huanmin
AU - Ng, Michael K.
N1 - This work was supported by Beijing Natural Science Foundation (No. 1232020), the NSF of China (Nos. 12371094, 12271050), Foundation of National Key Laboratory of Computational Physics (Grant No. 6142A05230503) and HKRGC GRF 12300218, 12300519, 17201020 and 17300021.
Publisher Copyright:
© 2024 Elsevier Inc.
PY - 2024/7/23
Y1 - 2024/7/23
N2 - Deep learning, particularly unsupervised techniques, has been widely used to solve linear inverse problems due to its flexibility. A notable unsupervised approach is the deep image prior (DIP), which employs a predetermined deep neural network to regularize inverse problems by imposing constraints on the generated image. This article introduces an optimization technique (DIP-AITV) by combining the DIP with the weighted anisotropic-isotropic total variation (AITV) regularization. Furthermore, we utilize the alternating direction method of multipliers (ADMM), a highly flexible optimization technique, to solve the DIP-AITV minimization problem effectively. To demonstrate the benefits of the proposed DIP-AITV method over the state-of-the-art DIP, DIP-TV, DIP-WTV and CS-DIP, we solve two linear inverse problems, i.e., image denoising and compressed sensing. Computation examples on the MSE and PSNR values show that our method outperforms the existing DIP-based methods in both synthetic and real grayscale and color images.
AB - Deep learning, particularly unsupervised techniques, has been widely used to solve linear inverse problems due to its flexibility. A notable unsupervised approach is the deep image prior (DIP), which employs a predetermined deep neural network to regularize inverse problems by imposing constraints on the generated image. This article introduces an optimization technique (DIP-AITV) by combining the DIP with the weighted anisotropic-isotropic total variation (AITV) regularization. Furthermore, we utilize the alternating direction method of multipliers (ADMM), a highly flexible optimization technique, to solve the DIP-AITV minimization problem effectively. To demonstrate the benefits of the proposed DIP-AITV method over the state-of-the-art DIP, DIP-TV, DIP-WTV and CS-DIP, we solve two linear inverse problems, i.e., image denoising and compressed sensing. Computation examples on the MSE and PSNR values show that our method outperforms the existing DIP-based methods in both synthetic and real grayscale and color images.
KW - Compressed sensing
KW - Deep image prior
KW - Image denoising
KW - The anisotropic-isotropic total variation
UR - http://www.scopus.com/inward/record.url?scp=85199214555&partnerID=8YFLogxK
U2 - 10.1016/j.amc.2024.128952
DO - 10.1016/j.amc.2024.128952
M3 - Journal article
AN - SCOPUS:85199214555
SN - 0096-3003
VL - 482
JO - Applied Mathematics and Computation
JF - Applied Mathematics and Computation
M1 - 128952
ER -