Stochastic blockmodels with growing number of classes

David S. Choi, Patrick J. Wolfe, Edoardo M. Airoldi

Introduction

The global structure of social, biological, and information networks is sometimes envisioned as the aggregate of many local interactions whose effects propagate in ways that are not yet well understood. There is increasing opportunity to collect data on an appropriate scale for such systems, but their analysis remains challenging (Goldenberg et al., 2009). Here we analyze a statistical model for network data known as the (single-membership) stochastic blockmodel. Its salient feature is that it partitions the NN nodes of a network into KK distinct classes whose members all interact similarly with the network. Blockmodels were first associated with the deterministic concept of structural equivalence in social network analysis (Lorrain & White, 1971), where two nodes were considered interchangeable if their connections were equivalent in a formal sense. This concept was adapted to stochastic settings and gave rise to the stochastic blockmodel in work by Holland et al. (1983) and Fienberg et al. (1985). The model and extensions thereof have since been applied in a variety of disciplines (Wang & Wong, 1987; Nowicki & Snijders, 2001; Girvan & Newman, 2002; Airoldi et al., 2005; Doreian et al., 2005; Newman, 2006; Handcock et al., 2007; Hoff, 2008; Airoldi et al., 2008; Copic et al., 2009; Mariadassou et al., 2010; Karrer & Newman, 2011).

In this work we provide a finite-sample confidence bound that can be used when estimating network structure from data modeled by independent Bernoulli random variates, and also show that under maximum likelihood fitting of a correctly specified KK-class blockmodel, the fraction of misclassified network nodes converges in probability to zero even when the number of classes KK grows with NN. As noted by Rohe et al. (2011), this is advantageous if we expect class sizes to remain relatively constant even as NN increases. Related results for fixed KK have been shown by Snijders & Nowicki (1997) for networks with linearly increasing degree, and in a stronger sense for sparse graphs with poly-logarithmically increasing degree by Bickel & Chen (2009).

Our results can be related to those of Rohe et al. (2011), who use spectral methods to bound the number of misclassified nodes in the stochastic blockmodel with increasing KK, although with the more restrictive requirement of nearly linearly increasing degree. As noted by those authors, this assumption may not hold in many practical settings. Our manner of proof requires only poly-logarithmically increasing degree, and is more closely related to the fixed-KK proof of Bickel & Chen (2009), although we note that spectral clustering as suggested by Rohe et al. (2011) provides a computationally appealing alternative to maximum likelihood fitting in practice.

As discussed by Bickel & Chen (2009), one may assume exchangeability in lieu of a generative KK-class blockmodel: An analogue to de Finetti’s theorem for exchangeable sequences states that the probability distribution of an infinite exchangeable random graph is expressible as a mixture of distributions whose components can be approximated by blockmodels (Kallenberg, 2005; Bickel & Chen, 2009). An observed network can then be viewed as a sample drawn from this infinite conceptual population, and so in this case the fitted blockmodel describes one mixture component thereof.

Statement of results

We consider likelihood-based inference for independent Bernoulli data {Aij} (i=1,…,N;j=i+1,…,N)\{A_{ij}\}\ (i=1,\ldots,N;j=i+1,\ldots,N), both when no structure linking the success probabilities {Pij}\{P_{ij}\} is assumed, as well as the special case when a stochastic blockmodel of known order KK is assumed to apply. To this end, let A∈{0,1}N×NA\in\{0,1\}^{N\times N} denote the symmetric adjacency matrix of a simple, undirected graph on NN nodes whose entries {Aij}\{A_{ij}\} for i<ji<j are assumed independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) random variates, and whose main diagonal {Aii}i=1N\{A_{ii}\}_{i=1}^{N} is fixed to zero. The average degree of this graph is 2M/N2M/N, where M=∑i<jPijM=\sum_{i<j}P_{ij} is its expected number of edges. Under a KK-class stochastic blockmodel, these edge probabilities are further restricted to satisfy

for some symmetric matrix θ∈K×K\theta\in^{K\times K} and membership vector z∈{1,…,K}Nz\in\{1,\ldots,K\}^{N}. Thus the probability of an edge between two nodes is assumed to depend only on the class of each node.

