Timezone: »

Stochastic Relational Models for Large-scale Dyadic Data using MCMC
Shenghuo Zhu · Kai Yu · Yihong Gong

Wed Dec 10 07:30 PM -- 12:00 AM (PST) @ None #None

Stochastic relational models provide a rich family of choices for learning and predicting dyadic data between two sets of entities. It generalizes matrix factorization to a supervised learning problem that utilizes attributes of objects in a hierarchical Bayesian framework. Previously empirical Bayesian inference was applied, which is however not scalable when the size of either object sets becomes tens of thousands. In this paper, we introduce a Markov chain Monte Carlo (MCMC) algorithm to scale the model to very large-scale dyadic data. Both superior scalability and predictive accuracy are demonstrated on a collaborative filtering problem, which involves tens of thousands users and a half million items.

Author Information

Shenghuo Zhu (NEC Laboratories America)
Kai Yu (Baidu)
Yihong Gong

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors