Gradient Descent is Pareto-Optimal in the Oracle Complexity and Memory Tradeoff for Feasibility Problems
In this paper we provide oracle complexity lower bounds for finding a point in a given set using a memory-constrained algorithm that has access to a separation oracle. We assume that the set is contained within the unit <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$d-\mathbf{dimensional}$</tex> ball and contains a ball of known radius <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\epsilon > 0$</tex>. This setup is commonly referred to as the feasibility problem. We show that to solve feasibility problems with accuracy <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\epsilon\geq e^{-d^{o(1)}}$</tex>, any deterministic algorithm either uses <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$d^{1+\delta}$</tex> bits of memory or must make at least <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$1/(d^{0.01\delta}\epsilon^{2\frac{1-\delta}{1+1.01\delta}-o(1)})$</tex> oracle queries, for any <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\delta\in[0,1]$</tex>. Additionally, we show that randomized algorithms either use <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$d^{1+\delta}$</tex> memory or make at least <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$1/(d^{2\delta}\epsilon^{2(1-4\delta)-o(1)})$</tex> queries for any <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\delta\in[0, \frac{1}{4}]$</tex>. Because gradient descent only uses linear memory <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathcal{O}(d\ln 1/\epsilon)$</tex> but makes <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathcal{O}(1/\epsilon^{2})$</tex> queries, our results imply that it is Pareto-optimal in the oracle complexity/memory tradeoff. Further, our results show that the oracle complexity for deterministic algorithms is always polynomial in <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$1/\epsilon$</tex> if the algorithm has less than quadratic memory in <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$d$</tex>. This reveals a sharp phase transition since with quadratic <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathcal{O}(d^{2}\ln 1/\epsilon)$</tex> memory, cutting plane methods only require <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathcal{O}(d\ln 1/\epsilon)$</tex> queries.
Authors
- Moïse Blanchard (ORCID: https://orcid.org/0000-0003-0593-8666)
Institutions
- Massachusetts Institute of Technology (US)
Publication Details
- Published
- 2024-10-27
- DOI
- https://doi.org/10.1109/focs61266.2024.00141
- Primary Topic
- Stochastic Gradient Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Office of Naval Research
- Air Force Office of Scientific Research