Let L(A;z,θ)L(A;z,\theta) denote the log-likelihood of observing data matrix AA under a KK-class blockmodel with parameters (z,θ)(z,\theta), and LˉP(z,θ)\bar{L}_{P}(z,\theta) its expectation:

For fixed class assignment zz, let NaN_{a} denote the number of nodes assigned to class aa, and let nabn_{ab} denote the maximum number of possible edges between classes aa and bb; i.e., nab=NaNbn_{ab}=N_{a}N_{b} if a≠ba\neq b and naa=(Na2)n_{aa}={N_{a}\choose 2}. Further, let θ^(z)\hat{\theta}^{(z)} and θˉ(z)\bar{\theta}^{(z)} be symmetric matrices in K×K^{K\times K}, with

defined whenever nab≠0n_{ab}\neq 0. Observe that θ^(z)\hat{\theta}^{(z)} comprises sample proportion estimators as a function of zz, whereas θˉ(z)\bar{\theta}^{(z)} is its expectation under the independent {Bernoulli⁡(Pij)}\{\operatorname{Bernoulli}(P_{ij})\} model. Taken over all class assignments z∈{1,…,K}Nz\in\{1,\ldots,K\}^{N}, the sets {θ^(z)}\{\hat{\theta}^{(z)}\} comprise a sufficient statistic for the family of KK-class stochastic blockmodels, and for each zz, θ^(z)\hat{\theta}^{(z)} maximizes L(A;z,⋅)L(A;z,\cdot). Analogously, the sets {θˉ(z)}\{\bar{\theta}^{(z)}\} are functions of the model parameters {Pij}i<j\{P_{ij}\}_{i<j}, and maximize LˉP(z,⋅)\bar{L}_{P}(z,\cdot). We write θ^\hat{\theta} and θˉ\bar{\theta} when the choice of zz is understood, and L(A;z)L(A;z) and LˉP(z)\bar{L}_{P}(z) to abbreviate sup⁡θL(A;z,θ)\sup_{\theta}L(A;z,\theta) and sup⁡θLˉP(z,θ)\sup_{\theta}\bar{L}_{P}(z,\theta) respectively.

Finally, observe that when a blockmodel with parameters (zˉ,θˉ)(\bar{z},\bar{\theta}) is in force, then Pij=θˉzˉizˉjP_{ij}=\bar{\theta}_{\bar{z}_{i}\bar{z}_{j}} in accordance with (1), and consequently LˉP\bar{L}_{P} is maximized by the true parameter values (zˉ,θˉ)(\bar{z},\bar{\theta}):

where D(p∣∣p′)D(p\mid\mid p^{\prime}) denotes the Kullback–Leibler divergence of a Bernoulli⁡(p′)\operatorname{Bernoulli}(p^{\prime}) distribution from a Bernoulli⁡(p)\operatorname{Bernoulli}(p) one.

2 Fitting a K𝐾K-class stochastic blockmodel to independent Bernoulli trials

Fitting a KK-class stochastic blockmodel to independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) trials yields estimates θ^(z)\hat{\theta}^{(z)} of averages θˉ(z)\bar{\theta}^{(z)} of subsets of the parameter set {Pij}\{P_{ij}\}, with each class assignment zz inducing a partition of that set. We begin with a basic lemma that expresses the difference L(A;z)−LˉP(z)L(A;z)-\bar{L}_{P}(z) in terms of θ^(z)\hat{\theta}^{(z)} and θˉ(z)\bar{\theta}^{(z)}, and follows directly from their respective maximizing properties.

Let {Aij}i<j\{A_{ij}\}_{i<j} comprise independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) trials. Then the difference sup⁡θL(A;z,θ)−sup⁡θLˉP(z,θ)\sup_{\theta}L(A;z,\theta)-\sup_{\theta}\bar{L}_{P}(z,\theta) can be expressed for X=∑i<jAijlog⁡{θˉzizj/(1−θˉzizj)}X=\sum_{i<j}A_{ij}\log\{\bar{\theta}_{z_{i}z_{j}}/(1-\bar{\theta}_{z_{i}z_{j}})\} as

