Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Liang, Yang, Junchi, Karbasi, Amin, He, Niao
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910292521254912
author Zhang, Liang
Yang, Junchi
Karbasi, Amin
He, Niao
author_facet Zhang, Liang
Yang, Junchi
Karbasi, Amin
He, Niao
contents Algorithmic reproducibility measures the deviation in outputs of machine learning algorithms upon minor changes in the training process. Previous work suggests that first-order methods would need to trade-off convergence rate (gradient complexity) for better reproducibility. In this work, we challenge this perception and demonstrate that both optimal reproducibility and near-optimal convergence guarantees can be achieved for smooth convex minimization and smooth convex-concave minimax problems under various error-prone oracle settings. Particularly, given the inexact initialization oracle, our regularization-based algorithms achieve the best of both worlds - optimal reproducibility and near-optimal gradient complexity - for minimization and minimax optimization. With the inexact gradient oracle, the near-optimal guarantees also hold for minimax optimization. Additionally, with the stochastic gradient oracle, we show that stochastic gradient descent ascent is optimal in terms of both reproducibility and gradient complexity. We believe our results contribute to an enhanced understanding of the reproducibility-convergence trade-off in the context of convex optimization.
format Preprint
id arxiv_https___arxiv_org_abs_2310_17759
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization
Zhang, Liang
Yang, Junchi
Karbasi, Amin
He, Niao
Machine Learning
Optimization and Control
Algorithmic reproducibility measures the deviation in outputs of machine learning algorithms upon minor changes in the training process. Previous work suggests that first-order methods would need to trade-off convergence rate (gradient complexity) for better reproducibility. In this work, we challenge this perception and demonstrate that both optimal reproducibility and near-optimal convergence guarantees can be achieved for smooth convex minimization and smooth convex-concave minimax problems under various error-prone oracle settings. Particularly, given the inexact initialization oracle, our regularization-based algorithms achieve the best of both worlds - optimal reproducibility and near-optimal gradient complexity - for minimization and minimax optimization. With the inexact gradient oracle, the near-optimal guarantees also hold for minimax optimization. Additionally, with the stochastic gradient oracle, we show that stochastic gradient descent ascent is optimal in terms of both reproducibility and gradient complexity. We believe our results contribute to an enhanced understanding of the reproducibility-convergence trade-off in the context of convex optimization.
title Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2310.17759