Abstract
The infinite relational model (IRM) is a Bayesian nonparametric stochastic block model; a generative model for random networks parameterized for uni-partite undirected networks by a partition of the node set and symmetric matrix of inter-partion link probabilities. The prior for the node clusters is the Chinese restaurant process, and the link probabilities are, in the most simple setting, modeled as iid. with a common symmetric Beta prior. More advanced priors such as separate asymmetric Beta priors for links within and between clusters have also been proposed. In this paper we investigate the importance of these priors for discovering latent clusters and for predicting links. We compare fixed symmetric priors and fixed asymmetric priors based on the empirical distribution of links with a Bayesian hierarchical approach where the parameters of the priors are inferred from data. On synthetic data, we show that the hierarchical Bayesian approach can infer the prior distributions used to generate the data. On real network data we demonstrate that using asymmetric priors significantly improves predictive performance and heavily influences the number of extracted partitions.
Original language | English |
---|---|
Title of host publication | Proceedings of the IEEE International Workshop on Machine Learning for Signal Processing (MLSP 2016) |
Number of pages | 6 |
Publisher | IEEE |
Publication date | 2016 |
ISBN (Print) | 978-1-5090-0746-2 |
DOIs | |
Publication status | Published - 2016 |
Event | 2016 IEEE 26th International Workshop on Machine Learning for Signal Processing - Vietri sul Mare, Italy Duration: 13 Sept 2016 → 16 Sept 2016 Conference number: 26 https://ieeexplore.ieee.org/xpl/conhome/7605057/proceeding |
Conference
Conference | 2016 IEEE 26th International Workshop on Machine Learning for Signal Processing |
---|---|
Number | 26 |
Country/Territory | Italy |
City | Vietri sul Mare |
Period | 13/09/2016 → 16/09/2016 |
Internet address |
Keywords
- Infinite relational model
- Hyperparameter inference
- Link-prediction
- Bayesian nonparametrics