A Generalization Result for Convergence in Learning-to-Optimize

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Sucker, Michael, Ochs, Peter
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908385851473920
author Sucker, Michael
Ochs, Peter
author_facet Sucker, Michael
Ochs, Peter
contents Learning-to-optimize leverages machine learning to accelerate optimization algorithms. While empirical results show tremendous improvements compared to classical optimization algorithms, theoretical guarantees are mostly lacking, such that the outcome cannot be reliably assured. Especially, convergence is hardly studied in learning-to-optimize, because conventional convergence guarantees in optimization are based on geometric arguments, which cannot be applied easily to learned algorithms. Thus, we develop a probabilistic framework that resembles classical optimization and allows for transferring geometric arguments into learning-to-optimize. Based on our new proof-strategy, our main theorem is a generalization result for parametric classes of potentially non-smooth, non-convex loss functions and establishes the convergence of learned optimization algorithms to critical points with high probability. This effectively generalizes the results of a worst-case analysis into a probabilistic framework, and frees the design of the learned algorithm from using safeguards.
format Preprint
id arxiv_https___arxiv_org_abs_2410_07704
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Generalization Result for Convergence in Learning-to-Optimize
Sucker, Michael
Ochs, Peter
Machine Learning
Optimization and Control
Probability
Learning-to-optimize leverages machine learning to accelerate optimization algorithms. While empirical results show tremendous improvements compared to classical optimization algorithms, theoretical guarantees are mostly lacking, such that the outcome cannot be reliably assured. Especially, convergence is hardly studied in learning-to-optimize, because conventional convergence guarantees in optimization are based on geometric arguments, which cannot be applied easily to learned algorithms. Thus, we develop a probabilistic framework that resembles classical optimization and allows for transferring geometric arguments into learning-to-optimize. Based on our new proof-strategy, our main theorem is a generalization result for parametric classes of potentially non-smooth, non-convex loss functions and establishes the convergence of learned optimization algorithms to critical points with high probability. This effectively generalizes the results of a worst-case analysis into a probabilistic framework, and frees the design of the learned algorithm from using safeguards.
title A Generalization Result for Convergence in Learning-to-Optimize
topic Machine Learning
Optimization and Control
Probability
url https://arxiv.org/abs/2410.07704