Breaking the Linear Iteration Cost Barrier for Some Well-known Conditional Gradient Methods Using MaxIP Data-structures
Zhaozhuo Xu, Zhao Song, Anshumali Shrivastava
Abstract
Conditional gradient methods (CGM) are widely used in modern machine learning. CGM's overall running time usually consists of two parts: the number of iterations and the cost of each iteration. Most efforts focus on reducing the number of iterations as a means to reduce the overall running time. In this work, we focus on improving the per iteration cost of CGM. The bottleneck step in most CGM is maximum inner product search (MaxIP), which requires a linear scan over the parameters. In practice, approximate MaxIP data-structures are found to be helpful heuristics. However, theoretically, nothing is known about the combination of approximate MaxIP data-structures and CGM. In this work, we answer this question positively by providing a formal framework to combine the locality sensitive hashing type approximate MaxIP data-structures with CGM algorithms. As a result, we show the first algorithm, where the cost per iteration is sublinear in the number of parameters, for many fundamental optimization algorithms, e.g., Frank-Wolfe, Herding algorithm, and policy gradient.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ffb0bc8e-d005-4078-9cc3-12c036c4e9b9Cited by top-tier papers14
- Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear TimeYuzhou Gu, Zhao Song, Junze Yin, Lichen ZhangICLR 2024 · 37 citations
- Sketching for First Order Method: Efficient Algorithm for Low-Bandwidth Channel and VulnerabilityZhao Song, Yitan Wang, Zheng Yu, Lichen ZhangICML 2023 · 35 citations
- Sketching Meets Differential Privacy: Fast Algorithm for Dynamic Kronecker Projection MaintenanceZhao Song, Xin Yang, Yuanyuan Yang, Lichen ZhangICML 2023 · 30 citations
- A Sublinear Adversarial Training AlgorithmYeqi Gao, Lianke Qin, Zhao Song, Yitan WangICLR 2024 · 27 citations
- Dynamic Tensor Product RegressionAravind Reddy, Zhao Song, Lichen ZhangNeurIPS 2022 · 22 citations
Builds on15
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Accelerating Large-Scale Inference with Anisotropic Vector QuantizationRuiqi Guo, Philip Sun, Erik Lindgren, Quan Geng et al.ICML 2020 · 539 citations
- A Deterministic Linear Program Solver in Current Matrix Multiplication TimeJan van den BrandSODA 2020 · 107 citations
- Learning Space Partitions for Nearest Neighbor SearchYihe Dong, Piotr Indyk, Ilya P. Razenshteyn, Tal WagnerICLR 2020 · 104 citations
- Bipartite Matching in Nearly-linear Time on Moderately Dense GraphsJan van den Brand, Yin Tat Lee, Danupon Nanongkai, Richard Peng et al.FOCS 2020 · 72 citations
Related papers
- Pairwise Conditional Gradients without Swap Steps and Sparser Kernel HerdingKazuma Tsuji, Ken'ichiro Tanaka, Sebastian PokuttaICML 2022 · 31 citations
- ProMIPS: Efficient High-Dimensional c-Approximate Maximum Inner Product Search with a Lightweight IndexYang Song, Yu Gu, Rui Zhang, Ge YuICDE 2021 · 16 citations
- SAH: Shifting-Aware Asymmetric Hashing for Reverse k Maximum Inner Product SearchQiang Huang, Yanhao Wang, Anthony K. H. TungAAAI 2023 · 6 citations
- Faster Randomized Infeasible Interior Point Methods for Tall/Wide Linear ProgramsAgniva Chowdhury, Palma London, Haim Avron, Petros DrineasNeurIPS 2020 · 7 citations
- Reusing Combinatorial Structure: Faster Iterative Projections over Submodular Base PolytopesJai Moondra, Hassan Mortagy, Swati GuptaNeurIPS 2021 · 5 citations
