New Algorithms for Learning Incoherent and Overcomplete Dictionaries

By Sanjeev Arora, Rong Ge, Ankur Moitra. 2013

212 citations recorded by the source corpus.

In sparse recovery we are given a matrix $A$ (the dictionary) and a vector of the form $A X$ where $X$ is sparse, and the goal is to recover $X$. This is a central notion in signal processing, statistics and machine learning. But in applications such as sparse coding, edge detection, compression and super resolution, the dictionary $A$ is unknown and has to be learned from random examples of the form $Y = AX$ where $X$ is drawn from an appropriate distribution --- this is the dictionary learning problem. In most settings, $A$ is overcomplete: it has more columns than rows. This paper presents a polynomial-time algorithm for learning overcomplete dictionaries; the only previously known algorithm with provable guarantees is the recent work of Spielman, Wang and Wright who gave an algorithm for the full-rank case, which is rarely the case in applications. Our algorithm applies to incoherent dictionaries which have been a central object of study since they were introduced in seminal work of Donoho and Huo. In particular, a dictionary is $\mu$-incoherent if each pair of columns has inner product at most $\mu / \sqrt{n}$. The algorithm makes natural stochastic assumptions about the unknown sparse vector $X$, which can contain $k \leq c \min(\sqrt{n}/\mu \log n, m^{1/2 -\eta})$ non-zero entries (for any $\eta > 0$). This is close to the best $k$ allowable by the best sparse recovery algorithms even if one knows the dictionary $A$ exactly. Moreover, both the running time and sample complexity depend on $\log 1/\epsilon$, where $\epsilon$ is the target accuracy, and so our algorithms converge very quickly to the true dictionary. Our algorithm can also tolerate substantial amounts of noise provided it is incoherent with respect to the dictionary (e.g., Gaussian). In the noisy setting, our running time and sample complexity depend polynomially on $1/\epsilon$, and this is necessary.

Subfields: Biomedical Engineering, Computational Mechanics, Statistical Learning Theory, Algorithms & Complexity, Signal Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Biomedical Engineering

  1. Yuanzhi Li
  2. Francis R. Bach
  3. Karin Schnass
  4. Zehao Dou
  5. Zeyuan Allen-Zhu
  6. Jean Ponce
  7. Tomoki Tsuchida
  8. Elad Hazan

In Computational Mechanics

  1. Zeyuan Allen-Zhu
  2. Yuanzhi Li
  3. Francis R. Bach
  4. Prateek Jain
  5. Peter Richtárik
  6. Tengyu Ma
  7. Julien Mairal
  8. Praneeth Netrapalli

In Statistical Learning Theory

  1. Yuanzhi Li
  2. Zeyuan Allen-Zhu
  3. Zehao Dou
  4. Wei Hu
  5. Tengyu Ma
  6. Jason D. Lee
  7. Sanjeev Arora
  8. Karin Schnass

In Algorithms & Complexity

  1. Yuanzhi Li
  2. Elad Hazan
  3. Karin Schnass
  4. Zehao Dou
  5. Zeyuan Allen-Zhu
  6. Francis R. Bach
  7. Tomoki Tsuchida
  8. Garrison W. Cottrell

In Signal Processing

  1. Zeyuan Allen-Zhu
  2. Yuanzhi Li
  3. Francis R. Bach
  4. Julien Mairal
  5. Zehao Dou
  6. Anima Anandkumar
  7. Jean Ponce
  8. Rong Ge

Citations in the corpus