Sketching meets random projection in the dual: A provable recovery algorithm for big and high-dimensional data

Jialei Wang, Jason D. Lee, Mehrdad Mahdavi, Mladen Kolar, Nathan Srebro

Research output: Contribution to journalArticle

9 Scopus citations

Abstract

Sketching techniques scale up machine learning algorithms by reducing the sample size or dimensionality of massive data sets, without sacrificing their statistical properties. In this paper, we study sketching from an optimization point of view. We first show that the iterative Hessian sketch is an optimization process with preconditioning and develop an accelerated version using this insight together with conjugate gradient descent. Next, we establish a primal-dual connection between the Hessian sketch and dual random projection, which allows us to develop an accelerated iterative dual random projection method by applying the preconditioned conjugate gradient descent on the dual problem. Finally, we tackle the problems of large sample size and high-dimensionality in massive data sets by developing the primal-dual sketch. The primal-dual sketch iteratively sketches the primal and dual formulations and requires only a logarithmic number of calls to solvers of small sub-problems to recover the optimum of the original problem up to arbitrary precision. Our iterative sketching techniques can also be applied for solving distributed optimization problems where data are partitioned by samples or features. Experiments on synthetic and real data sets complement our theoretical results.

Original languageEnglish (US)
Pages (from-to)4896-4944
Number of pages49
JournalElectronic Journal of Statistics
Volume11
Issue number2
DOIs
StatePublished - Jan 1 2017
Externally publishedYes

All Science Journal Classification (ASJC) codes

  • Statistics and Probability
  • Statistics, Probability and Uncertainty

Keywords

  • Acceleration
  • Conjugate gradient
  • Dual random projection
  • Iterative Hessian sketch
  • Preconditioning
  • Primal-dual conversion
  • Primal-dual sketch

Fingerprint Dive into the research topics of 'Sketching meets random projection in the dual: A provable recovery algorithm for big and high-dimensional data'. Together they form a unique fingerprint.

  • Cite this