Data-Driven Performance Guarantees for Classical and Learned Optimizers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sambharya, Rajiv, Stellato, Bartolomeo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911193323536384
author Sambharya, Rajiv
Stellato, Bartolomeo
author_facet Sambharya, Rajiv
Stellato, Bartolomeo
contents We introduce a data-driven approach to analyze the performance of continuous optimization algorithms using generalization guarantees from statistical learning theory. We study classical and learned optimizers to solve families of parametric optimization problems. We build generalization guarantees for classical optimizers, using a sample convergence bound, and for learned optimizers, using the Probably Approximately Correct (PAC)-Bayes framework. To train learned optimizers, we use a gradient-based algorithm to directly minimize the PAC-Bayes upper bound. Numerical experiments in signal processing, control, and meta-learning showcase the ability of our framework to provide strong generalization guarantees for both classical and learned optimizers given a fixed budget of iterations. For classical optimizers, our bounds are much tighter than those that worst-case guarantees provide. For learned optimizers, our bounds outperform the empirical outcomes observed in their non-learned counterparts.
format Preprint
id arxiv_https___arxiv_org_abs_2404_13831
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data-Driven Performance Guarantees for Classical and Learned Optimizers
Sambharya, Rajiv
Stellato, Bartolomeo
Optimization and Control
Machine Learning
We introduce a data-driven approach to analyze the performance of continuous optimization algorithms using generalization guarantees from statistical learning theory. We study classical and learned optimizers to solve families of parametric optimization problems. We build generalization guarantees for classical optimizers, using a sample convergence bound, and for learned optimizers, using the Probably Approximately Correct (PAC)-Bayes framework. To train learned optimizers, we use a gradient-based algorithm to directly minimize the PAC-Bayes upper bound. Numerical experiments in signal processing, control, and meta-learning showcase the ability of our framework to provide strong generalization guarantees for both classical and learned optimizers given a fixed budget of iterations. For classical optimizers, our bounds are much tighter than those that worst-case guarantees provide. For learned optimizers, our bounds outperform the empirical outcomes observed in their non-learned counterparts.
title Data-Driven Performance Guarantees for Classical and Learned Optimizers
topic Optimization and Control
Machine Learning
url https://arxiv.org/abs/2404.13831