Online learning in the embedded manifold of low-rank matrices

Uri Shalit*, Daphna Weinshall, Gal Chechik

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

44 Scopus citations

Abstract

When learning models that are represented in matrix forms, enforcing a low-rank constraint can dramatically improve the memory and run time complexity, while providing a natural regularization of the model. However, naive approaches to minimizing functions over the set of low-rank matrices are either prohibitively time consuming (repeated singular value decomposition of the matrix) or numerically unstable (optimizing a factored representation of the low-rank matrix). We build on recent advances in optimization over manifolds, and describe an iterative online learning procedure, consisting of a gradient step, followed by a second-order retraction back to the manifold. While the ideal retraction is costly to compute, and so is the projection operator that approximates it, we describe another retraction that can be computed efficiently. It has run time and memory complexity of O((n+m)κ) for a rank-κ matrix of dimension m × n, when using an online procedure with rank-one gradients. We use this algorithm, LORETA, to learn a matrix-form similarity measure over pairs of documents represented as high dimensional vectors. LORETA improves the mean average precision over a passive-aggressive approach in a factorized model, and also improves over a full model trained on pre-selected features using the same memory requirements. We further adapt LORETA to learn positive semi-definite low-rank matrices, providing an online algorithm for low-rank metric learning. LORETA also shows consistent improvement over standard weakly supervised methods in a large (1600 classes and 1 million images, using ImageNet) multi-label image classification task.

Original languageAmerican English
Pages (from-to)429-458
Number of pages30
JournalJournal of Machine Learning Research
Volume13
StatePublished - Feb 2012

Keywords

  • Low rank
  • Metric learning
  • Multitask learning
  • Online learning
  • Retractions
  • Riemannian manifolds

Fingerprint

Dive into the research topics of 'Online learning in the embedded manifold of low-rank matrices'. Together they form a unique fingerprint.

Cite this