We first bound the former quantity in this expression, which provides a measure of the distance between θ^\hat{\theta} and its estimand θˉ\bar{\theta} under the setting of Lemma 2.1. The bound is used in subsequent asymptotic results, and also yields a kind of confidence measure on θ^\hat{\theta} in the finite-sample regime.

Suppose that a KK-class stochastic blockmodel is fitted to data {Aij}i<j\{A_{ij}\}_{i<j} comprising (N2)\binom{N}{2} independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) trials, where, for any class assignment zz, estimate θ^\hat{\theta} maximizes the blockmodel log-likelihood L(A;z,⋅)L(A;z,\cdot). Then with probability at least 1−δ1-\delta,

Theorem 2.2 is proved in the Appendix via the method of types: for fixed zz, the probability of any realization of θ^\hat{\theta} is first bounded by exp⁡{−∑a≤bnabD(θ^ab∣∣θˉab)}\exp\{-\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab})\}. A counting argument then yields a deviation result in terms of (N/K+1)K2+K(N/K+1)^{K^{2}+K}, and finally a union bound is applied so that the result holds uniformly over all KNK^{N} possible choices of assignment vector zz.

Our second result is asymptotic, and combines Theorem 2.2 with a Bernstein inequality for bounded random variables, applied to the latter terms X−E(X)X-E(X) in Lemma 2.1. To ensure boundedness we assume minimal restrictions on each PijP_{ij}; this Bernstein inequality, coupled with a union bound to ensure that the result holds uniformly over all zz, dictates growth restrictions on KK and MM.

Assume the setting of Theorem 2.2, whereby a KK-class blockmodel is fitted to (N2)\binom{N}{2} independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) random variates {Aij}i<j\{A_{ij}\}_{i<j}, and further assume that 1/N2≤Pij≤1−1/N21/N^{2}\leq P_{ij}\leq 1-1/N^{2} for all NN and i<ji<j. Then if K=O(N1/2)K=\mathcal{O}(N^{1/2}) and M=ω(N(log⁡N)3+δ)M=\omega(N(\log N)^{3+\delta}) for some δ>0\delta>0,

Thus whenever each PijP_{ij} is bounded away from 0 and 1 in the manner above, the maximized log-likelihood function L(A;z)=sup⁡θL(A;z,θ)L(A;z)=\sup_{\theta}L(A;z,\theta) is asymptotically well behaved in network size NN as long as the network’s average degree 2M/N2M/N grows faster than (log⁡N)3+δ(\log N)^{3+\delta} and the number KK of classes fitted to it grows no faster than N1/2N^{1/2}.

3 Fitting a correctly specified K𝐾K-class stochastic blockmodel

The above results apply to the general case of independent Bernoulli data {Aij}\{A_{ij}\}, with no additional structure assumed amongst the set of success probabilities {Pij}\{P_{ij}\}; if we further assume the data to be generated by a KK-class stochastic blockmodel whose parameters (zˉ,θˉ)(\bar{z},\bar{\theta}) are subject to suitable identifiability conditions, it is possible to characterize the behavior of the class assignment estimator z^\hat{z} under maximum likelihood fitting of a correctly specified KK-class blockmodel.

If the conclusion max⁡z∣L(A;z)−LˉP(z)∣=oP(M)\max_{z}|L(A;z)-\bar{L}_{P}(z)|=o_{P}(M) of Theorem 2.3 holds, and data are generated according to a KK-class blockmodel with membership vector zˉ\bar{z}, then

with respect to the maximum-likelihood KK-class blockmodel class assignment estimator z^\hat{z}.

(i) for all blockmodel classes a=1,…,Ka=1,\ldots,K, class size NaN_{a} grows as min⁡a{Na}=Ω(N/K)\min_{a}\{N_{a}\}=\Omega(N/K);

(ii) the following holds over all distinct class pairs (a,b)(a,b) and all classes cc:

Numerical results

