Timezone: »

 
Poster
Recovery Guarantee of Non-negative Matrix Factorization via Alternating Updates
Yuanzhi Li · Yingyu Liang · Andrej Risteski

Wed Dec 07 09:00 AM -- 12:30 PM (PST) @ Area 5+6+7+8 #79
in Wednesday Posters »

Non-negative matrix factorization is a popular tool for decomposing data into feature and weight matrices under non-negativity constraints. It enjoys practical success but is poorly understood theoretically. This paper proposes an algorithm that alternates between decoding the weights and updating the features, and shows that assuming a generative model of the data, it provably recovers the ground-truth under fairly mild conditions. In particular, its only essential requirement on features is linear independence. Furthermore, the algorithm uses ReLU to exploit the non-negativity for decoding the weights, and thus can tolerate adversarial noise that can potentially be as large as the signal, and can tolerate unbiased noise much larger than the signal. The analysis relies on a carefully designed coupling between two potential functions, which we believe is of independent interest.

Keywords: Learning Theory · Matrix Factorization

Author Information

Yuanzhi Li (Princeton University)
Yingyu Liang (Princeton University)
Andrej Risteski (Princeton University)

More from the Same Authors