Streaming Semidefinite Programs: O(√n) Passes, Small Space and Fast Runtime

We study the problem of solving semidefinite programs (SDP) in the streaming model. Specifically, $m$ constraint matrices and a target matrix $C$, all of size $n\times n$ together with a vector $b\in \mathbb{R}^m$ are streamed to us one-by-one. The goal is to find a matrix $X\in \mathbb{R}^{n\times n}$ such that $\langle C, X\rangle$ is maximized, subject to $\langle A_i, X\rangle=b_i$ for all $i\in [m]$ and $X\succeq 0$. Previous algorithmic studies of SDP primarily focus on \emph{time-efficiency}, and all of them require a prohibitively large $\Omega(mn^2)$ space in order to store \emph{all the constraints}. Such space consumption is necessary for fast algorithms as it is the size of the input. In this work, we design an interior point method (IPM) that uses $\widetilde O(m^2+n^2)$ space, which is strictly sublinear in the regime $n\gg m$. Our algorithm takes $O(\sqrt n\log(1/\epsilon))$ passes, which is standard for IPM. Moreover, when $m$ is much smaller than $n$, our algorithm also matches the time complexity of the state-of-the-art SDP solvers. To achieve such a sublinear space bound, we design a novel sketching method that enables one to compute a spectral approximation to the Hessian matrix in $O(m^2)$ space. To the best of our knowledge, this is the first method that successfully applies sketching technique to improve SDP algorithm in terms of space (also time).

[1]  V. V. Williams,et al.  New Bounds for Matrix Multiplication: from Alpha to Omega , 2023, ArXiv.

[2]  Yuzhou Gu,et al.  A Nearly-Linear Time Algorithm for Structured Support Vector Machines , 2023, ArXiv.

[3]  Franccois Le Gall Faster Rectangular Matrix Multiplication by Combination Loss Analysis , 2023, ArXiv.

[4]  Mingquan Ye,et al.  Efficient Alternating Minimization with Applications to Weighted Low Rank Approximation , 2023, ArXiv.

[5]  A. Jambulapati,et al.  Linear-Sized Sparsifiers via Near-Linear Time Discrepancy Theory , 2023, ArXiv.

[6]  Zhao Song,et al.  Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Time , 2023, ArXiv.

[7]  Zhao Song,et al.  A Faster Small Treewidth SDP Solver , 2022, ArXiv.

[8]  Xin Yang,et al.  Sketching Meets Differential Privacy: Fast Algorithm for Dynamic Kronecker Projection Maintenance , 2022, ICML.

[9]  Ran Duan,et al.  Faster Matrix Multiplication via Asymmetric Hashing , 2022, ArXiv.

[10]  Zhao Song,et al.  Dynamic Tensor Product Regression , 2022, NeurIPS.

[11]  Samson Zhou,et al.  Learning-Augmented Algorithms for Online Linear and Semidefinite Programming , 2022, NeurIPS.

[12]  Sayan Bhattacharya,et al.  Dynamic Algorithms for Packing-Covering LPs via Multiplicative Weight Updates , 2022, SODA.

[13]  Zhaozhuo Xu,et al.  Speeding Up Sparsification using Inner Product Search Data Structures , 2022, ArXiv.

[14]  Omri Weinstein,et al.  A Faster Interior-Point Method for Sum-of-Squares Optimization , 2022, Algorithmica.

[15]  David P. Woodruff,et al.  Leverage Score Sampling for Tensor Product Matrices in Input Sparsity Time , 2022, ICML.

[16]  Zhao Song,et al.  Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Time , 2021, ITCS.

[17]  Prasad Raghavendra,et al.  Matrix discrepancy from Quantum communication , 2021, STOC.

[18]  Yeshwanth Cherapanamjeri,et al.  Terminal Embeddings in Sublinear Time , 2021, 2021 IEEE 62nd Annual Symposium on Foundations of Computer Science (FOCS).

[19]  David P. Woodruff,et al.  Fast Sketching of Polynomial Kernels of Polynomial Degree , 2021, ICML.

[20]  Jinwoo Shin,et al.  Scaling Neural Tangent Kernels via Sketching and Random Features , 2021, NeurIPS.

[21]  Shunhua Jiang,et al.  Solving SDP Faster: A Robust IPM Framework and Efficient Implementation , 2021, 2022 IEEE 63rd Annual Symposium on Foundations of Computer Science (FOCS).

[22]  Josh Alman,et al.  A Refined Laser Method and Faster Matrix Multiplication , 2020, SODA.

