Abstract
We propose a novel limited-memory stochastic block BFGS update for incorporating enriched curvature information in stochastic approximation methods. In our method, the estimate of the inverse Hessian matrix that is maintained by it, is updated at each iteration using a sketch of the Hessian, i.e., a randomly generated compressed form of the Hessian. We propose several sketching strategies, present a new quasi-Newton method that uses stochastic block BFGS updates combined with the variance reduction approach SVRG to compute batch stochastic gradients, and prove linear convergence of the resulting method. Numerical tests on large-scale logistic regression problems reveal that our method is more robust and substantially outperforms current state-of-the-art methods.
Original language | English (US) |
---|---|
Title of host publication | 33rd International Conference on Machine Learning, ICML 2016 |
Editors | Kilian Q. Weinberger, Maria Florina Balcan |
Publisher | International Machine Learning Society (IMLS) |
Pages | 2774-2783 |
Number of pages | 10 |
ISBN (Electronic) | 9781510829008 |
State | Published - 2016 |
Externally published | Yes |
Event | 33rd International Conference on Machine Learning, ICML 2016 - New York City, United States Duration: Jun 19 2016 → Jun 24 2016 |
Publication series
Name | 33rd International Conference on Machine Learning, ICML 2016 |
---|---|
Volume | 4 |
Other
Other | 33rd International Conference on Machine Learning, ICML 2016 |
---|---|
Country/Territory | United States |
City | New York City |
Period | 06/19/16 → 06/24/16 |
Bibliographical note
Publisher Copyright:© 2016 by the author(s).
ASJC Scopus subject areas
- Artificial Intelligence
- Software
- Computer Networks and Communications