We now present results of a small simulation study undertaken to investigate the assumptions and conditions of Theorems 2.2–2.4 above, in which KK-class blockmodels were fitted to various networks generated at random from models corresponding to each of the three theorems. Because exact maximization in zz of the blockmodel log-likelihood L(A;z,θ)L(A;z,\theta) is computationally intractable even for moderate NN, we instead employed Gibbs sampling to explore the function max⁡θL(A;z,θ)\max_{\theta}L(A;z,\theta) and recorded the best value of zz visited by the sampler. As the results of Theorems 2.2 and 2.3 hold uniformly in zz, however, we expect θˉ\bar{\theta} and LˉP(z)\bar{L}_{P}(z) to be close to their empirical estimates whenever NN is sufficiently large, regardless of the approach employed to select zz. This fact also suggests that a single-class (Erdös-Rényi) blockmodel may come closest to achieving equality in Theorems 2.2 and 2.3, as many class assignments are equally likely a priori to have high likelihood. By similar reasoning, a weakly identifiable model should come closest to achieving the error bound in Theorem 2.4, such as one with nearly identical within- and between-class edge probabilities. We describe each of these cases empirically in the remainder of this section.

First, the tightness of the confidence bound of (2) from Theorem 2.2 was investigated by fitting KK-class blockmodels to Erdös-Rényi networks comprising (N2)\binom{N}{2} independent Bernoulli⁡(p)\operatorname{Bernoulli}(p) trials, with N=500N=500 nodes and p=\0⋅\cdot075 chosen to match the data analysis example in the sequel, and K∈{5,10,20,30,40,50}K\in\{5,10,20,30,40,50\}. For each KK, the error terms ∑a≤bnabD(θ^ab∣∣θˉab)\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab}) and {∑a≤bnab(θ^ab−θˉab)2}1/2\{\sum_{a\leq b}n_{ab}(\hat{\theta}_{ab}-\bar{\theta}_{ab})^{2}\}^{1/2} were recorded for each of 100 trials and compared to the respective 95% confidence bounds (\delta=\0⋅\cdot05) derived from Theorem 2.2. The bounds overestimated the respective errors by a factor of 3 to 7 on average, with small standard deviation. In this worst-case scenario the bound is loose, but not unusable; the errors never exceeded the 95% confidence bounds in any of the trials.

To test whether the assumptions of Theorem 2.3 are necessary as well as sufficient to obtain convergence of L(A;z)/ML(A;z)/M to LˉP(z)/M\bar{L}_{P}(z)/M, blockmodels were next fitted to Erdös-Rényi networks of increasing size, for NN in the range 50–1050. The corresponding normalized log-likelihood error ∣L(A;z)−LˉP(z)∣/M|L(A;z)-\bar{L}_{P}(z)|/M for different rates of growth in the expected number of edges MM and the number of fitted classes KK is shown in Fig. 1. Observe from the leftmost panel that when M=N(log⁡N)4M=N(\log N)^{4} and K=N1/2K=N^{1/2}, as prescribed by the theorem, this error decreases in NN. If the edge density is reduced to M/N=(log⁡N)2M/N=(\log N)^{2}, we observe in the center panel convergence when K=N1/2K=N^{1/2} and divergence when K=N3/5K=N^{3/5}. This suggests that the error as a function of KK follows Theorem 2.3 closely, but that the network can be somewhat more sparse than it requires.

Network data example

To illustrate the use of our results in the fitting of KK-class stochastic blockmodels to network data, we employed a publicly available social network dataset containing N=553N=553 undergraduate Facebook profiles from the California Institute of Technology (people.maths.ox.ac.uk/∼\simporterm/data/facebook5.zip). These profiles indicate whenever a pair of students have identified one another as friends, yielding a network of 11 51111\,511 edges and accompanying covariate information including gender, class year, and hall of residence.

Traud et al. (2011) applied community detection algorithms to this network, and compared their output to partitions based on categorical covariates such as those identified above. They concludes that a grouping of students by residence hall was most similar to the best algorithmic grouping obtained, and thus that shared residence hall membership was the best predictor for the formation of community structure. This structure is reflected in the leftmost panel of Fig. 2, which shows the network adjacency structure under an ordering of students by residence hall.

2 Logit blockmodel parameterization and fitting procedure