[23]  Yin Tat Lee,et al.  A Faster Interior Point Method for Semidefinite Programming , 2020, 2020 IEEE 61st Annual Symposium on Foundations of Computer Science (FOCS).

[24]  Hengjie Zhang,et al.  Space-Efficient Interior Point Method, with applications to Linear Programming and Maximum Weight Bipartite Matching , 2020, ICALP.

[25]  Omri Weinstein,et al.  Training (Overparametrized) Neural Networks in Near-Linear Time , 2020, ITCS.

[26]  Kevin Tian,et al.  Robust Sub-Gaussian Principal Component Analysis and Width-Independent Schatten Packing , 2020, NeurIPS.

[27]  Omri Weinstein,et al.  Faster Dynamic Matrix Inverse for Faster LPs , 2020, ArXiv.

[28]  Yin Tat Lee,et al.  An improved cutting plane method for convex optimization, convex-concave games, and its applications , 2020, STOC.

[29]  Ankur Moitra,et al.  Learning Structured Distributions From Untrusted Batches: Faster and Simpler , 2020, NeurIPS.

[30]  Yin Tat Lee,et al.  Solving Linear Programs with Sqrt(rank) Linear System Solves , 2019, ArXiv.

[31]  David P. Woodruff,et al.  Optimal Sketching for Kronecker Product Regression and Low Rank Approximation , 2019, NeurIPS.

[32]  Samuel B. Hopkins,et al.  Quantum Entropy Scoring for Fast Robust Mean Estimation and Improved Outlier Detection , 2019, NeurIPS.

[33]  Yin Tat Lee,et al.  Solving Empirical Risk Minimization in the Current Matrix Multiplication Time , 2019, COLT.

[34]  Qin Zhang,et al.  Distributed and Streaming Linear Programming in Low Dimensions , 2019, PODS.

[35]  Nikhil Bansal,et al.  On a generalization of iterated and randomized rounding , 2018, STOC.

[36]  Aditi Raghunathan,et al.  Semidefinite relaxations for certifying robustness to adversarial examples , 2018, NeurIPS.

[37]  Man-Chung Yue,et al.  Universal Barrier Is n-Self-Concordant , 2018, Math. Oper. Res..

[38]  David P. Woodruff,et al.  Sketching for Kronecker Product Regression and P-splines , 2017, AISTATS.

[39]  David P. Woodruff,et al.  Relative Error Tensor Low Rank Approximation , 2017, Electron. Colloquium Comput. Complex..

[40]  Yin Tat Lee,et al.  An SDP-based algorithm for linear-sized spectral sparsification , 2017, STOC.

[41]  Shachar Lovett,et al.  Towards a Constructive Version of Banaszczyk's Vector Balancing Theorem , 2016, APPROX-RANDOM.

[42]  Nikhil Bansal,et al.  Algorithmic discrepancy beyond partial coloring , 2016, STOC.

[43]  Nikhil Bansal,et al.  An Algorithm for Komlós Conjecture Matching Banaszczyk's Bound , 2016, 2016 IEEE 57th Annual Symposium on Foundations of Computer Science (FOCS).

[44]  Daniel M. Kane,et al.  Robust Estimators in High Dimensions without the Computational Intractability , 2016, 2016 IEEE 57th Annual Symposium on Foundations of Computer Science (FOCS).

[45]  Yin Tat Lee,et al.  A Faster Cutting Plane Method and its Implications for Combinatorial and Convex Optimization , 2015, 2015 IEEE 56th Annual Symposium on Foundations of Computer Science.

[46]  Yin Tat Lee,et al.  Using Optimization to Obtain a Width-Independent, Parallel, Simpler, and Faster Positive SDP Solver , 2015, SODA.

[47]  David P. Woodruff,et al.  Subspace Embeddings for the Polynomial Kernel , 2014, NIPS.

[48]  Yin Tat Lee,et al.  Path Finding Methods for Linear Programming: Solving Linear Programs in Õ(vrank) Iterations and Faster Algorithms for Maximum Flow , 2014, 2014 IEEE 55th Annual Symposium on Foundations of Computer Science.

[49]  Andrew McGregor,et al.  Graph stream algorithms: a survey , 2014, SGMD.

[50]  François Le Gall,et al.  Powers of tensors and fast matrix multiplication , 2014, ISSAC.

[51]  Rasmus Pagh,et al.  Fast and scalable polynomial kernels via explicit feature maps , 2013, KDD.

