Analyzing inexact hypergradients for bilevel learning

• Published in: IMA journal of applied mathematics Vol. 89; no. 1; pp. 254 - 278
• Main Authors: Ehrhardt, Matthias J, Roberts, Lindon
• Format: Journal Article
• Language: English
• Published: Oxford University Press 21.06.2024
• Subjects: automatic differentiation; hyperparameter optimization; bilevel optimization
• ISSN: 0272-4960; 1464-3634
• DOI: 10.1093/imamat/hxad035

Estimating hyperparameters has been a long-standing problem in machine learning. We consider the case where the task at hand is modeled as the solution to an optimization problem. Here the exact gradient with respect to the hyperparameters cannot be feasibly computed and approximate strategies are required. We introduce a unified framework for computing hypergradients that generalizes existing methods based on the implicit function theorem and automatic differentiation/backpropagation, showing that these two seemingly disparate approaches are actually tightly connected. Our framework is extremely flexible, allowing its subproblems to be solved with any suitable method, to any degree of accuracy. We derive a priori and computable a posteriori error bounds for all our methods and numerically show that our a posteriori bounds are usually more accurate. Our numerical results also show that, surprisingly, for efficient bilevel optimization, the choice of hypergradient algorithm is at least as important as the choice of lower-level solver.