OCLGMLFeb 12, 2024

Tuning-Free Stochastic Optimization

Princeton
arXiv:2402.07793v213 citationsh-index: 14ICML
Originality Incremental advance
AI Analysis

This addresses the prohibitive cost of hyperparameter tuning for practitioners in machine learning, though it is incremental as it builds on existing algorithms and theory.

The paper tackles the problem of hyperparameter tuning in large-scale machine learning by formalizing 'tuning-free' algorithms that match optimally-tuned Stochastic Gradient Descent (SGD) performance up to polylogarithmic factors, showing feasibility in bounded domains and impossibility in unbounded ones under certain conditions.

Large-scale machine learning problems make the cost of hyperparameter tuning ever more prohibitive. This creates a need for algorithms that can tune themselves on-the-fly. We formalize the notion of "tuning-free" algorithms that can match the performance of optimally-tuned optimization algorithms up to polylogarithmic factors given only loose hints on the relevant problem parameters. We consider in particular algorithms that can match optimally-tuned Stochastic Gradient Descent (SGD). When the domain of optimization is bounded, we show tuning-free matching of SGD is possible and achieved by several existing algorithms. We prove that for the task of minimizing a convex and smooth or Lipschitz function over an unbounded domain, tuning-free optimization is impossible. We discuss conditions under which tuning-free optimization is possible even over unbounded domains. In particular, we show that the recently proposed DoG and DoWG algorithms are tuning-free when the noise distribution is sufficiently well-behaved. For the task of finding a stationary point of a smooth and potentially nonconvex function, we give a variant of SGD that matches the best-known high-probability convergence rate for tuned SGD at only an additional polylogarithmic cost. However, we also give an impossibility result that shows no algorithm can hope to match the optimal expected convergence rate for tuned SGD with high probability.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes