Efficient gradient-based methods for bilevel learning via recycling Krylov subspaces

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ehrhardt, Matthias J., Gazzola, Silvia, Scott, Sebastian J.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915539288326144
author Ehrhardt, Matthias J.
Gazzola, Silvia
Scott, Sebastian J.
author_facet Ehrhardt, Matthias J.
Gazzola, Silvia
Scott, Sebastian J.
contents Many optimization problems require hyperparameters, i.e., parameters that must be pre-specified in advance, such as regularization parameters and parametric regularizers in variational regularization methods for inverse problems, and dictionaries in compressed sensing. A data-driven approach to determine appropriate hyperparameter values is via a nested optimization framework known as bilevel learning. Even when it is possible to employ a gradient-based solver to the bilevel optimization problem, construction of the gradients, known as hypergradients, is computationally challenging, each one requiring both a solution of a minimization problem and a linear system solve. These systems do not change much during the iterations, which motivates us to apply recycling Krylov subspace methods, wherein information from one linear system solve is re-used to solve the next linear system. Existing recycling strategies often employ eigenvector approximations called Ritz vectors. In this work we propose a novel recycling strategy based on a new concept, Ritz generalized singular vectors, which acknowledge the bilevel setting. Additionally, while existing iterative methods primarily terminate according to the residual norm, this new concept allows us to define a new stopping criterion that directly approximates the error of the associated hypergradient. The proposed approach is validated through extensive numerical testing in the context of inverse problems in imaging.
format Preprint
id arxiv_https___arxiv_org_abs_2412_08264
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient gradient-based methods for bilevel learning via recycling Krylov subspaces
Ehrhardt, Matthias J.
Gazzola, Silvia
Scott, Sebastian J.
Optimization and Control
Numerical Analysis
65F10
Many optimization problems require hyperparameters, i.e., parameters that must be pre-specified in advance, such as regularization parameters and parametric regularizers in variational regularization methods for inverse problems, and dictionaries in compressed sensing. A data-driven approach to determine appropriate hyperparameter values is via a nested optimization framework known as bilevel learning. Even when it is possible to employ a gradient-based solver to the bilevel optimization problem, construction of the gradients, known as hypergradients, is computationally challenging, each one requiring both a solution of a minimization problem and a linear system solve. These systems do not change much during the iterations, which motivates us to apply recycling Krylov subspace methods, wherein information from one linear system solve is re-used to solve the next linear system. Existing recycling strategies often employ eigenvector approximations called Ritz vectors. In this work we propose a novel recycling strategy based on a new concept, Ritz generalized singular vectors, which acknowledge the bilevel setting. Additionally, while existing iterative methods primarily terminate according to the residual norm, this new concept allows us to define a new stopping criterion that directly approximates the error of the associated hypergradient. The proposed approach is validated through extensive numerical testing in the context of inverse problems in imaging.
title Efficient gradient-based methods for bilevel learning via recycling Krylov subspaces
topic Optimization and Control
Numerical Analysis
65F10
url https://arxiv.org/abs/2412.08264