Here we build on the results of Traud et al. (2011) by taking covariate information explicitly into account when fitting the Facebook dataset described above. Specifically, by assuming only that links are independent Bernoulli variates and then employing confidence bounds to assess fitted blocks by way of parameter θˉ(z)\bar{\theta}^{(z)}, we examine these data for residual community structure beyond that well explained by the covariates themselves.

Since the results of Theorems 2.2 and 2.3 hold uniformly over all choices of blockmodel membership vector zz, we may select zz in any manner, including those that depend on covariates. For this example, we determined an approximate maximum likelihood estimate z^\hat{z} under a logit blockmodel that allows the direct incorporation of covariates. The model is parameterized such that the log-odds ratio of an edge occurrence between nodes ii and jj is given by

3 Data analysis

We fitted the logit blockmodel of (4) for values of KK ranging from 11 to 5050 using the stochastic maximization procedure described in the preceding paragraph, and gauged model order by the Bayesian information criterion and out-of-sample prediction using five-fold cross validation, shown respectively in the center and rightmost panels of Fig. 2. These plots suggest a relatively low model order, beginning around K=4K=4. The corresponding 95% confidence bounds on the divergence of θ^(z)\hat{\theta}^{(z)} from θˉ(z)\bar{\theta}^{(z)} provided by Theorem 2.2 also yield small values for KK in the range 4–7: for example, when K=5K=5, the normalized sum of Kullback–Leibler divergences (N2)−1∑a≤bnabD(θ^ab∣∣θˉab){N\choose 2}{}^{-1}\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab}) is bounded by 0⋅\cdot0067. Corresponding normalized root-mean-square error bounds over this range of KK are approximately one order of magnitude larger.

We then examined approximate maximum likelihood estimates of zz for KK in the range 4–7, as shown in the top two rows of Fig. 3; larger values of KK also reveal block structure, but exhibit correspondingly larger confidence bound evaluations. The permuted adjacency structures under each estimated class assignment z^\hat{z} are shown in the top row, along with the corresponding values of θ^\hat{\theta} below in the second row. The structure of θ^\hat{\theta} over this range of KK suggests that after covariates are taken into account, it is possible to identify a subset of students who divide naturally into two residual “meta-groups” that interact less frequently with one another in comparison to the remaining subjects in the dataset; the precision of the corresponding estimates θ^\hat{\theta} can be quantified by Theorem 2.2, as in the caption of Fig. 3.

As KK increases, these groups become more tightly concentrated, as extra blocks absorb students whose connections are more evenly distributed. While the exact membership of each group varied over KK, in part due to stochasticity in the fitting algorithm employed, we observed 199 students whose meta-group membership remained constant. The bottom row of Fig. 3 shows the 8 residence halls identified for these sets of students, with the ninth category indicating unreported; observe that the effect of residence hall is still visible in that the left-hand grouping has more students in halls 4–7, while the right-hand grouping has more students in halls 1, 2, and 8.

Acknowledgement

Work supported in part by the National Science Foundation, National Institute of Health, Army Research Office and the Office of Naval Research, U.S.A. Additional funding provided by the Harvard Medical School’s Milton Fund.

Appendix

To begin, observe that for any fixed class assignment zz, every θ^ab\hat{\theta}_{ab} is a sum of nabn_{ab} independent Bernoulli random variables, with corresponding mean θˉab\bar{\theta}_{ab}. A Chernoff bound (Dubhashi & Panconesi, 2009) shows

Since these bounds also hold respectively for pr⁡(θ^ab=θˉab±t)\operatorname{pr}(\hat{\theta}_{ab}=\bar{\theta}_{ab}\pm t), we may bound the probability of any given realization ϑ∈{0,1/nab,…,1}\vartheta\in\{0,1/n_{ab},\ldots,1\} of θ^ab\hat{\theta}_{ab} in terms of the Kullback–Leibler divergence of θˉab\bar{\theta}_{ab} from ϑ\vartheta:

By independence of the {Aij}i<j\{A_{ij}\}_{i<j}, this implies a corresponding bound on the probability of any θ^\hat{\theta}:

Now, let Θ^\widehat{\Theta} denote the range of θ^\hat{\theta} for fixed zz, and observe that since each of the (K+12)\binom{K+1}{2} lower-diagonal entries {θ^ab}a≤b\{\hat{\theta}_{ab}\}_{a\leq b} of θ^\hat{\theta} can independently take on nab+1n_{ab}+1 distinct values, we have that ∣Θ^∣=∏a≤b(nab+1)|\widehat{\Theta}|=\prod_{a\leq b}(n_{ab}+1). Subject to the constraint that ∑a≤bnab=(N2)\sum_{a\leq b}n_{ab}=\binom{N}{2}, we see that this quantity is maximized when nab=(N2)/(K+12)n_{ab}=\binom{N}{2}/\binom{K+1}{2} for all a≤ba\leq b, and hence

Now consider the event that ∑a≤bnabD(θ^ab∣∣θˉab)\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab}) is at least as large as some ϵ>0\epsilon>0; the probability of this event is given by pr⁡(Θ^ϵ)\operatorname{pr}(\widehat{\Theta}_{\epsilon}) for

Since ∑a≤bnabD(θ^ab∣∣θˉab)≥ϵ\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab})\geq\epsilon for all θ^∈Θ^ϵ\hat{\theta}\in\widehat{\Theta}_{\epsilon}, we have from (5) and (7) that

and since ∣Θ^ϵ∣≤∣Θ^∣|\widehat{\Theta}_{\epsilon}|\leq|\widehat{\Theta}|, we may use (6) to obtain, for fixed class assignment zz,

Appealing to a union bound over all KNK^{N} possible class assignments and setting ϵ=log⁡[KN(N/K+1)K2+K/δ]\epsilon=\log[K^{N}\left(N/K+1\right)^{K^{2}+K}/\delta] then yields the claimed result.

By Lemma 2.1, the difference L(A;z)−LˉP(z)L(A;z)-\bar{L}_{P}(z) can be expressed for any fixed class assignment zz as ∑a≤bnabD(θ^ab∣∣θˉab)+X−E(X)\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab})+X-E(X), where the first term satisfies the deviation bound of (8), and X=∑i<jAijlog⁡{θˉzizj/(1−θˉzizj)}X=\sum_{i<j}A_{ij}\log\{\bar{\theta}_{z_{i}z_{j}}/(1-\bar{\theta}_{z_{i}z_{j}})\} comprises a weighted sum of independent Bernoulli⁡(Pij)\operatorname{Bernoulli}(P_{ij}) random variables.

To bound the quantity ∣X−E(X)∣|X-E(X)|, observe that since by assumption N−2≤Pij≤1−N−2N^{-2}\leq P_{ij}\leq 1-N^{-2}, the same is true for each corresponding average θˉzizj\bar{\theta}_{z_{i}z_{j}}. As a result, the random variables Xij=Aijlog⁡{θˉzizj/(1−θˉzizj)}X_{ij}=A_{ij}\log\{\bar{\theta}_{z_{i}z_{j}}/(1-\bar{\theta}_{z_{i}z_{j}})\} comprising XX are each bounded in magnitude by C=2log⁡NC=2\log N. This allows us to apply a Bernstein inequality for sums of bounded independent random variables due to Chung & Lu (2006, Theorems 2.8 and 2.9, p. 27), which states that for any ϵ>0\epsilon>0,

Finally, observe that since the event ∣L(A;z)−LˉP(z)∣>2ϵM|L(A;z)-\bar{L}_{P}(z)|>2\epsilon M implies either the event ∑a≤bnabD(θ^ab∣∣θˉab)≥ϵM\sum_{a\leq b}n_{ab}D(\hat{\theta}_{ab}\mid\mid\bar{\theta}_{ab})\geq\epsilon M or the event ∣X−E(X)∣≥ϵM|X-E(X)|\geq\epsilon M, we have for fixed assignment zz that

Summing the right-hand sides of (8) and (9), and then over all KNK^{N} possible assignments, yields

where we have used the fact that ∑i<jE(Xij2)≤4Mlog⁡2N\sum_{i<j}E(X_{ij}^{2})\leq 4M\log^{2}N in (9). It follows directly that if K=O(N1/2)K=\mathcal{O}(N^{1/2}) and M=ω(N(log⁡N)3+δ)M=\omega(N(\log N)^{3+\delta}), then lim⁡N→∞pr⁡{max⁡z∣L(A;z)−LˉP(z)∣/M≥ϵ}=0\lim_{N\rightarrow\infty}\operatorname{pr}\{\max_{z}|L(A;z)-\bar{L}_{P}(z)|/M\geq\epsilon\}=0 for every fixed ϵ>0\epsilon>0 as claimed.

