Bilevel learning of regularization models and their discretization for image deblurring and super-resolution

Bilevel learning is a powerful optimization technique that has extensively been employed in recent years to bridge the world of model-driven variational approaches with data-driven methods. Upon suitable parametrization of the desired quantities of interest (e.g., regularization terms or discretization filters), such approach computes optimal parameter values by solving a nested optimization problem where the variational model acts as a constraint. In this work, we consider two different use cases of bilevel learning for the problem of image restoration. First, we focus on learning scalar weights and convolutional filters defining a Field of Experts regularizer to restore natural images degraded by blur and noise. For improving the practical performance, the lower-level problem is solved by means of a gradient descent scheme combined with a line-search strategy based on the Barzilai-Borwein rule. As a second application, the bilevel setup is employed for learning a discretization of the popular total variation regularizer for solving image restoration problems (in particular, deblurring and super-resolution). Numerical results show the effectiveness of the approach and their generalization to multiple tasks.

[1]  S. Bonettini,et al.  Explainable bilevel optimization: an application to the Helsinki deblur challenge , 2022, Inverse Problems and Imaging.

[2]  D. Lorenz,et al.  Learning Variational Models with Unrolling and Bilevel Optimization , 2022, ArXiv.

[3]  A. Chambolle,et al.  Convergence of a Piggyback-Style Method for the Differentiation of Solutions of Standard Saddle-Point Problems , 2022, SIAM J. Math. Data Sci..

[4]  F. Sgallari,et al.  ADMM-Based Residual Whiteness Principle for Automatic Parameter Selection in Single Image Super-Resolution Problems , 2022, Journal of Mathematical Imaging and Vision.

[5]  Matthias Joachim Ehrhardt,et al.  Regularization of inverse problems: deep equilibrium models versus bilevel learning , 2022, Numerical Algebra, Control and Optimization.

[6]  J. Bolte,et al.  Automatic differentiation of nonsmooth iterative algorithms , 2022, NeurIPS.

[7]  J. C. De Los Reyes,et al.  Parameter space study of optimal scale-dependent weights in TV image denoising , 2022, Applicable Analysis.

[8]  J. Fessler,et al.  Bilevel Methods for Image Reconstruction , 2021, Found. Trends Signal Process..

[9]  L. Calatroni,et al.  Constrained and Unconstrained Inverse Potts Modelling for Joint Image Super-Resolution and Segmentation , 2022, Image Process. Line.

[10]  Julianne Chung,et al.  Learning regularization parameters of inverse problems via deep neural networks , 2021, Inverse Problems.

[11]  Antonin Chambolle,et al.  Learning Consistent Discretizations of the Total Variation , 2020, SIAM J. Imaging Sci..

[12]  Kaiyi Ji,et al.  Bilevel Optimization: Convergence Analysis and Enhanced Design , 2020, ICML.

[13]  Matthias J. Ehrhardt,et al.  Inexact Derivative-Free Optimization for Bilevel Learning , 2020, Journal of Mathematical Imaging and Vision.

[14]  Marta D'Elia,et al.  Bilevel Parameter Learning for Nonlocal Image Denoising Models , 2019, Journal of Mathematical Imaging and Vision.

[15]  Valentin De Bortoli,et al.  Maximum Likelihood Estimation of Regularization Parameters in High-Dimensional Inverse Problems: An Empirical Bayesian Approach Part I: Methodology and Experiments , 2019, SIAM J. Imaging Sci..

[16]  L. Calatroni,et al.  Analysis and automatic parameter selection of a variational model for mixed Gaussian and salt-and-pepper noise removal , 2018, Inverse Problems.

[17]  Paolo Frasconi,et al.  Bilevel Programming for Hyperparameter Optimization and Meta-Learning , 2018, ICML.

[18]  Karl Kunisch,et al.  A bilevel approach for parameter learning in inverse problems , 2018, Inverse Problems.

[19]  Valeria Ruggiero,et al.  On the steplength selection in gradient methods for unconstrained optimization , 2018, Appl. Math. Comput..

[20]  Martin Burger,et al.  Modern regularization methods for inverse problems , 2018, Acta Numerica.

[21]  Laurent Condat,et al.  Discrete Total Variation: New Definition and Minimization , 2017, SIAM J. Imaging Sci..

[22]  Carola-Bibiane Schönlieb,et al.  Bilevel Parameter Learning for Higher-Order Total Variation Regularisation Models , 2015, Journal of Mathematical Imaging and Vision.

[23]  Lea Fleischer,et al.  Regularization of Inverse Problems , 1996 .

[24]  A. Basarab,et al.  Fast Single Image Super-resolution using a New Analytical Solution for l2-l2 Problems. , 2016, IEEE transactions on image processing : a publication of the IEEE Signal Processing Society.

[25]  Thomas Brox,et al.  Bilevel Optimization with Nonsmooth Lower Level Problems , 2015, SSVM.

[26]  Carola-Bibiane Schönlieb,et al.  Bilevel approaches for learning of variational imaging models , 2015, ArXiv.

[27]  M. Hintermüller,et al.  Functional-analytic and numerical issues in splitting methods for total variation-based image reconstruction , 2014 .

[28]  Yunjin Chen,et al.  Insights Into Analysis Operator Learning: From Patch-Based Sparse Models to Higher Order MRFs , 2014, IEEE Transactions on Image Processing.

[29]  C. Schönlieb,et al.  Image denoising: Learning the noise model via nonsmooth PDE-constrained optimization , 2013 .

[30]  Karl Kunisch,et al.  A Bilevel Optimization Approach for Parameter Learning in Variational Models , 2013, SIAM J. Imaging Sci..

[31]  Antonin Chambolle,et al.  Diagonal preconditioning for first order primal-dual algorithms in convex optimization , 2011, 2011 International Conference on Computer Vision.

[32]  Antonin Chambolle,et al.  A First-Order Primal-Dual Algorithm for Convex Problems with Applications to Imaging , 2011, Journal of Mathematical Imaging and Vision.

[33]  Daniela Calvetti,et al.  Introduction to Bayesian Scientific Computing: Ten Lectures on Subjective Computing , 2007 .

[34]  Jianhong Shen,et al.  Deblurring images: Matrices, spectra, and filtering , 2007, Math. Comput..

[35]  Michael J. Black,et al.  Fields of Experts: a framework for learning image priors , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[36]  A. Griewank,et al.  Piggyback Differentiation and Optimization , 2003 .

[37]  Jitendra Malik,et al.  A database of human segmented natural images and its application to evaluating segmentation algorithms and measuring ecological statistics , 2001, Proceedings Eighth IEEE International Conference on Computer Vision. ICCV 2001.

[38]  Marcos Raydan,et al.  The Barzilai and Borwein Gradient Method for the Large Scale Unconstrained Minimization Problem , 1997, SIAM J. Optim..

[39]  L. Rudin,et al.  Nonlinear total variation based noise removal algorithms , 1992 .

[40]  J. Borwein,et al.  Two-Point Step Size Gradient Methods , 1988 .