Variational inference for neural network matrix factorization and its application to stochastic blockmodeling

Onno Kampman, Creighton Heaukulani

We consider the probabilistic analogue to neural network matrix factorization (Dziugaite & Roy, 2015), which we construct with Bayesian neural networks and fit with variational inf…