Command Palette
Search for a command to run...
Collaborative Filtering with Graph Information: Consistency and Scalable Methods
{Pradeep K. Ravikumar Hsiang-Fu Yu Nikhil Rao Inderjit S. Dhillon}

Abstract
Low rank matrix completion plays a fundamental role in collaborative filtering applications, the key idea being that the variables lie in a smaller subspace than the ambient space. Often, additional information about the variables is known, and it is reasonable to assume that incorporating this information will lead to better predictions. We tackle the problem of matrix completion when pairwise relationships among variables are known, via a graph. We formulate and derive a highly efficient, conjugate gradient based alternating minimization scheme that solves optimizations with over 55 million observations up to 2 orders of magnitude faster than state-of-the-art (stochastic) gradient-descent based methods. On the theoretical front, we show that such methods generalize weighted nuclear norm formulations, and derive statistical consistency guarantees. We validate our results on both real and synthetic datasets.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-douban | GRALS | RMSE: 0.714 |
| collaborative-filtering-on-flixster | GRALS | RMSE: 0.845 |
| collaborative-filtering-on-movielens-100k | GRALS | RMSE (u1 Splits): 0.945 |
| collaborative-filtering-on-yahoomusic | GRALS | RMSE: 22.872 |
| recommendation-systems-on-douban-monti | GRALS | RMSE: 0.8326 |
| recommendation-systems-on-flixster-monti | GRALS | RMSE: 1.2447 |
| recommendation-systems-on-yahoomusic-monti | GRALS | RMSE: 38.0423 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.