[52]  Huy L. Nguyen,et al.  OSNAP: Faster Numerical Linear Algebra Algorithms via Sparser Subspace Embeddings , 2012, 2013 IEEE 54th Annual Symposium on Foundations of Computer Science.

[53]  David P. Woodruff,et al.  Low rank approximation and regression in input sparsity time , 2012, STOC '13.

[54]  Virginia Vassilevska Williams,et al.  Multiplying matrices faster than coppersmith-winograd , 2012, STOC '12.

[55]  Sanjeev Arora,et al.  The Multiplicative Weights Update Method: a Meta-Algorithm and Applications , 2012, Theory Comput..

[56]  Rahul Jain,et al.  A parallel approximation algorithm for mixed packing and covering semidefinite programs , 2012, ArXiv.

[57]  Richard Peng,et al.  Faster and simpler width-independent parallel algorithms for positive semidefinite programming , 2012, SPAA '12.

[58]  Rahul Jain,et al.  A Parallel Approximation Algorithm for Positive Semidefinite Programming , 2011, 2011 IEEE 52nd Annual Symposium on Foundations of Computer Science.

[59]  Chih-Jen Lin,et al.  LIBSVM: A library for support vector machines , 2011, TIST.

[60]  Daniel M. Kane,et al.  Sparser Johnson-Lindenstrauss Transforms , 2010, JACM.

[61]  Benjamin Recht,et al.  A Simpler Approach to Matrix Completion , 2009, J. Mach. Learn. Res..

[62]  Emmanuel J. Candès,et al.  The Power of Convex Relaxation: Near-Optimal Matrix Completion , 2009, IEEE Transactions on Information Theory.

[63]  Emmanuel J. Candès,et al.  Exact Matrix Completion via Convex Optimization , 2008, Found. Comput. Math..

[64]  M. Wainwright,et al.  High-dimensional analysis of semidefinite relaxations for sparse principal components , 2008, 2008 IEEE International Symposium on Information Theory.

[65]  Sanjeev Arora,et al.  A combinatorial, primal-dual approach to semidefinite programs , 2007, STOC '07.

[66]  Tamás Sarlós,et al.  Improved Approximation Algorithms for Large Matrices via Random Projections , 2006, 2006 47th Annual IEEE Symposium on Foundations of Computer Science (FOCS'06).

[67]  Thorsten Joachims,et al.  Training linear SVMs in linear time , 2006, KDD '06.

[68]  Santosh S. Vempala,et al.  Solving convex programs by random walks , 2004, JACM.

[69]  Satish Rao,et al.  Expander flows, geometric embeddings and graph partitioning , 2004, STOC '04.

[70]  Michael I. Jordan,et al.  A Direct Formulation for Sparse Pca Using Semidefinite Programming , 2004, SIAM Rev..

[71]  Moses Charikar,et al.  Finding frequent items in data streams , 2002, Theor. Comput. Sci..

[72]  Kurt M. Anstreicher,et al.  The Volumetric Barrier for Semidefinite Programming , 2000, Math. Oper. Res..

[73]  Noga Alon,et al.  The space complexity of approximating the frequency moments , 1996, STOC '96.

[74]  David R. Karger,et al.  Approximate graph coloring by semidefinite programming , 1994, Proceedings 35th Annual Symposium on Foundations of Computer Science.

[75]  David P. Williamson,et al.  .879-approximation algorithms for MAX CUT and MAX 2SAT , 1994, STOC '94.

[76]  Pravin M. Vaidya,et al.  A new algorithm for minimizing convex functions over convex sets , 1989, 30th Annual Symposium on Foundations of Computer Science.

[77]  N. Z. Shor Cut-off method with space extension in convex programming problems , 1977, Cybernetics.

[78]  Danyang Zhuo,et al.  An Online and Unified Algorithm for Projection Matrix Vector Multiplication with Application to Empirical Risk Minimization , 2023, AISTATS.

[79]  Oblivious Sketching-based Central Path Method for Solving Linear Programming Problems , 2020 .

[80]  John E. Mitchell,et al.  Properties of a Cutting Plane Method for Semidefinite Programming 1 , 2007 .

[81]  Yurii Nesterov,et al.  Interior-point polynomial algorithms in convex programming , 1994, Siam studies in applied mathematics.

[82]  A. S. Nemirovsky,et al.  Conic formulation of a convex programming problem and duality , 1992 .

[83]  Michael J. Todd,et al.  Polynomial Algorithms for Linear Programming , 1988 .