A Probabilistic Approach to Robust Matrix Factorization (ECCV2012')

Naiyan Wang, Tiansheng Yao, Jingdong Wang, and Dit-Yan Yeung.


Matrix factorization underlies a large variety of computer vision applications. It is a particularly challenging problem for large-scale applications and when there exist outliers and missing data. In this paper, we propose a novel probabilistic model called Probabilistic Robust Matrix Factorization (PRMF) to solve this problem. In particular, PRMF is formulated with a Laplace error and a Gaussian prior which correspond to an l1 loss and an l2 regularizer, respectively. For model learning, we devise a parallelizable expectation-maximization (EM) algorithm which can potentially be applied to large-scale applications. We also propose an online extension of the algorithm for sequential data to offer further scalability. Experiments conducted on both synthetic data and some practical computer vision applications show that PRMF is comparable to other state-of-the-art robust matrix factorization methods in terms of accuracy and outperforms them particularly for large data matrices.

[pdf] [BibTex]

Demo Video

Batch Mode
Online Mode

Demo Codes

Matlab Code

Related Project

[BRMF Project Page]