Interpretable model learning in variational imaging: a bilevel optimization approach

De los Reyes, Juan Carlos, Villacís, David

IMA Journal of Applied Mathematics, volume 89, number 1, pages 85-122, September 2023, doi: 10.1093/imamat/hxad024

Abstract

In this paper, we investigate the use of bilevel optimization for model learning in variational imaging problems. Bilevel learning is an alternative approach to deep learning methods, which leads to fully interpretable models. However, it requires a detailed analytical insight into the underlying mathematical model. We focus on the bilevel learning problem for total variation models with spatially- and patch-dependent parameters. Our study encompasses the directional differentiability of the solution mapping, the derivation of optimality conditions, and the characterization of the Bouligand subdifferential of the solution operator. We also propose a two-phase trust-region algorithm for solving the problem and present numerical tests using the CelebA dataset.

Bibtex

@article{10.1093/imamat/hxad024,
  author   = {De los Reyes, Juan Carlos and Villacís, David},
  title    = {{Interpretable model learning in variational imaging: a bilevel optimization approach}},
  journal  = {IMA Journal of Applied Mathematics},
  volume   = {89},
  number   = {1},
  pages    = {85-122},
  year     = {2023},
  month    = {09},
  abstract = {In this paper, we investigate the use of bilevel optimization for model learning in variational imaging problems. Bilevel learning is an alternative approach to deep learning methods, which leads to fully interpretable models. However, it requires a detailed analytical insight into the underlying mathematical model. We focus on the bilevel learning problem for total variation models with spatially- and patch-dependent parameters. Our study encompasses the directional differentiability of the solution mapping, the derivation of optimality conditions, and the characterization of the Bouligand subdifferential of the solution operator. We also propose a two-phase trust-region algorithm for solving the problem and present numerical tests using the CelebA dataset.},
  issn     = {0272-4960},
  doi      = {10.1093/imamat/hxad024},
  url      = {https://doi.org/10.1093/imamat/hxad024},
  eprint   = {https://academic.oup.com/imamat/article-pdf/89/1/85/58325932/hxad024.pdf}
}