Memory-Query Tradeoffs for Randomized Convex Optimization
Xi Chen, Binghui Peng
摘要
We show that any randomized first-order algorithm which minimizes a d-dimensional, 1-Lipschitz convex function over the unit ball must either use Ω(d 2-δ ) bits of memory or make Ω(d 1+δ/6-o(1) ) queries, for any constant δ ∈ (0, 1) and when the precision ǫ is quasipolynomially small in d. Our result implies that cutting plane methods, which use Õ(d 2 ) bits of memory and Õ(d) queries, are Pareto-optimal among randomized first-order algorithms, and quadratic memory is required to achieve optimal query complexity for convex optimization.
Warm up: Encoding in the bit model Our encoding argument is delicate and we first illustrate the basic idea in a simpler "bit model" of the correlated orthogonal vector game: Alice sends n bits (instead rows) in the third round. This is strictly weaker than the original model because n rows can transmit at least n bits of message (even when they are required to be row vectors of A).
In this model, Bob's output vector x, is a function of the first round message M ∈ 0, 1 kd , the third round message b ∈ 0, 1 n and its input vector v ∈ 1 √ d H d , and we denote as x M,v,b . Fix a matrix A ∈ -1, 1 (d/2)×d and the first round message M A , the choice of v is still uniformly at random, and we observe that the collection x M A ,v,b v∈ 1 √ d H d ,b∈0,1 n should contain at least s linearly independent vectors that are orthogonal to A, and this continues to hold w.h.p. when one restricts to a subset V * ⊆ H d of size O(s). That is to say, using a probabilistic argument, there exists a set V * ⊆ 1
such that x M A ,v,b v∈V * ,b∈0,1 n contains at least s linearly independent vectors that are orthogonal to A, for at least 1/2-fraction of matrices A ∈ 0, 1 (d/2)×d . We denote this set as A nice
Consider the following (one-shot) encoding strategy: Fix a message M ∈ 0, 1 kd , a s-tuple x 1 , . . . , x s ∈ x M,v,b v∈V * ,b∈0,1 n that are linearly independent, include all matrices A ∈ -1, 1 (d/2)×d whose rows are orthogonal to x 1 , . . . , x s . The above argument implies that all matrices A ∈ A nice are included, but on the hand, the encoding scheme only encodes
Here the first term of LHS is the number of message M, the second term is the number of s-tuple x 1 , . . . , x s (note this is the place we need |V * | ≤ O(s)) and the third term is the number of matrices A that are orthogonal to the s-tuple.
Encoding in the row model We next design an encoding strategy based on the (too-goodto-be-true) communication protocol that sends n rows instead of n bits, which turns out to be much more challenging and requires an iterative encoding. In the row model, the output x of Bob depends on the message M ∈ 0, 1 kd , the vector v ∈ 1 √ d
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper2
- I/O Complexity of Attention, or How Optimal is FlashAttention?Barna Saha, Christopher YeICML 2024 · 被引用 4 次
- Gradient Descent is Pareto-Optimal in the Oracle Complexity and Memory Tradeoff for Feasibility ProblemsMoïse BlanchardFOCS 2024 · 被引用 3 次
它引用的顶会 Paper10
- An improved cutting plane method for convex optimization, convex-concave games, and its applicationsHaotian Jiang, Yin Tat Lee, Zhao Song, Sam Chiu-wai WongSTOC 2020 · 被引用 54 次
- When is memorization of irrelevant training data necessary for high-accuracy learning?Gavin Brown, Mark Bun, Vitaly Feldman, Adam D. Smith 等STOC 2021 · 被引用 33 次
- Does learning require memorization? a short tale about a long tailVitaly FeldmanSTOC 2020 · 被引用 28 次
- Towards a Combinatorial Characterization of Bounded-Memory LearningAlon Gonen, Shachar Lovett, Michal MoshkovitzNeurIPS 2020 · 被引用 9 次
- Memory Bounds for Continual LearningXi Chen, Christos H. Papadimitriou, Binghui PengFOCS 2022 · 被引用 6 次
相关 Paper
- Memory-Constrained Algorithms for Convex OptimizationMoïse Blanchard, Junhui Zhang, Patrick JailletNeurIPS 2023 · 被引用 4 次
- The Communication Complexity of OptimizationSantosh S. Vempala, Ruosong Wang, David P. WoodruffSODA 2020 · 被引用 16 次
- Tight Bounds for the Subspace Sketch Problem with ApplicationsYi Li, Ruosong Wang, David P. WoodruffSODA 2020 · 被引用 5 次
- A Whole New Ball Game: A Primal Accelerated Method for Matrix Games and Minimizing the Maximum of Smooth FunctionsYair Carmon, Arun Jambulapati, Yujia Jin, Aaron SidfordSODA 2024
- Near-Optimal Quantum Algorithm for Minimizing the Maximal LossHao Wang, Chenyi Zhang, Tongyang LiICLR 2024 · 被引用 1 次