Proof of Theorem 2.4

To begin, note that Theorem 2.3 holds uniformly in zz, and thus implies that

Since z^\hat{z} is the maximum-likelihood estimate of class assignment zˉ\bar{z}, we know that L(A;z^)≥L(A;zˉ)L(A;\hat{z})\geq L(A;\bar{z}), implying that L(A;z^)=L(A;zˉ)+δL(A;\hat{z})=L(A;\bar{z})+\delta for some δ≥0\delta\geq 0. Thus, by the triangle inequality,

and since LˉP(zˉ)≥LˉP(z^)\bar{L}_{P}(\bar{z})\geq\bar{L}_{P}(\hat{z}) under any blockmodel with parameter zˉ\bar{z}, we have LˉP(zˉ)−LˉP(z^)=oP(M)\bar{L}_{P}(\bar{z})-\bar{L}_{P}(\hat{z})=o_{P}(M).

Under conditions (i) and (ii) of Theorem 2.4, we will now show that also

To show (10), first observe that any blockmodel class assignment vector zz induces a corresponding partition of the set {Pij}i<j\{P_{ij}\}_{i<j} according to (i,j)↦(zi,zj)(i,j)\mapsto(z_{i},z_{j}). Formally, zz partitions {Pij}i<j\{P_{ij}\}_{i<j} into LL subsets (S1,…,SL)(S_{1},\ldots,S_{L}) via the mapping

This partition is separable in the sense that there exists a bijection between {1,…,L}\{1,\ldots,L\} and the upper triangular portion of blockmodel parameter θ\theta, such that we write θζij=θzizj\theta_{\zeta_{ij}}=\theta_{z_{i}z_{j}} for membership vector zz. More generally, for any partition Π\Pi of {Pij}i<j\{P_{ij}\}_{i<j}, we may define θˉl=∣Sl∣−1∑i<jPij 1{Pij∈Sl}\bar{\theta}_{l}=|S_{l}|^{-1}\sum_{i<j}P_{ij}\,1\{P_{ij}\in S_{l}\} as the arithmetic average over all PijP_{ij} in the subset SlS_{l} indexed by ζij=l\zeta_{ij}=l. Thus we may also define

so that LˉP∗\bar{L}_{P}^{*} and LˉP\bar{L}_{P} coincide on partitions corresponding to admissible blockmodel assignments zz.

Consider a KK-class stochastic blockmodel with membership vector zˉ\bar{z}, and let Πz\Pi^{z} denote the partition of its associated {Pij}1≤i<j≤N\{P_{ij}\}_{1\leq i<j\leq N} induced by any z∈{1,…,K}Nz\in\{1,\ldots,K\}^{N}. For every Πz\Pi^{z}, there exists a partition Π∗\Pi^{*} that refines Πz\Pi^{z} and with the property that, if conditions (i) and (ii) of Theorem 2.4 hold,

Let Π′\Pi^{\prime} be a refinement of any partition Π\Pi of the set {Pij}i<j\{P_{ij}\}_{i<j}; then LˉP∗(Π′)≥LˉP∗(Π)\bar{L}_{P}^{*}(\Pi^{\prime})\geq\bar{L}_{P}^{*}(\Pi).

