Directional Smoothness and Gradient Methods: Convergence and Adaptivity

Aaron Mishkin, Ahmed Khaled, Yuanhao Wang, Aaron Defazio, Robert M. Gower
3/6/2024
cs.LGmath.OC

No official code found. Best implementation: aaronpmishkin/directional_smoothness_poster (0 stars, last commit 2025-01-11). 1 more below.

Abstract

We develop new sub-optimality bounds for gradient descent (GD) that depend on the conditioning of the objective along the path of optimization rather than on global, worst-case constants. Key to our proofs is directional smoothness, a measure of gradient variation that we use to develop upper-bounds on the objective. Minimizing these upper-bounds requires solving implicit equations to obtain a sequence of strongly adapted step-sizes; we show that these equations are straightforward to solve for convex quadratics and lead to new guarantees for two classical step-sizes. For general functions, we prove that the Polyak step-size and normalized GD obtain fast, path-dependent rates despite using no knowledge of the directional smoothness. Experiments on logistic regression show our convergence guarantees are tighter than the classical theory based on $L$-smoothness.

View on arXivView PDF

Code Implementations(2)

LaTeX code for NeurIPS 2024 poster on "Directional Smoothness and Gradient Methods: Convergence and Adaptivity".

00TeX, PerlJan 11, 20251 years ago
BSD-3-Clause

Code to replicate experiments in the NeurIPS 2024 paper "Directional Smoothness and Gradient Methods: Convergence and Adaptivity".

00Shell, PythonJan 11, 20251 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{mishkin2024directional,
  title  = {Directional Smoothness and Gradient Methods: Convergence and Adaptivity},
  author = {Aaron Mishkin and Ahmed Khaled and Yuanhao Wang and Aaron Defazio and Robert M. Gower},
  year   = {2024},
  eprint = {2403.04081},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2403.04081},
  url    = {http://arxiv.org/abs/2403.04081v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.04081.svg)](https://codeofpaper.com/paper/2403.04081)

Discussion