Proceedings

Efficient Convex Optimization Requires Superlinear Memory (Extended Abstract)

Download BibTeX Cite

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Sister Conferences Best Papers. Pages 6468-6473.

https://doi.org/10.24963/ijcai.2023/722

Abstract

Minimizing a convex function with access to a first order oracle---that returns the function evaluation and (sub)gradient at a query point---is a canonical optimization problem and a fundamental primitive in machine learning. Gradient-based methods are the most popular approaches used for solving the problem, owing to their simplicity and computational efficiency. These methods, however, do not achieve the information-theoretically optimal query complexity for minimizing the underlying function to small error, which are achieved by more expensive techniques based on cutting-plane methods. Is it possible to achieve the information-theoretically query complexity without using these more complex and computationally expensive methods? In this work, we use memory as a lens to understand this, and show that is is not possible to achieve optimal query complexity without using significantly more memory than that used by gradient descent.

  1. [Alon et al., 1999] Noga Alon, Yossi Matias, and Mario Szegedy. The space complexity of approximating the fre- quency moments. Journal of Computer and system sci- ences, 58(1):137-147, 1999.
  2. [Anstreicher, 2000] Kurt M Anstreicher. The volumetric bar- rier for semidefinite programming. Mathematics of Oper- ations Research, 25(3):365-380, 2000.
  3. [Arjevani and Shamir, 2015] Yossi Arjevani and Ohad Shamir. Communication complexity of distributed convex learning and optimization. Advances in neural information processing systems, 28, 2015.
  4. [Atkinson and Vaidya, 1995] David S Atkinson and Pravin M Vaidya. A cutting plane algorithm for convex programming that uses analytic centers. Mathematical Programming, 69(1):1-43, 1995.
  5. [Balcan et al., 2012] Maria Florina Balcan, Avrim Blum, Shai Fine, and Yishay Mansour. Distributed learning, communication complexity and privacy. In Conference on Learning Theory, pages 26-1. JMLR Workshop and Con- ference Proceedings, 2012.
  6. [Balkanski and Singer, 2018] Eric Balkanski and Yaron Singer. Parallelization does not accelerate convex opti- mization: Adaptivity lower bounds for non-smooth convex minimization. arXiv preprint arXiv:1808.03880, 2018.
  7. [Bar-Yossef et al., 2004] Ziv Bar-Yossef, Thathachar S Jayram, Ravi Kumar, and D Sivakumar. An information statistics approach to data stream and communication complexity. Journal of Computer and System Sciences, 68(4):702-732, 2004.
  8. [Beame et al., 2018] Paul Beame, Shayan Oveis Gharan, and Xin Yang. Time-space tradeoffs for learning finite func- tions from random evaluations, with applications to poly- nomials. In Conference On Learning Theory, pages 843- 856. PMLR, 2018.
  9. [Bertsimas and Vempala, 2004] Dimitris Bertsimas and San- tosh Vempala. Solving convex programs by random walks. Journal of the ACM (JACM), 51(4):540-556, 2004.
  10. [Braun et al., 2017] Gábor Braun, Cristóbal Guzmán, and Sebastian Pokutta. Lower bounds on the oracle complexity of nonsmooth convex optimization via information theory. IEEE Transactions on Information Theory, 63(7), 2017.
  11. [Braverman et al., 2016] Mark Braverman, Ankit Garg, Tengyu Ma, Huy L Nguyen, and David P Woodruff. Com- munication lower bounds for statistical estimation prob- lems via a distributed data processing inequality. In Pro- ceedings of the forty-eighth annual ACM symposium on Theory of Computing, pages 1011-1020, 2016.
  12. [Braverman et al., 2020] Mark Braverman, Elad Hazan, Max Simchowitz, and Blake Woodworth. The gradient complexity of linear regression. In Conference on Learn- ing Theory, pages 627-647. PMLR, 2020.
  13. [Bubeck et al., 2019] Sébastien Bubeck, Qijia Jiang, Yin-Tat Lee, Yuanzhi Li, and Aaron Sidford. Complexity of highly parallel non-smooth convex optimization. Advances in Neural Information Processing Systems, 32, 2019.
  14. [Bubeck, 2014] Sébastien Bubeck. Convex optimiza- tion: Algorithms and complexity. arXiv preprint arXiv:1405.4980, 2014.
  15. [Clarkson and Woodruff, 2009] Kenneth L Clarkson and David P Woodruff. Numerical linear algebra in the stream- ing model. In Proceedings of the forty-first annual ACM symposium on Theory of computing, pages 205-214, 2009.
  16. [Dagan and Shamir, 2018] Yuval Dagan and Ohad Shamir. Detecting correlations with little memory and communi- cation. In Conference On Learning Theory, pages 1145- 1198. PMLR, 2018.
  17. [Dagan et al., 2019] Yuval Dagan, Gil Kur, and Ohad Shamir. Space lower bounds for linear prediction in the streaming model. In Conference on Learning Theory, pages 929-954. PMLR, 2019.
  18. [Duchi et al., 2013] John C Duchi, Michael I Jordan, and Martin J Wainwright. Local privacy and statistical mini- max rates. In 2013 IEEE 54th Annual Symposium on Foun- dations of Computer Science, pages 429-438. IEEE, 2013.
  19. [Fletcher and Reeves, 1964] R. Fletcher and C. M. Reeves. Function minimization by conjugate gradients. The Com- puter Journal, 7(2), 1964.
  20. [Garg et al., 2014] Ankit Garg, Tengyu Ma, and Huy Nguyen. On communication cost of distributed statistical estimation and dimensionality. Advances in Neural Infor- mation Processing Systems, 27, 2014.
  21. [Garg et al., 2018] Sumegha Garg, Ran Raz, and Avishay Tal. Extractor-based time-space lower bounds for learning. In Proceedings of the 50th Annual ACM SIGACT Sympo- sium on Theory of Computing, pages 990-1002, 2018.
  22. [Hestenes and Stiefel, 1952] M.R. Hestenes and E. Stiefel. Methods of conjugate gradients for solving linear systems. Journal of Research of the National Bureau of Standards, 49(6), 1952.
  23. [Khachiyan et al., 1988] Leonid G Khachiyan, Sergei Pavlovich Tarasov, and II Erlikh. The method of inscribed ellipsoids. In Soviet Math. Dokl, volume 37, pages 226-230, 1988.
  24. [Kol et al., 2017] Gillat Kol, Ran Raz, and Avishay Tal. Time-space hardness of learning sparse parities. In Pro- ceedings of the 49th Annual ACM SIGACT Symposium on Theory of Computing, pages 1067-1080, 2017.
  25. [Levin, 1965] Anatoly Yur'evich Levin. An algorithm for minimizing convex functions. In Doklady Akademii Nauk, volume 160, pages 1244-1247. Russian Academy of Sci- ences, 1965.
  26. [Liu and Nocedal, 1989] Dong C. Liu and Jorge Nocedal. On the limited memory BFGS method for large scale op- timization. Mathematical Programming, 45(1-3), 1989.
  27. [McCormick, 2005] S Thomas McCormick. Submodular function minimization. Handbooks in operations research and management science, 12:321-391, 2005.
  28. [Moshkovitz and Moshkovitz, 2017] Dana Moshkovitz and Michal Moshkovitz. Mixing implies lower bounds for space bounded learning. In Conference on Learning The- ory, pages 1516-1566. PMLR, 2017.
  29. [Moshkovitz and Moshkovitz, 2018] Dana Moshkovitz and Michal Moshkovitz. Entropy samplers and strong generic lower bounds for space bounded learning. In 9th In- novations in Theoretical Computer Science Conference. Schloss Dagstuhl-Leibniz-Zentrum fuer Informatik, 2018.
  30. [Nemirovski and Yudin, 1983] Arkadij Nemirovski and David Yudin. Problem complexity and method efficiency in optimization. Wiley-Interscience, 1983.
  31. [Nemirovski, 1994] Arkadi Nemirovski. On parallel com- plexity of nonsmooth convex optimization. Journal of Complexity, 10(4):451-463, 1994.
  32. [Nesterov, 1989] Ju E Nesterov. Self-concordant func- tions and polynomial-time methods in convex program- ming. Report, Central Economic and Mathematic Insti- tute, USSR Acad. Sci, 1989.
  33. [Nesterov, 2003] Yurii Nesterov. Introductory lectures on convex optimization: A basic course, volume 87. Springer Science & Business Media, 2003.
  34. [Nocedal, 1980] Jorge Nocedal. Updating quasi-Newton ma- trices with limited storage. Mathematics of Computation, 35(151), 1980.
  35. [Papadimitriou and Roughgarden, 2008] Christos H Pa- padimitriou and Tim Roughgarden. Computing correlated equilibria in multi-player games. Journal of the ACM (JACM), 55(3):1-29, 2008.
  36. [Pilanci and Wainwright, 2017] Mert Pilanci and Martin J Wainwright. Newton sketch: A near linear-time optimiza- tion algorithm with linear-quadratic convergence. SIAM Journal on Optimization, 27(1):205-245, 2017.
  37. [Raz, 2017] Ran Raz. A time-space lower bound for a large class of learning problems. In 2017 IEEE 58th Annual Symposium on Foundations of Computer Science, pages 732-742. IEEE, 2017.
  38. [Raz, 2018] Ran Raz. Fast learning requires good memory: A time-space lower bound for parity learning. Journal of the ACM, 66(1):1-18, 2018.
  39. [Shamir, 2014] Ohad Shamir. Fundamental limits of online and distributed algorithms for statistical learning and esti- mation. Advances in Neural Information Processing Sys- tems, 27, 2014.
  40. [Sharan et al., 2019] Vatsal Sharan, Aaron Sidford, and Gre- gory Valiant. Memory-sample tradeoffs for linear regres- sion with small error. In Proceedings of the 51st Annual ACM SIGACT Symposium on Theory of Computing, pages 890-901, 2019.
  41. [Shor, 1977] Naum Z Shor. Cut-off method with space ex- tension in convex programming problems. Cybernetics, 13(1):94-96, 1977.
  42. [Simchowitz et al., 2018] Max Simchowitz, Ahmed El Alaoui, and Benjamin Recht. Tight query com- plexity lower bounds for pca via finite sample deformed wigner law. In Proceedings of the 50th Annual ACM SIGACT Symposium on Theory of Computing, pages 1249-1259, 2018.
  43. [Simchowitz, 2018] Max Simchowitz. On the randomized complexity of minimizing a convex quadratic function. arXiv preprint arXiv:1807.09386, 2018.
  44. [Steinhardt and Duchi, 2015] Jacob Steinhardt and John Duchi. Minimax rates for memory-bounded sparse lin- ear regression. In Conference on Learning Theory, pages 1564-1587. PMLR, 2015.
  45. [Steinhardt et al., 2016] Jacob Steinhardt, Gregory Valiant, and Stefan Wager. Memory, communication, and statis- tical queries. In Conference on Learning Theory, pages 1490-1516. PMLR, 2016.
  46. [Sun et al., 2021] Xiaoming Sun, David P Woodruff, Guang Yang, and Jialin Zhang. Querying a matrix through matrix-vector products. ACM Transactions on Algorithms (TALG), 17(4):1-19, 2021.
  47. [Vaidya, 1989] Pravin M Vaidya. A new algorithm for mini- mizing convex functions over convex sets. In 30th Annual Symposium on Foundations of Computer Science, pages 338-343. IEEE Computer Society, 1989.
  48. [Woodworth and Srebro, 2016] Blake E Woodworth and Nati Srebro. Tight complexity bounds for optimizing com- posite objectives. Advances in neural information process- ing systems, 29, 2016.
  49. [Woodworth and Srebro, 2017] Blake Woodworth and Nathan Srebro. Lower bound for randomized first order convex optimization. arXiv preprint arXiv:1709.03594, 2017.
  50. [Woodworth and Srebro, 2019] Blake Woodworth and Nathan Srebro. Open problem: The oracle complexity of convex optimization with limited memory. In Conference on Learning Theory, pages 3202-3210. PMLR, 2019.
  51. [Woodworth et al., 2021] Blake E Woodworth, Brian Bullins, Ohad Shamir, and Nathan Srebro. The min-max complexity of distributed stochastic convex optimization with intermittent communication. In Conference on Learning Theory, pages 4386-4437. PMLR, 2021.
  52. [Xu et al., 2020] Peng Xu, Fred Roosta, and Michael W Ma- honey. Newton-type methods for non-convex optimiza- tion under inexact hessian information. Mathematical Pro- gramming, 184(1):35-70, 2020.
  53. [Yudin and Nemirovskii, 1976] DB Yudin and Arkadi S Ne- mirovskii. Informational complexity and efficient methods for the solution of convex extremal problems. Matekon, 13(2):22-45, 1976.
  54. [Zhang et al., 2013] Yuchen Zhang, John C Duchi, Michael I Jordan, and Martin J Wainwright. Information-theoretic lower bounds for distributed statistical estimation with communication constraints. In Conference on Neural In- formation Processing Systems, pages 2328-2336. Cite- seer, 2013.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards