[edit]
Zero-Truncated Poisson Model for Scalable Bayesian Factorization of Massive Binary Tensors with Mode-Networks
Proceedings of the 31st Conference on Uncertainty in Artificial Intelligence, PMLR R13:950-959, 2015.
Abstract
We present a scalable Bayesian model for low-rank factorization of massive tensors with binary observations. The proposed model has the following key properties: (1) in contrast to the models based on logistic or probit likelihood, using a zero-truncated Poisson likelihood for binary data allows our model to scale up in the number of ones in the tensor, without sacrificing on the quality of the results; (2) side-information in form of binary pairwise relationships (e.g., an adjacency network) between objects in any tensor mode can also be leveraged, which can be especially useful in “cold-start” settings; and (3) the model admits simple inference via batch, as well as online MCMC; the latter allows us to scale up even for dense binary data (i.e., when the number of ones in the tensor/network is also massive). In addition, non-negative factor matrices in our model provide easy interpretability, and the tensor rank is inferred from data. We apply our model on several real-world massive binary tensors, and on massive binary tensors with binary mode-network(s) as side-information.