Robust and Scalable Bayes via a Median of Subset Posterior Measures
Stanislav Minsker, Sanvesh Srivastava, Lizhen Lin, David B. Dunson
Introduction
Contemporary data analysis problems pose several general challenges. One is resource limitations: massive data require computer clusters for storage and processing. Another problem occurs when data are severely contaminated by “outliers” that are not easily identified and removed. Following Box and Tiao (1968), an outlier can be defined as “being an observation which is suspected to be partially or wholly irrelevant because it is not generated by the stochastic model assumed.” While the topic of robust estimation has occupied an important place in the statistical literature for several decades and significant progress has been made in the theory of point estimation, robust Bayesian methods are not sufficiently well-understood.
Our main goal is to make a step towards solving these problems, proposing a general Bayesian approach that is
provably robust to the presence of outliers in the data without any specific assumptions on their distribution or reliance on preprocessing;
scalable to big data sets through allowing computational algorithms to be implemented in parallel for different data subsets prior to an efficient aggregation step.
The proposed approach consists in splitting the sample into disjoint parts, implementing Markov chain Monte Carlo (MCMC) or another posterior sampling method to obtain draws from each “subset posterior” in parallel, and then using these draws to obtain weighted samples from the median posterior (or M-Posterior), a new probability measure which is a (properly defined) median of a collection of subset posterior distributions. We show that, despite the loss of “interactions” among the data in different groups, the final result still admits strong guarantees; moreover, splitting the data gives certain advantages in terms of robustness to outliers.
In particular, we demonstrate that the M-posterior is a probability measure centered at the “robust” estimator of the unknown parameter, the associated credible sets are often of the same “width” as the credible sets obtained from the usual posterior distribution and admit strong “frequentist” coverage guarantees (see section 3.3 for exact statements).
The paper is organized as follows: section 1.1 contains an overview of the existing literature and explains the goals that we aim to achieve in this work. Section 2 introduces the mathematical background and key facts used throughout the paper. Section 3 describes the main theoretical results for the median posterior. Section 4 presents details of algorithms, implementation, and numerical performance of the median posterior for several models. The simulation study and analysis of data examples convincingly show the robustness properties of the median posterior. We have also implemented the matrix completion example based on the MovieLens data set (GroupLens Research, 2013) which illustrates the scalability of the model. Proofs that are omitted in the main text are contained in the appendix.
A. Dasgupta remarks that (see the discussion following Berger (1994)): “Exactly what constitutes a study of Bayesian robustness is of course impossible to define.” The popular definition (which also indicates the main directions of research in this area) is due to J. Berger (Berger, 1994): “Robust Bayesian analysis is the study of the sensitivity of Bayesian answers to uncertain inputs. These uncertain inputs are typically the model, prior distribution, or utility function, or some combination thereof.” Outliers are typically accommodated by either employing heavy-tailed likelihoods (e.g., Svensen and Bishop (2005)) or by attempting to identify and remove them as a first step (as in Box and Tiao (1968) or Bayarri and Berger (1994)). The usual assumption in the Bayesian literature is that the distribution of the outliers can be modeled (e.g., using a -distribution, contamination by a larger variance parametric distribution, etc). In this paper, we instead bypass the need to place a model on the outliers and do not require their removal prior to analysis. We base inference on the median posterior, whose robustness can be formally and precisely quantified in terms of concentration properties around the true delta measure under the potential influence of outliers and contaminations of arbitrary nature.
Also relevant is the recent progress in scalable Bayesian algorithms. Most methods designed for distributed computing share a common feature: they efficiently use the data subset available to a single machine and combine the “local” results for “global” learning, while minimizing communication among cluster machines (Smola and Narayanamurthy (2010)). A wide variety of optimization-based approaches are available for distributed learning (Boyd et al., 2011); however, the number of similar Bayesian methods is limited. One of the reasons for this limitation is related to Markov chain Monte Carlo (MCMC), the dominating approach for approximating the posterior distribution of parameters in Bayesian models. While there are many efficient MCMC techniques for sampling from posterior distributions based on small subsets of the data (called “subset posteriors” in the sequel), to the best of our knowledge, there is no general rigorously justified approach for combining the subset posteriors into a single distribution for improved performance.
Three major approaches exist for scalable Bayesian learning in a distributed setting. The first approach independently evaluates the likelihood for each data subset across multiple machines and returns the likelihoods to a “master” machine, where they are appropriately combined with the prior using conditional independence assumptions of the probabilistic model. These two steps are repeated at every MCMC iteration (see Smola and Narayanamurthy (2010); Agarwal and Duchi (2012)). This approach is problem-specific and involves extensive communication among machines. The second approach uses a so-called stochastic approximation (SA) and successively learns “noisy” approximations to the full posterior distribution using data in small mini-batches. The accuracy of SA increases as it uses more data. A group of methods based on this approach uses sampling-based techniques to explore the posterior distribution through modified Hamiltonian or Langevin dynamics (e.g., Welling and Teh (2011); Ahn, Korattikara and Welling (2012); Korattikara, Chen and Welling (2013)). Unfortunately, these methods fail to accommodate discrete-valued parameters and multimodality. Another subgroup of methods uses deterministic variational approximations and learns the variational parameters of the approximated posterior through an optimization-based approach (see Wang, Paisley and Blei (2011); Hoffman et al. (2013); Broderick et al. (2013)). Although these techniques often have excellent predictive performance, it is well known (Bishop, 2006) that variational methods tend to substantially underestimate posterior uncertainty and provide a poor characterization of posterior dependence, while lacking theoretical guarantees.
Our approach instead falls in a third class of methods which avoid extensive communication among machines by running independent MCMC chains for each data subset and obtaining draws from subset posteriors. These subset posteriors can be combined in a variety of ways. Some of these methods simply average draws from each subset (Scott et al., 2013). Other alternatives use an approximation to the full posterior distribution based on kernel density estimates (Neiswanger, Wang and Xing, 2013) or the so-called Weierstrass transform (Wang and Dunson, 2013). These methods have limitations related to the dimension of the parameter, moreover, their applicability and theoretical justification are restricted to parametric models. Unlike the method proposed below, none of the aforementioned algorithms are provably robust.
Our work was inspired by recent multivariate median-based techniques for robust estimation developed in Minsker (2013) (see also Hsu and Sabato (2013); Alon, Matias and Szegedy (1996); Lerasle and Oliveira (2011); Nemirovski and Yudin (1983) where similar ideas were applied in different frameworks).
Preliminaries
We proceed by recalling key definitions and facts which will be used throughout the paper.
Finally, given two nonnegative sequences and , we write if for some and all . Other objects and definitions are introduced in the course of exposition when necessity arises.
2 Generalizations of the univariate median
and set
We will say that is the metric median of . Note that always belongs to by definition. Advantages of this definition are its generality (only metric space structure is assumed) and simplicity of numerical evaluation since only the pairwise distances are required to compute the median. This construction was previously employed in Nemirovski and Yudin (1983) in the context of stochastic optimization and is further studied in Hsu and Sabato (2013). A closely related notion of the median was used in Lopuhaa and Rousseeuw (1991) under the name of the “minimal volume ellipsoid” estimator.
Finally, we recall an important property of the median (shared both by and ) which states that it transforms a collection of independent, “weakly concentrated” estimators into a single estimator with significantly stronger concentration properties. Given such that , define a nonnegative function via
The following result is an adaptation of Theorem 3.1 in Minsker (2013):
Let be the geometric median of . Then
Let . Then
While we require above for clarity and to keep the constants small, we prove a slightly more general result that holds for any .
Theorem 2.1 implies that the concentration of the geometric median of independent estimators around the “true” parameter value improves geometrically fast with respect to the number of such estimators, while the estimation rate is preserved, up to a constant. In our case, the role of ’s will be played by posterior distributions based on disjoint subsets of observations, viewed as elements of the space of signed measures equipped with a suitable distance.
Parameter allows taking corrupted observations into account: if the initial sample contains not more than outliers (of arbitrary nature), then at most estimators amongst can be affected but their median remains stable, still being close to the unknown with high probability. To clarify the notion of “robustness” that such a statement provides, assume that are consistent estimators of based on disjoint samples of size each. If , then , hence the breakdown point of the estimator is is general. However, it is able to handle a number of outliers that grows like while preserving consistency, which is the best one can hope for without imposing any additional assumptions on the underlying distribution, parameter of interest or nature of the outliers.
Let us also mention that the the geometric median of a collection of points in a Hilbert space belongs to the convex hull of these points. Thus, one can think about “downweighing” some observations (potential outliers) and increasing the weight of others, and geometric median gives a way to formalize this approach. The median defined in (2.3) corresponds to the extreme case when all but one weight are equal to . Its potential advantage lies in the fact that its evaluation requires only the knowledge of pairwise distances , see (2.2).
3 Distances between probability measures
Next, we discuss the special family of distances between probability measures that will be used throughout the paper. These distances provide the necessary structure to define and evaluate medians in the space of measures, as discussed above. Since one of our goals was to develop computationally efficient techniques, we focus on distances that admit accurate numerical approximation.
Important special cases include the situation when
where is the Lipschitz constant of .
It is well-known (Dudley (2002), Theorem 11.8.2) that in this case is equal to the Wasserstein distance (also known as the Kantorovich-Rubinstein distance)
where denotes the law of a random variable and the infimum on the right is taken over the set of all joint distributions of with marginals and .
Note that when and are discrete measures (e.g., and ), then
In this paper, we will only consider characteristic kernels, which means that if and only if . It follows from Theorem 7 in Sriperumbudur et al. (2010) that a sufficient condition for to be characteristic is its strict positive definiteness: we say that is strictly positive definite if it is bounded, measurable, and such that for all non-zero signed Borel measures
If is compactly supported, then is characteristic.
For i.i.d samples, a useful and favorable fact is that often does not depend on : under weak assumptions on kernel , has an upper bound of order (that is, can be made arbitrarily small by choosing big enough, see Corollary 12 in Sriperumbudur et al. (2009)). On the other hand, the bound for the (stronger) Wasserstein distance is not dimension-free and is of order . Similar error rates hold for empirical measures based on samples from Markov Chains used to approximate invariant distributions, including MCMC samples (see Boissard and Le Gouic (2014) and Fournier and Guillin (2013)).
it will be useful to assume that the class is chosen such that the distance between the measures is lower bounded by the distance between their means, namely
Then is characteristic and satisfies (2.13) with .
The total variation distance between two probability measures defined on a -algebra is
Contributions and main results
This section explains the construction of “median posterior” (or M-Posterior) distribution, along with the theoretical guarantees for its performance.
and assume that the metric space is separable.
Let be a characteristic kernel defined on . Kernel defines a metric on via
All subsequent results apply to this special case. While this is a “natural” metric for the problem, the disadvantage of is that it is often difficult to evaluate numerically. Instead, we will consider metrics that are “dominated” by (this is formalized in assumption 3.4).
for all Borel measurable sets . It is known (see Ghosal, Ghosh and Van Der Vaart (2000)) that under rather general assumptions the posterior distribution “contracts” towards , meaning that
almost surely or in probability as for a suitable sequence .
One of the questions that we address can be formulated as follows: what happens if some observations in are corrupted, e.g., if contains outliers of arbitrary nature and magnitude? Even if there is only one “outlier”, the usual posterior distribution might concentrate most of its mass “far” from the true value .
We proceed with a general description of our proposed algorithm for constructing a robust version of the posterior distribution. Let be an integer. Divide the sample into disjoint groups of size each:
A good choice of efficiently exploits the available computational resource while ensuring that the groups s are sufficiently large.
Let be a prior distribution over , and let
be the family of subset posterior distributions depending on disjoint subgroups :
where the medians and are evaluated with respect to or introduced in section 2.2 above. Note that and are always probability measures: indeed, due to the aforementioned properties of a geometric median, there exists such that , and by definition.
To overcome this difficulty, we propose a modification of our approach where the random measures are replaced by the stochastic approximations , of the full posterior distribution. To this end, define the “stochastic approximation” based on the subsample as
where we assume that is an integrable function for all . In other words, is obtained as a posterior distribution given that each data point from is observed times. While each of might underestimate uncertainly, the median (or ) of these random measures yields credible sets with much better coverage. This approach shows good performance in numerical experiments. One of our main results (see section 3.3) provides a justification for this observation (albeit, under rather strong assumptions and for the parametric case).
2 Convergence of posterior distribution and robust Bayesian inference
In this subsection, we study the contraction and robustness properties of the median posterior.
Our first result establishes the “weak concentration” property of the posterior distribution around the true parameter. Let be the Dirac measure supported on . Recall the following version of Theorem 2.1 in Ghosal, Ghosh and Van Der Vaart (2000) (we state the result for the Wasserstein distance rather than the (closely related) contraction rate of the posterior distribution). Here, the Wasserstein distance is evaluated with respect to the “Hellinger metric” defined in (3.1).
Let be an i.i.d. sample from . Assume that and are such that for some constant
The proof closely mimics the argument behind Theorem 2.1 in Ghosal, Ghosh and Van Der Vaart (2000). Details are outlined in section A.2. ∎
Conditions of Theorem 3.1 are standard assumptions guaranteeing that the resulting posterior distribution contracts to the true parameter at the rate . Note that the bounds for the distance slightly differ from the contraction rate itself: indeed, we have
hence to obtain the inequality , we usually require
which adds an extra logarithmic factor in the parametric case.
Therefore, and , where and were defined in (2.10) and (2.8) respectively, and the underlying metric structure is given by . In particular, convergence with respect to implies convergence with respect to .
Let be an i.i.d. sample from , and assume that is defined with respect to the norm as in (3.4) above. Set , assume that conditions of Theorem 3.1 hold, and, moreover, that satisfies
It is enough to apply part (a) of Theorem 2.1 with to the independent random measures . Note that the “weak concentration” assumption (A.2) is implied by (3.6). ∎
Once again, note the exponential improvement of concentration as compared to Theorem 3.1. It is easy to see that a similar statement holds for the median defined in (3.4) (even for the stronger Wasserstein distance ), modulo changes in constants.
While the result of the previous statement is promising, numerical approximation and sampling from the “robust posterior” is often problematic due to the underlying geometry defined by the Hellinger metric, and the associated distance is hard to estimate in practice. Our next goal is to derive similar guarantees for the M-posterior evaluated with respect to the computationally tractable family of distances discussed in section 2.3 above.
To transfer the conclusions of Theorem 3.1 and Corollary 3.2 to the case of other kernels and associated metrics , we need to guarantee the existence of tests versus the complements of the balls in these distances. Such tests can be obtained from comparison inequalities between distances.
where is the Hellinger distance or the Euclidean distance (in the parametric case).
When is the Euclidean distance, we will impose an additional mild assumption guaranteeing existence of test versus the complements of the balls (for the Hellinger distance, this is always true, see Ghosal, Ghosh and Van Der Vaart (2000)). Namely, we will assume that for every and every pair , there exists a test such that for some and a universal constant
Below, we provide several examples of kernels satisfying the stated assumption.
For finite-dimensional models, we will be especially interested in kernels such that the associated metric is bounded by the Euclidean distance. The following proposition gives a sufficient condition for this to hold.
Moreover, the result of the previous proposition clearly remains valid for kernels of the form
We are ready to state our main result for convergence with respect to the RKHS-induced distance .
Assume that conditions of Theorem 3.1 hold with being the Hellinger or the Euclidean distance, and that assumption 3.4 is satisfied. In addition, let prior be such that
The result essentially follows from the combination of Theorem 3.1 and assumption 3.4, see section A.3 in the appendix for details. ∎
Theorem 3.8 yields the “weak” estimate that is needed to obtain the stronger bound for the M-Posterior distribution . This is summarized in the following corollary:
Let be an i.i.d. sample from , and assume that is defined with respect to the distance as in (3.4) above. Let . Assume that conditions of Theorem 3.8 hold, and, moreover, is such that
It is enough to apply parts (a) and (b) of Theorem 2.1 with to the independent random measures . Note that the “weak concentration” assumption (A.1) is implied by (3.10). ∎
where is the mean of . In other words, this shows that the M-posterior mean is the “robust” estimator of .
3 Bayesian inference based on stochastic approximation of the posterior distribution
As we have already mentioned in section 3.1, when the number of disjoint subgroups is large, the resulting M-Posterior distribution is “too flat”, which results in large credible sets and overestimation of uncertainty. Clearly, the source of the problem is the fact that each individual random measure is based on sample of size which can be much smaller than .
where we have assumed that is integrable. Here, can be viewed as an approximation of the full data likelihood. We call the random measure the -th stochastic approximation to the full posterior distribution.
Of course, such a “correction” negatively affects coverage properties of the credible sets associated with each measure . However, taking the median of stochastic approximations yields improved coverage of the resulting M-posterior distribution. The main goal of this section is to establish an asymptotic statement in spirit of a Bernstein-von Mises theorem for the M-posterior based on stochastic approximations .
We will start by showing that under certain assumptions the upper bounds for the convergence rates of towards are the same as for , the “standard” posterior distribution given .
be the bracketing entropy. In what follows, denotes the “Hellinger ball” of radius centered at .
There exist constants and such that if
In particular, one can choose and .
whenever , and
there exists such that for ,
Application of theorem 3.10 to the analysis of “stochastic approximations” yields the following result.
Let be such that conditions of Theorem 3.10 hold with , and
Note that for the kernel of the form (3.9), assumption 3.4 reduces to the inequality between the Hellinger and Euclidean distances.
As before, Theorem 2.1 combined with the “weak concentration” inequality of Theorem 3.11 gives stronger guarantees for the median (or its alternative ) of . Exact statement is very similar in spirit to Corollary 3.9.
We will first state a preliminary result for each individual “subset posterior” distribution:
Let be an i.i.d. sample from for some in the interior of . Assume that
the family is differentiable in quadratic mean;
the prior has a density (with respect to the Lebesgue measure) that is continuous and positive in the neighborhood of ;
conditions of Theorem 3.10 hold with for some and large enough.
in -probability as .
The proof follows standard steps (e.g., Theorem 10.1 in Van der Vaart (2000)), where the existence of tests is substituted by the inequality of Theorem 3.10. See section A.5 in the appendix for more details. ∎
The implication of this result for the M-posterior is the following: if is the kernel of type (3.9), for sufficiently regular parametric families (differentiable in quadratic mean, with “well-behaved” bracketing numbers, satisfying assumption 3.4 for the Euclidean distance with ) and regular priors, then
the M-posterior is well approximated by a normal distribution centered at the “robust” estimator of unknown ;
the estimator is a center of the confidence set of level and diameter of order (same as we would expect for this level for the usual posterior distribution - however, the bound for the M-posterior holds for finite sample sizes).
in -probability when , where is the mean of .
Assume that conditions (a), (b) of Theorem 3.11 hold with
and . Then for all and large enough,
(a) It is easy to see that convergence in total variation norm, together with an assumption that the prior distribution satisfies
implies that the expectations converge in -probability as well:
Together with an observation that the total variation distance between and is bounded by the multiple of , it implies that we can replace by the mean
in other words, the conclusion of Proposition 3.13 can be stated as
in -probability. Now assume that is fixed, and let . As before, let be disjoint groups of i.i.d. observations from of cardinality each. Recall that, by the definition (2.3) of , for some , and is the mean of . Clearly, we have
(b) Let where large enough so that
In particular, for and , we obtain the bound
for some constant independent of . Note that itself depends on , hence this bound is not uniform, and holds only for a given confidence level .
It is convenient to interpret this (informally) in terms of the credible sets: to obtain the credible set with “frequentist” coverage level , pick and use the - credible set of the M-posterior .
Numerical algorithms and examples
In this section, we consider examples and applications in which comparisons are made for the inference based on the usual posterior distribution and on the M-Posterior. One of the well-known and computationally efficient ways to find the geometric median in Hilbert spaces is the famous Weiszfeld’s algorithm (introduced in Weiszfeld (1936)). Details of implementation are described in Algorithms 1 and 2. Algorithm 1 is a particular case of Weiszfeld’s algorithm applied to subset posterior distributions and distance , while Algorithm 2 shows how to obtain an approximation to M-Posterior given the samples from . Note that the subset posteriors whose “weights” in the expression of the M-Posterior are small (in our case, smaller than ) are excluded from the analysis. Our extensive simulations show the empirical evidence in favor of this additional thresholding step.
Detailed discussion of convergence rates and acceleration techniques for Weiszfeld’s method from the viewpoint of modern optimization can be found in Beck and Sabach (2013). For alternative approaches and extensions of Weiszfeld’s algorithm, see Bose, Maheshwari and Morin (2003), Ostresh (1978), Overton (1983), Chandrasekaran and Tamir (1990), Cardot, Cénac and Zitt (2012), Cardot, Cénac and Zitt (2013), among other works.
In all numerical simulations below, we use “stochastic approximations” and the corresponding median measure , unless noted otherwise.
Before presenting the results of numerical analysis, let us remark on two important computational aspects.
The number of subsets appears as a “free parameter” entering the theoretical guarantees for M-Posterior. One interpretation of (in terms of the credible sets) is given in the end of section 3.3. Our results also imply that partitioning the data into subsets guarantees robustness to the presence of outliers of arbitrary nature.
In many applications, is dictated by the sample size and computational resources (e.g., the number of available machines). In section B.3 of the appendix, we describe a heuristic approach to selection of that shows good practical performance. As a rule of a thumb, we recommend choosing as larger values of lead to an M-posterior that overestimates uncertainty. This heuristic is supported by the numerical results presented below.
It is easy to get a general idea regarding the potential improvement in computational time complexity achieved by the M-Posterior. Given the data set of size , let be the running time of the algorithm (e.g., MCMC) that outputs a single observation from the posterior distribution . If the goal is to obtain samples from the posterior, then the total running time is . Let us compare this time with the running time needed to obtain samples from the -posterior given that the algorithm is running on machines in parallel. In this case, we need to generate samples from each of subset posteriors, which is done in time , where is typically large and . According to Theorem 7.1 in Beck and Sabach (2013), Weiszfeld’s algorithm approximates the M-Posterior to degree of accuracy in at most steps, and each of these steps has complexity (which follows from (2.12)), so that the total running time is
The term can be refined in several ways via application of more advanced optimization techniques (see the aforementioned references). If, for example, for some , then which should be compared to required by the standard approach.
To give a specific example, consider an application of (4.1) in the context of Gaussian process (GP) regression. If is the number of training samples, then GP regression has asymptotic time complexity to obtain samples from the posterior distribution of GP (Rasmussen and Williams, 2006, Algorithm 2.1). Assuming we have access to machines, the time complexity to obtain samples from M-Posterior in GP regression is . If for example for some and , we get improvement in running time.
In many cases, replacing the “subset posterior” by the stochastic approximation does not result in increased sampling complexity: indeed, the log-likelihood in the sampling algorithm for the subset posterior is simply multiplied by to obtain the sampler for the stochastic approximation. We have included the description of a modified Dirichlet mixture model in section B.2 of the appendix as an illustration.
M-Posterior was more robust than its competitors and its performance improved with increasing magnitude of the outlier. We compared the performance of “consensus posterior”, the overall posterior, and the M-Posterior using the empirical coverage of (1-)100% credible intervals (CIs) calculated across 50 replications for , and 0.05. The empirical coverages of M-Posterior’s CIs showed robustness to magnitude of the outlier. On the contrary, performance of the consensus and overall posteriors deteriorated fairly quickly across all ’s leading to 0% empirical coverage as magnitude of the outlier increased from to (Figure 1). We compared uncertainty quantification of the M-Posterior with that of the overall posterior using relative lengths of their CIs, with zero value corresponding to identical lengths and a positive value to wider CIs of the M-Posterior. We found that widths of CIs for both posteriors were fairly similar for , with M-Posterior’s CIs being slightly wider in absence of large outliers (Figure 2).
Stochastic approximation was important for proper calibration of uncertainty quantification. The empirical coverages of (1-)100% CIs of the M-Posterior without stochastic approximation overcompensated for uncertainty at all levels of (Figure 3). Similarly, lengths of the CIs of M-Posterior without stochastic approximation are wider than those with stochastic approximation (Figure 4). Both these observations showed that stochastic approximation led to shorter CIs for M-Posterior that had empirical coverages close to their theoretical values.
The number of subsets () had an effect on credible interval length of the M-Posterior. We modified the simulation above and generated 1000 observations , with the last 10 observations in being outliers with value , . The simulation setup was replicated 50 times. M-Posteriors were obtained for . Across all values of , M-Posterior’s CI was compared to the CI of the overall posterior after removing the outliers; the relative difference of M-Posterior and the overall posterior CI lengths decreases for , where is the number of outliers, remains stable as increases to , and grows for larger values of (Figure 5a). This demonstrates that inference based on M-Posterior was not too sensitive to the choice of for a wide range of values.
2 Real data analysis: General social survey
The General Social Survey (GSS; gss.norc.org) has collected responses to questions about evolution of American society since 1972. We selected data for 9 questions from different social topics: “happy” (happy), “Bible is a word of God” (bible), “support capital punishment” (cap), “support legalization of marijuana” (grass), “support premarital sex” (pre), “approve bible prayer in public schools” (prayer), “expect US to be in world war in 10 years” (uswar), “approve homosexual sex relations” (homo), “support abortion” (abort). These questions were in the survey since 1988 and their answers were converted to two levels: yes or no. Missing data were imputed based on the average, resulting in a data set with approximately 28,000 respondents.
M-Posterior had similar uncertainty quantification as the overall posterior while being more efficient. M-Posterior was at least 10 () and 8 times () faster than the overall posterior and it used less than 25% of the memory resources required by the overall posterior (Figure 5b). The overall posterior was more concentrated than the M-Posterior for and 20; however, its coverage of maximum likelihood estimators for obtained from the test data was worse than that of the two M-Posteriors (Table 1).
References
Appendix A Remaining proofs.
We will prove a slightly more general result:
Let be the geometric median of . Then
where .
Let . Then
To get the bound stated in the paper, take and in part (a) and in part b.
We start by proving part a. To this end, we will need the following lemma (see lemma 2.1 in Minsker (2013)):
and . Then there exists a subset of cardinality such that for all , .
Assume that event occurs. Lemma A.1 implies that there exists a subset of cardinality such that for all , hence
If has Binomial distribution , then
(see Lemma 23 in Lerasle and Oliveira (2011) for a rigorous proof of this fact). Chernoff bound (e.g., Proposition A.6.1 in van der Vaart and Wellner (1996)), together with an obvious bound , implies that
To establish part b, we proceed as follows: let be the event
The rest of the proof repeats the argument of part a since
where is the complement of .
A.2 Proof of Theorem 3.1
By the definition of Wasserstein distance ,
(recall that is the Hellinger distance). Let be a large enough constant to be determined later. Note that the Hellinger distance is uniformly bounded by 1. Using (A.3), it is easy to see that
To this end, it remains to estimate the second term in the sum above. We will follow the proof of Theorem 2.1 in Ghosal, Ghosh and Van Der Vaart (2000). Bayes formula implies that
For any , Lemma 8.1 Ghosal, Ghosh and Van Der Vaart (2000) yields
for every probability measure on the set . Moreover, by the assumption on the prior ,
Consequently, with probability at least ,
Define the event B_{l}=\left\{\int\limits_{\Theta}\prod\limits_{i=1}^{l}\frac{p_{\theta}}{p_{0}}(X_{i})d\Pi(\theta)\leq\exp\Big{(}-(1+C_{1}+C)l\varepsilon_{l}^{2}\Big{)}\right\}.
Let be the set satisfying conditions of Theorem 3.1. Then by Theorem 7.1 in Ghosal, Ghosh and Van Der Vaart (2000), there exist test functions and a universal constant such that
Next, by the definition of , we have
To estimate the second term of last equation, note that
for . Set and note that with probability . It follows from (A.6), (A.2) and (A.2) and Chebyshev’s inequality that for any
A.3 Proof of Theorem 3.8
From equation (3.7) in the paper and proceeding as in proof of Theorem 3.2, we get that
Following in the proof of Theorem 3.1, we note that (where constants are the same as in the proof of Theorem 3.1). Also, note that
By Theorem 7.1 in Ghosal, Ghosh and Van Der Vaart (2000), there exist test functions and a universal constant such that
Repeating the steps of the proof of Theorem 3.1, we see that if is chosen large enough, then
A.4 Proof of Theorem 3.11
The proof strategy is similar to Theorem 3.1. Note that
where is the Hellinger distance.
Let . By the definition of , we have
To bound the denominator from below, we proceed as before. Let
where is a probability measure supported on . Lemma 8.1 in Ghosal, Ghosh and Van Der Vaart (2000) yields that for any , in particular, for the conditional distribution . We conclude that
To estimate the numerator in (A.11), note that if Theorem 3.10 holds for , then it also holds for for any . This observation implies that
with probability , hence
with the same probability. Choose large enough so that . Putting the bounds for the numerator and denominator of (A.11) together, we get that with probability ,
A.5 Proof of Proposition 3.13
The proof follows a standard pattern: on the first step, we show that it is enough to consider the posterior obtained from the prior restricted to a large compact set, and then proving the theorem for the prior with compact support. The second part mimics the classical argument exactly (e.g., see Van der Vaart (2000)).
To show that one can restrict the prior to the compact set, it is enough to establish that for large enough and ,
can be made arbitrarily small. This follows from the inclusion
(due to the assumed inequality between Hellinger and Euclidean distances) and the bounds for the numerator and the denominator of (A.12) established in the proof of Theorem 3.11.
Appendix B Numerical simulation: additional examples and details.
The generative model of p-parafac has two levels. First, prior probabilities of latent classes and parameters are sampled. Discrete random measure is generated using the stick-breaking construction of DP
where is the prior probability of responders responses belonging to the latent class . The prior probability of a response to th question depending on the latent class
Second, latent variables and parameters are generated, which are specific to responders in the training data. The latent class of th responder
Finally, the response of th responder for question
This generative model in turn implies that
All sampling algorithms were implemented in Matlab. The samplers ran for 10,000 iterations and every fifth sample was collected after a burn-in of 5000. All experiments were performed on Oracle Grid Engine cluster with 2.6GHz 16 core compute nodes. Memory resources were capped at 16GB and 64GB for sampling from subset and overall posteriors, respectively.
B.2 Dirichlet process mixture model sampling for the stochastic approximation
As we have mentioned, using the stochastic approximation in place of the subset posterior does not lead to increased sample complexity in many cases. Many Bayesian models involve hierarchical exponential family specifications, in which case conditional distributions in Gibbs sampling or acceptance probabilities in Metropolis-Hastings algorithms can be trivially modified to account for the weighted likelihood. Here, we present one particular example of the Dirichlet process mixture model. We augment the original sampling model with latent variables and raise the complete data likelihood to an appropriate power. We have observed excellent performance with this approach in other contexts, including the matrix completion example.
and ; and are respectively assigned Inverse-Gamma() and Gamma(, ) priors. Assume that data are partitioned into subsets of size such that data on subset are . The complete data likelihood for subset after stochastic approximation is
where is an indicator function, is the maximum number of atoms in the stick breaking representation for , and also depends on latent variables and . Full conditionals of latent variables and unknown parameters are tractable in terms of standard distributions. The Gibbs sampler iterates between the following five steps:
Sample from Normal(, ) for , where
Sample for from the categorical distribution
Sample from Inverse-Gamma(), where
Sample from Beta(, ) for .
Sample from Gamma(, ).
We fix and following standard conventions.
B.3 Selection of the optimal number of subsets m𝑚m
The following heuristic approach picks the median among the candidate -posteriors. Namely, start by evaluating the M-Posterior for each in the range of candidate values :
and choose such that
where is the metric median defined in (2.3) in the section 2 of the paper.