Research paper · in preparation

Infinite stochastic block models

How many communities does a graph have? Instead of fixing the number in advance, let the data decide.

With Abdulkadir Çelikkanat · Department of Computer Science, Aalborg University

Draft. The question and model are written from your ISBM notes; the contribution and status need your words.
Hidden structure, recovered. Left: the adjacency matrix of a synthetic 120-node graph in random order. Right: the same matrix sorted by the clusters a collapsed Gibbs sampler inferred. All 6 planted clusters are recovered (ARI = 1.0).

The question

A stochastic block model explains a graph by grouping its nodes: two nodes connect with a probability that depends only on their groups. Classic SBMs need the number of groups $K$ up front. The infinite relational model puts a Chinese restaurant process prior on the grouping, so $K$ is inferred along with everything else.

Approach

The model is $z \sim \mathrm{CRP}(\alpha)$, $\theta_{kl} \sim \mathrm{Beta}(a,b)$ and $x_{ij} \sim \mathrm{Bernoulli}(\theta_{z_i z_j})$. Integrating out $\theta$ gives a collapsed sampler that only moves the labels $z$. [What the paper adds: the method, comparison or setting that is new.]

Status

[Target venue and timeline, or 'in preparation'.]