Since Lemma .4 applies to any admissible blockmodel assignment vector zz, it also applies to the maximum-likelihood estimate z^\hat{z} for any realization of the data; each z^\hat{z} in turn induces a partition Π ^ \Pi\,\hat{}\, of blockmodel edge probabilities {Pij}i<j\{P_{ij}\}_{i<j}, and (11) holds with respect to its refinement Π∗\Pi^{*}. By Lemma .5, LˉP∗(Π ^ )≤LˉP∗(Π∗)\bar{L}_{P}^{*}(\Pi\,\hat{}\,)\leq\bar{L}_{P}^{*}(\Pi^{*}). Finally, observe that LˉP(z^)=LˉP∗(Π ^ )\bar{L}_{P}(\hat{z})=\bar{L}_{P}^{*}(\Pi\,\hat{}\,) by the definition of LˉP∗\bar{L}_{P}^{*}, and so LˉP(zˉ)−LˉP(z^)≥LˉP(zˉ)−LˉP∗(Π∗)\bar{L}_{P}(\bar{z})-\bar{L}_{P}(\hat{z})\geq\bar{L}_{P}(\bar{z})-\bar{L}_{P}^{*}(\Pi^{*}), thereby establishing (10).

The construction of Π∗\Pi^{*} will take several steps. For a given membership class under zz, partition the corresponding set of nodes into subclasses according to the true class assignment zˉ\bar{z} of each node. Then remove one node from each of the two largest subclasses so obtained, and group them together as a pair; continue this pairing process until no more than one nonempty subclass remains, then terminate. Observe that if we denote pairs by their node indices as (i,j)(i,j), then by construction zi=zjz_{i}=z_{j} but zˉi≠zˉj\bar{z}_{i}\neq\bar{z}_{j}.

Repeat the above procedure for each class under zz, and let C1C_{1} denote the total number of pairs thus formed. For each of the C1C_{1} pairs (i,j)(i,j), find all other distinct indices kk for which the following holds:

where CC is the constant from condition (ii) of Theorem 2.4, and indices ikik and jkjk in (12) are to be interpreted respectively as kiki whenever k<ik<i, and kjkj whenever k<jk<j. Let C2C_{2} denote the total number of distinct triples that can be formed in this manner.

We are now ready to construct the partition Π∗\Pi^{*} of the probabilities {Pij}1≤i<j≤N\{P_{ij}\}_{1\leq i<j\leq N} as follows: For each of the C2C_{2} triples (i,j,k)(i,j,k), remove PikP_{ik} (or PkiP_{ki} if k<ik<i) and PjkP_{jk} (or PkjP_{kj}) from their previous subset assignment under Πz\Pi^{z}, and place them both in a new, distinct two-element subset. We observe the following:

(i) The partition Π∗\Pi^{*} is a refinement of the partition Πz\Pi^{z} induced by zz: Since nodes ii and jj have the same class label under zz in that zi=zjz_{i}=z_{j}, it follows that for any kk, PikP_{ik} and PjkP_{jk} are in the same subset under Πz\Pi^{z}.

(iii) Condition (ii) of Theorem 2.4 implies that for every pair of classes (a,b)(a,b), there exists at least one class cc for which (12) holds eventually. Thus eventually, for any of the C1C_{1} pairs (i,j)(i,j), we obtain a number of triples at least as large as the cardinality of class cc. Condition (i) in turn implies that the cardinality of the smallest class grows as Ω(N/K)\Omega(N/K), and thus we may write C2=C1 Ω(N/K)C_{2}=C_{1}\,\Omega(N/K).

We can now express the difference LˉP(zˉ)−LˉP∗(Π∗)\bar{L}_{P}(\bar{z})-\bar{L}_{P}^{*}(\Pi^{*}) as a sum of nonnegative divergences D(Pij∣∣θˉζij∗)D(P_{ij}\mid\mid\bar{\theta}_{\zeta_{ij}^{*}}), where ζij∗\zeta_{ij}^{*} is the assignment mapping associated to Π∗\Pi^{*}, and use (12) to lower-bound this difference:

Let Π′\Pi^{\prime} be a refinement of any partition Π\Pi of the set {Pij}i<j\{P_{ij}\}_{i<j}, and given a∈{1,…,L′}a\in\{1,\ldots,L^{\prime}\} indexing Sa′S_{a}^{\prime}, let F(a)F(a) denote its index under Π\Pi. We show that LˉP∗(Π′)≥LˉP∗(Π)\bar{L}_{P}^{*}(\Pi^{\prime})\geq\bar{L}_{P}^{*}(\Pi) as follows:

where the first inequality holds by nonnegativity of Kullback–Leibler divergence, and the second equality follows from the fact that Π′\Pi^{\prime} is a refinement of Π\Pi.

References