AAAI Publications, Thirtieth AAAI Conference on Artificial Intelligence

Font Size: 
Discriminative Nonparametric Latent Feature Relational Models with Data Augmentation
Bei Chen, Ning Chen, Jun Zhu, Jiaming Song, Bo Zhang

Last modified: 2016-02-21

Abstract


We present a discriminative nonparametric latent feature relational model (LFRM) for link prediction to automatically infer the dimensionality of latent features. Under the generic RegBayes (regularized Bayesian inference) framework, we handily incorporate the prediction loss with probabilistic inference of a Bayesian model; set distinct regularization parameters for different types of links to handle the imbalance issue in real networks; and unify the analysis of both the smooth logistic log-loss and the piecewise linear hinge loss. For the nonconjugate posterior inference, we present a simple Gibbs sampler via data augmentation, without making restricting assumptions as done in variational methods. We further develop an approximate sampler using stochastic gradient Langevin dynamics to handle large networks with hundreds of thousands of entities and millions of links, orders of magnitude larger than what existing LFRM models can process. Extensive studies on various real networks show promising performance.

Keywords


Link Prediction; Bayesian Nonparametrics; Latent Feature Model; Data Augmentation

Full Text: PDF