Bayesian Poisson Tensor Factorization for Inferring Multilateral Relations from Sparse Dyadic Event Counts

Abstract

We present a Bayesian tensor factorization model for inferring latent group structures from dynamic pairwise interaction patterns. For decades, political scientists have collected and analyzed records of the form "country $i$ took action $a$ toward country $j$ at time $t$"---known as dyadic events---in order to form and test theories of international relations. We represent these event data as a tensor of counts and develop Bayesian Poisson tensor factorization to infer a low-dimensional, interpretable representation of their salient patterns. We demonstrate that our model's predictive performance is better than that of standard non-negative tensor factorization methods. We also provide a comparison of our variational updates to their maximum likelihood counterparts. In doing so, we identify a better way to form point estimates of the latent factors than that typically used in Bayesian Poisson matrix factorization. Finally, we showcase our model as an exploratory analysis tool for political scientists. We show that the inferred latent factor matrices capture interpretable multilateral relations that both conform to and inform our knowledge of international affairs.

To appear in Proceedings of the 21st ACM SIGKDD Conference of Knowledge Discovery and Data Mining (KDD 2015)

References in corpus (5)

• On Tensors, Sparsity, and Nonnegative Factorizations
• Scalable Recommendation with Poisson Factorization
• Nonnegative Factorization and The Maximum Edge Biclique Problem
• A Bayesian Tensor Factorization Model via Variational Inference for Link Prediction
• Equivariant and scale-free Tucker decomposition models

Cited by in corpus (6)

• Variational Auto-encoder Based Bayesian Poisson Tensor Factorization for Sparse and Imbalanced Count Data
• Fragmentation Coagulation Based Mixed Membership Stochastic Blockmodel
• Non-negative Factorization of the Occurrence Tensor from Financial Contracts
• PASTA: A Parallel Sparse Tensor Algorithm Benchmark Suite
• Explaining reviews and ratings with PACO: Poisson Additive Co-Clustering
• Understanding International Migration using Tensor Factorization