A Size-Free CLT for Poisson Multinomials and its Applications
Constantinos Daskalakis, Anindya De, Gautam Kamath, Christos Tzamos
Introduction
In theoretical computer science, PMDs are commonly used in the analysis of randomized algorithms, often through large deviation inequalities. They have also found applications in algorithmic problems where one is looking for a collection of random vectors optimizing a certain probabilistic objective, or satisfying probabilistic constraints. For example, understanding the behavior of PMDs has led to polynomial-time approximation schemes for anonymous games [Mil96, Blo99, Blo05, Kal05, DP07, DP08, DP09], despite the PPAD-completeness of their exact equilibria [CDO15]. Anonymous games are games where a large number of players share the same strategies, and each player’s utility only depends on his own choice of strategy and the number of other players that chose each of the strategies. In particular, the expected payoff of each player depends on the PMD resulting from the mixed strategies of the other players. It turns out that understanding the behavior of PMDs provides a handle on the computation of approximate Nash equilibria. One of our main contributions is to advance the state of the art for computing approximate Nash equilibria in anonymous games. We will come to this contribution shortly.
Recently Valiant and Valiant have used PMDs to obtain sample complexity lower bounds for testing symmetric properties of distributions [VV11]. The workhorse in their lower bounds is a new CLT bounding the total variation distance between a -GMD and a multidimensional Gaussian with the same mean vector and covariance matrix. Since they are comparing a discrete to a continuous distribution under the total variation distance, they need to discretize the Gaussian by rounding its coordinates to their closest point in the integer lattice. If is distributed according to some -GMD with mean vector and covariance matrix , and is distributed according to the multi-dimensional Gaussian , [VV11] shows that:
where is the minimum eigenvalue of and denotes the rounding of to the closest point in the integer lattice. The dependence of the bound on the dimension and the minimum eigenvalue is necessary, and quite typical of Berry-Esseen type bounds. Answering a question raised in [VV11], we prove a qualitatively stronger CLT by showing that the explicit dependence of the bound on can be removed (hence, the CLT is “size-free”).
Suppose that is distributed according to some -GMD with mean and covariance matrix , and . There exists some constant such that
where is the minimum eigenvalue of .
Interestingly, Theorem 1 is proven by bootstrapping the Valiant-Valiant CLT itself. Indeed, this CLT was used as one of the key ingredients in a recent structural characterization of PMDs [DKT15], where it was shown that any -Poisson multinomial random vector is -close in total variation distance to the sum of an (appropriately discretized) Gaussian and a -Poisson multinomial random vector; see Theorem 6. In turn, we prove Theorem 1 by using Theorem 6 as a black box.
For more details on our proof’s approach, see Section 3.
In the remainder of this section we discuss the algorithmic applications of our CLT, concluding with our improved algorithms for learning PMDs using Fourier analysis.
We have already discussed anonymous games earlier in this section, where we have also explained their relation to PMDs. In particular, the expected utility of some player in a -player -strategy anonymous game only depends on his own choice of mixed strategy and the -Poisson multinomial random vector aggregating the mixed strategies of his opponents. It is therefore natural to expect that a better understanding of the structure of PMDs could lead to improved algorithms for computing Nash equilibria in these games. Indeed, earlier work [DP08, DP15] has exploited this connection to obtain algorithms for approximate Nash equilibria, whose running time is
While clearly of theoretical interest, this bound shows that anonymous games are one of the few classes of games where approximate equilibria can be efficiently computed, while exact equilibria are PPAD-hard [CDO15], even for -player -strategy anonymous games. Exploiting our CLT we obtain a significant improvement over [DP08].
An -approximately well supported Nash equilibrium of an -player -strategy anonymous games whose utilities are in to make these approximations meaningful.
The salient feature of Theorem 2 is the polynomial dependence of the running time on and its quasi-polynomial dependence on . In terms of these dependencies our algorithm matches the best known algorithm for -strategy anonymous games [DP09], where much more is known given the single-dimensional nature of -PMDs.
Moreover, the recent hardness results for anonymous games [CDO15] establish that not only finding an exact but also a -approximate Nash equilibrium is PPAD-hard. An interesting corollary of Theorem 2 is that this cannot be pushed to -approximations, unless PPAD can be solved in quasi-polynomial time.
Unless PPAD Quasi-PTIME, it is not PPAD-hard to find a -approximately well supported Nash equilibrium in anonymous games, for any .
It is interesting to contrast this corollary with normal-form games where it is known that computing inverse polynomial approximations is PPAD-hard [DGP09, CDT09].
From a technical standpoint, our algorithm for anonymous games uses the structural understanding of PMDs as follows. Since every player views the aggregate strategies of the other players as a PMD, one approach would be to guess each player’s view using a cover as developed in [DKT15]. However, this approach gives a runtime which is exponential in , since it requires us to enumerate the cover for each player. An alternative approach is to guess the overall PMD which occurs at a Nash equilibrium, and guess appropriate “corrections” that allow us to infer each player’s view. To do this, we must find an alternative PMD which approximately matches the PMD at Nash in the following sense:
The PMD that results by removing the CRV corresponding to a player should be close to the view that the player observes;
A player’s CRV must only assign probability to strategies which are approximate best responses to his view.
It turns out that these conditions can be satisfied by using a careful dynamic program together with the structural understanding provided by [DKT15] and the CLT of Theorem 1. According to this structural result, we can partition the players into a “sparse” and a “Gaussian” component. Moreover, our CLT implies that matching the first two moments of the Gaussian suffices to approximate this component. This allows us to perform guesses at a different granularity for the sparse and Gaussian components. Roughly speaking, our dynamic program guesses a succinct representation of the two components and tries to compute CRVs which obey this representation and satisfy the conditions outlined above.
For more details on our PTAS, refer to Section 4.
Moreover, we can efficiently enumerate this cover in time polynomial in its size.
It is important to contrast Theorem 3 with Theorem 2 in [DKT15], which provides a non-proper cover whose size is similar, albeit with a leading factor of . Instead, our cover is proper, which is important for approximation algorithms that require searching over PMDs. Its dependence on is also optimal, as the number of -PMDs whose summands are deterministic is already . Moreover, we provide a lower bound for the dependence on , establishing that the quasi-polynomial dependence is also essentially optimal.
We describe our proper cover construction in two parts. First, we give details on how to construct a non-proper cover of size . The main tool we use is the existence of spectral sparsifiers for Laplacian matrices. Our non-proper cover sparsifies the non-proper cover of [DKT15], showing how its leading factor of can be reduced to . Roughly speaking, the factor of was due to spectrally approximating all possible covariance matrices , whose entries are bounded by . These covariance matrices corresponded to covariance matrices of -PMDs, and the cover maintained for each such some such that . (We call this guarantee a “-spectral approximation.”) The realization leading to our sparsification result is that covariance matrices of PMDs are in fact graph Laplacians. Indeed, a -PMD, , has covariance matrix, , corresponding to the sum of the covariance matrices of its summands. Now the covariance matrix of a -CRV, , is actually the Laplacian of a graph that has one node per dimension, along with an edge from node to node of weight ; and the covariance matrix of a -PMD is the Laplacian of the graph with the sum of the weights from each constituent -CRV– see Observation 1. We show that Laplacians corresponding to -PMDs can be -spectrally covered with a set of covariance matrices of size .
We appeal to recent results in spectral sparsification of Laplacian matrices [ST11, SS11, BSS12, BSST13]. In particular, we use the result of Batson, Spielman, and Srivastava [BSS12] (Theorem 8) to argue that the underlying graph can be sparsified to linearly many edges in the dimension . We do this in the hopes that we would have fewer parameters in the covariance matrix to guess. Unfortunately, the [BSS12] sparsification theorem has polynomial dependence in the accuracy. So applying it with a -approximation error, which is what we need, gives a meaningless result (namely no sparsification at all). Instead, we only use this theorem to get a rough -spectral cover of -PMD covariance matrices. Around every covariance matrix in this rough cover we grow a local -spectral cover. Roughly speaking, as the -spectral cover provides multiplicative approximation to the variance in every direction , every covariance matrix in this cover gives us a multiplicative handle on the eigenvalues of the matrices approximated by it. This is sufficient information to cover these matrices to -spectral error with a “local” spectral cover of size – see Lemma 6. Putting everything together, we get a -spectral cover of all covariance matrices of -PMDs of size – see Section 5.1.3. As covering these matrices was the bottleneck in the size of the non-proper cover, this completes the construction of a non-proper cover whose size is (4).
Further details on our non-proper construction are provided in Section 5.
Details on this conversion process are provided in Section 6.
Our lower bound is described further in Section 7. Our technique shows a lower bound on the metric entropy of a polynomial map of the moments of PMDs using an extension of Bézout’s theorem and other tools from algebraic geometry.
Finally, we give a new learning algorithm for PMDs:
This improves the learning algorithm from [DKT15] by eliminating the superpolynomial dependence on in the running time that was obtained in that paper. Our algorithm exploits properties of the continuous Fourier transform of a PMD, as opposed to recent work by Diakonikolas, Kane and Stewart on learning univariate sums of independent integer random variables, which uses the discrete Fourier transform [DKS16b]. They also apply similar discrete Fourier techniques in their simultaneous work on PMDs [DKS16a].
We note that such Fourier-based learning algorithms may simply output a description of the Fourier transform of a distribution. This allows one to compute the PMF of the distribution at any point of interest, but it is not obvious how to sample from such a description. Our algorithm outputs an explicit description of a distribution, which allows one to efficiently (i.e., in time independent of ) draw samples from the distribution. In contrast, they output the Fourier transform of a distribution and describe how to sample from it.
For more details on our learning algorithm, refer to Section 8.
1 Comparison of Results with [DKS16a]
Simultaneous to our work, Diakonikolas, Kane, and Stewart also studied Poisson Multinomial distributions [DKS16a]. In this section, we describe and compare their results with ours. While both papers independently prove many qualitatively similar results, the techniques are quite different, and thus both may be of independent interest.
Both papers prove new CLTs, which manage to remove the dependence on which is found in the CLT of [VV11], while the dependence on and remains polynomial. Additionally, both works improve upon the previous best covers for PMDs [DKT15]. First, both manage to reduce the size of the cover – interestingly, the two improvements seem to be orthogonal. Our result improves the dependence on from to , while theirs improves the dependence on and from to . We note that this upper bound holds for : for , [DKS16b] proves the tight cover size bound of . Furthermore, both papers describe how to efficiently achieve a proper cover of this size. These cover sizes are asymptotically optimal, as shown by lower bounds in both papers. In particular, the double-exponential dependence in is necessary. Both works also consider the problem of finding approximate Nash equilibria in anonymous games. The complexity of both algorithms is roughly comparable to the PMD cover size. Finally, both papers study the learning of PMDs, obtaining algorithms with sample complexity . The runtime of our algorithm is , and the runtime of their algorithm is , both in the standard word RAM model.
Preliminaries
We more formally define several of the distribution classes we consider.
A -Categorical Random Variable (-CRV) is a random variable that takes values in where is the -dimensional unit vector along direction . is the probability of observing .
An -Poisson Multinomial Distribution (-PMD) is given by the law of the sum of independent but not necessarily identical -CRVs. An -PMD is parameterized by a nonnegative matrix each of whose rows sum to is denoted by , and is defined by the following random process: for each row of matrix interpret it as a probability distribution over the columns of and draw a column index from this distribution. Finally, return a row vector recording the total number of samples falling into each column (the histogram of the samples).
We note that a sample from an -PMD is redundant – given coordinates of a sample, we can recover the final coordinate by noting that the sum of all coordinates is . For instance, while a Binomial distribution is over a support of size , a sample is -dimensional since the frequency of the other coordinate may be inferred given the parameter . With this inspiration in mind, we define the Generalized Multinomial Distribution, which is the primary object of study in [VV11].
A Truncated -Categorical Random Variable is a random variable that takes values in where is the -dimensional unit vector along direction , and is the dimensional zero vector. is the probability of observing the zero vector, and is the probability of observing .
An -Generalized Multinomial Distribution (-GMD) is given by the law of the sum of independent but not necessarily identical truncated -CRVs. A GMD is parameterized by a nonnegative matrix each of whose rows sum to at most is denoted by , and is defined by the following random process: for each row of matrix interpret it as a probability distribution over the columns of – including, if , an “invisible” column – and draw a column index from this distribution. Finally, return a row vector recording the total number of samples falling into each column (the histogram of the samples).
For both -PMDs and -GMDs, we will refer to and as the size and dimension, respectively.
We note that a PMD corresponds to a GMD where the “invisible” column is the zero vector, and thus the definition of GMDs is more general than that of PMDs. However, whenever we refer to a GMD in this paper, it will explicitly have a non-zero invisible column.
While we will approximate the Multinomial distribution with Gaussian distributions, it does not make sense to compare discrete distributions with continuous distributions, since the total variation distance is always . As such, we must discretize the Gaussian distributions. We will use the notation to say that is rounded to the nearest integer (with ties being broken arbitrarily). If is a vector, we round each coordinate independently to the nearest integer.
As seen in the definition of an -GMD, we have one coordinate which is equal to minus the sum of the other coordinates. We define a similar notion for a discretized Gaussian. However, we go one step further, to take care of when there are several such Gaussians which live in disjoint dimensions. By this, we mean that given two Gaussians, the set of directions in which they have a non-zero variance are disjoint. Without loss of generality (because we can simply relabel the dimensions), we assume all of a Gaussian’s non-zero variance directions are consecutive, i.e., the covariance matrix is all zeros, except for a single block on the diagonal. Therefore, when we add the covariance matrices, the result is block diagonal. The resulting distribution is described in the following definition.
The structure preserving rounding of a multidimensional Gaussian Distribution takes as input a multi-dimensional Gaussian with in block-diagonal form. It chooses one coordinate as a “pivot” in each block, samples from the Gaussian ignoring these pivots and rounds each value to the nearest integer. Finally, the pivot coordinate of each block is set by taking the difference between the sum of the means and the sum of the values sampled within the block.
Finally, we formally define the notion of a cover.
2 Probability Metrics
To compare probability distributions, we will require the total variation and Kolmogorov distances:
The total variation distance between two probability measures and on a -algebra is defined by
Unless explicitly stated otherwise, in this paper, when two distributions are said to be -close, we mean in total variation distance.
The Kolmogorov distance between two probability measures and with CDFs and is defined by
We note that Kolmogorov distance is, in general, weaker than total variation distance. In particular, total variation distance between two distributions is lower bounded by the Kolmogorov distance.
3 Miscellaneous Lemmata
We will use the following tools for bounding total variation distance between various random variables.
Let be two random variables over a domain . Fix any (possibly randomized) function on (which may be viewed as a distribution over deterministic functions on ) and let be the random variable such that a draw from is obtained by drawing independently from and from and then outputting (likewise for ). Then we have
Let be independent random variables, with , and define . Then for an absolute constant ,
Given two -dimensional Gaussians such that for all , , and the minimum eigenvalue of is at least ,
In addition, we prove the following general purpose lemma showing that two multivariate Gaussians with spectrally-close moments are close in total variation distance. This is intended to be a multivariate version of Proposition B.4 of [DDO+13], which proves a similar statement for univariate Gaussians. The proof appears in Section A.
Suppose there exist two -dimensional Gaussians, and , such that for all unit vectors ,
4 Results on PMDs from [DKT15]
Our work builds upon recent structural results on PMDs [DKT15]. We recall some of the key results which we will refer to in this paper.
Two key parameters used in this paper are and , set as and , for constants .
The main tool from this paper we will use is the structural characterization, stating that every PMD is close to the sum of an appropriately discretized Gaussian and a “sparse” PMD.
For parameters and as described above, every -Poisson multinomial random vector is -close to the sum of a Gaussian with a structure preserving rounding and a -Poisson multinomial random vector. For each block of the Gaussian, the minimum non-zero eigenvalue of is at least .
Finally, we will also use their rounding procedure, which relates a PMD to a nearby PMD with all parameters either equal to or sufficiently far from and :
A Size-Free CLT
We overview our proof of Theorem 1. Recall that the Central Limit Theorem of Valiant and Valiant, (1), has a poly-logarithmic dependence on the size parameter of the GMD. Their work raised the question whether this CLT could be made size-independent, and we resolve this conjecture by showing that it can be. This qualitative improvement comes at a quantitative loss in the polynomial dependence of the bound on the parameters and .
Our CLT builds off of the structural result of [DKT15], Theorem 6, which we use as a black box. This structural result says that every -PMD is -close to the sum of an appropriately discretized Gaussian and a -PMD. We note that the statement of Theorem 6 does not tell us anything about the moments of this Gaussian and sparse PMD, while our new CLT requires that the discretized Gaussian has the same moments as the original PMD. We prove this CLT in two steps. First, we show that the original PMD and the discretized Gaussian from the cover are close in total variation distance, i.e., we show that we can “drop” the sparse PMD component from Theorem 6 in the relevant approximation regime. Then, we bound the distance between the discretized Gaussian from the cover, , and a discretized Gaussian with the same mean and covariance as the original PMD, . The proof is concluded by combining these two bounds using the triangle inequality.
Next, we bound the distance between the discretized Gaussian from the cover, , and a discretized Gaussian with the same moments as the original PMD, . At this point, we know that and are close in total variation distance. By projecting both distributions in some direction and considering true Gaussians with the same moments as and , it can be shown that the first two moments are similar in this direction – otherwise, the true Gaussians would be far from each other in the Kolmogorov metric. This implies that the first two moments of and are close in every direction, as guaranteed by Proposition 8. Applying Lemma 2 tells us that bona-fide Gaussians with moments which are close in every direction are therefore close in total variation distance. The proof is concluded by applying the Data Processing inequality, which shows that the corresponding discretized Gaussians and are close as well.
We state and prove many useful lemmas in Section 3.1, which we combine to complete the proof of Theorem 1 in Section 3.2.
The following two propositions bound the Kolmogorov distance between a univariate Gaussian and the projection of a GMD or a discretized Gaussian, respectively.
Suppose that there exists an -generalized multinomial random vector , with mean vector and covariance matrix . Then for any unit vector ,
where is the minimum eigenvalue of .
We apply the Berry-Esseen theorem (Proposition 1). Let to recenter the random variables, and we will now compare with . We note that . Letting and , this implies that , and thus the Berry-Esseen bound gives
Suppose there exists a random variable . Then for any unit vector ,
where is the minimum eigenvalue of .
Let . We first show , which holds by Cauchy-Schwarz: and . Thus,
because the two distributions are univariate Gaussians with the same variance (which is at least and means shifted by . This implies
The following proposition compares a Gaussian and an arbitrary distribution . It shows that if ’s variance is much smaller than ’s, then they must be far in Kolmogorov distance.
Suppose there exists a univariate Gaussian with variance , and a distribution with variance . Then the Kolmogorov distance between and is at least .
We consider the event that a sample falls in an interval of width centered at . As a certificate of a large Kolmogorov distance between and , we show that the probability assigned to this interval is very different for versus .
First, by Chebyshev’s inequality, we know that
where the last inequality uses the Taylor expansion of the error function.
The difference in probability assigned to this interval is at least
Setting gives
The following proposition tells us if we are considering the sum of two random variables, one being a Gaussian with a large variance and one being an arbitrary distribution with a small support, we can remove all contribution from the distribution with small support and not pay a large cost in total variation distance.
We start by applying a law of total probability for total variation distance:
Using the data processing inequality for total variation distance (Lemma 1):
The next proposition tells us that Kolmogorov closeness implies parameter closeness for univariate Gaussians.
Without loss of generality, assume . We will first show the conclusion assuming the means are separated, and then assuming the variances are separated.
Our final proposition in this section applies the previous proposition, showing that total variation closeness implies parameter closeness (in any projection) when considering a GMD and a discretized Gaussian.
Consider the projections of and onto . By Propositions 3 and 4 and the triangle inequality, the Kolmogorov distance between the univariate Gaussians with the same mean and variance is at most . Applying Proposition 7 implies the desired result. ∎
2 Proof of Theorem 1
We will prove the statement for a sufficiently large constant . Thus we only need examine the case
otherwise the conclusion of the theorem statement is vacuous since total variation distance is at most .
As a starting point, we convert from a GMD to the corresponding -Poisson multinomial random vector and apply Theorem 6 with . This gives us that
where is a Gaussian with a structure preserving rounding and is a -Poisson multinomial random vector. By the definition of in Section 2.4, we have that for some constant . Thus, is a -Poisson multinomial random vector.
Now, applying Proposition 6 and the triangle inequality, we get
where .
We use the Data Processing Inequality (Lemma 1) followed by Lemma 2 with these guarantees to give:
Finally, applying the triangle inequality with Equation (7) gives
Choosing the constant sufficiently large completes the proof.
A PTAS for Anonymous Games
Indeed, if the above guarantee held, then the expected payoff of every player from any pure strategy would not change by more than an additive if we changed the strategies of all other players from to . So, if were a Nash equilibrium and , it would follow that is an approximate best response of player to . So would be an approximate equilibrium.
Unfortunately, we do not know how to construct a proper -cover of all -PMDs that has size of order (3) and such that for any there exists some satisfying Condition (8). Nevertheless, we can exploit our CLT and the structural result of [DKT15] (restated as Theorem 6 in this paper) to bypass this difficulty. Roughly speaking [DKT15] approximate a given by first discretizing the parameters of all ’s into fine enough accuracy (this is shown to only cost some in total variation distance), then partitioning the ’s into a small group of size that are left intact, and a large group whose sum is approximated by a discretized multidimensional Gaussian (up to another cost of in total variation distance). It is further shown that the distribution of the sum of variables in can be summarized through the vector of its first moments (at a loss of an additional in total variation distance), while the discretized Gaussian through its first two moments . Moreover, it is shown that the Gaussian has at least variance in all directions where it has non-zero variance. By enumerating over all possible summary statistics , a non-proper cover of all -PMDs can be obtained, whose size is of the order of (3).
Suppose now that is a Nash equilibrium whose approximating statistic in the non-proper cover is some . Given a correct guess for this statistic, our goal is to uncover an approximate Nash equilibrium of the game. By the construction of the cover, we know that every player either contributed his discretized to the discretized Gaussian with parameters , or to the small group of variables with moments . So, letting be the set of -CRVs whose parameters have the discretization accuracy used in the construction of the cover, we need to assign some to each player such that:
There exists a -size subset of players such that has vector of moments , while has first two moments .
For all , is a best response to .
To find a good assignment, we first construct a compatibility graph between players and mixed strategies in . We add an edge between some and some iff at least one of the following two conditions is met. We also annotate the edge with all conditions that are met:
( is compatible with ): is an approximate best response to the “environment” would observe if contributed to . If contributed to and Condition (a) were met, then we can deduce what PMD player would see in his environment. Indeed, this would be within some in total variation distance to a the sum of a Gaussian random vector with parameters and a PMD whose first moments are the same as after removing the contribution of . The updated moment vector can be computed from and as moments are symmetric polynomials of the underlying parameters. Given the updated moment vector, the PMD is determined to within in total variation distance, so its sum with the discretized Gaussian is also determined, and we can also efficiently determine whether is an approximate best response of player to that distribution.
( is compatible with ): is an approximate best response to the “environment” would observe if contributed to the discretized Gaussian with parameters . First, for this to be the case must be “compatible” with , i.e. not correlating uncorrelated pairs of dimensions/adding variance in zero-variance dimensions (or in other words, the block structure of should be preserved). Moreover, since all non-zero eigenvalues of are at least -large, the discretized Gaussian with parameters and are approximately the same(Proposition 2). At the same time, due to the largeness of the non-zero eigenvalues of , if condition (a) were eventually true, then our CLT (Theorem 1) would imply that is well-approximated by the discretized Gaussian with parameters , and hence by that with parameters . So, if , is assigned , and Condition (a) is eventually met, then the PMD that player sees in his environment is pinned down to within in total variation distance: it is approximately the sum of the discretized Gaussian with parameters and a PMD with moments . We can therefore check if is an approximate best response to that distribution.
After constructing the compatibility graph as above, we need to see if there is an assignment of players to compatible mixed strategies from so that (a) is satisfied. This looks non-trivial, but it can be done using dynamic programming. We sweep through the players, maintaining as state all possible leftover moments that may arise from assignments of a prefix of players to compatible mixed strategies. Given the discretization of , the set of possible states is bounded by (3). Importantly, the compatibility graph has the property that player is happy when given a compatible strategy as long as the overall assignment matches .
A mixed strategy profile is a set of distributions , where by we denote the -dimensional simplex, or, equivalently, the set of distributions over . A mixed strategy profile is an -approximately well supported Nash equilibrium (or an -Nash equilibrium, for short) if, for all and ,
where is the distribution over obtained by drawing random samples from independently according to the distributions , and forming the induced partition. We note that this an -Nash equilibrium is stronger than the related concept of an -approximate Nash equilibrium (see, i.e., [DGP09] for further discussion of this distinction). Throughout this paper, we solely consider the harder problem of computing an -Nash equilibrium.
A -Nash equilibrium is simply called a Nash equilibrium and it is always guaranteed to exist by Nash’s theorem.
2 An Algorithm for Anonymous Games
In a Nash equilibrium of an anonymous game every player uses a mixed strategy selecting strategy with probability . The distribution of the number of players which select each of the strategies is an -PMD. Using the fact that there exist small size -covers for PMDs, we can efficiently search over the space of all strategies and identify a mixed strategy profile that produces an -Nash equilibrium. We show that there exists an efficient polynomial time approximation scheme (EPTAS) for computing an -Nash equilibrium, thus proving Theorem 2.
The algorithm works by guessing an aggregate statistic that describes the overall behavior of all players. This statistic is based on the structural theorem shown in [DKT15], which shows that the overall PMD that describes the mixed strategy profile can be approximately written as sum of a discretized Gaussian and a sparse PMD with only components. Moreover, for the sparse PMD knowledge of the moments (which is equivalent to knowing the power-sums of all the summands up to , suffices to describe it within in total variation distance. Thus, the algorithm requires guessing the power-sums of the sparse PMD and the mean and covariance of the discretized Gaussian.
As we will show, knowledge of an individual’s strategy together with the aggregate statistic for the overall mixed strategy profile, allows us to compute an approximate distribution that discribes the player’s view about the aggregate strategy of everyone else. If we manage to assign strategies to every player so that approximately matched and additionally each player only chooses strategies that corresponds to approximate best responses with respect to his view we will obtain an -Nash equilibrium. The following lemma formalizes this intuition and is the main tool we use in the proof of Theorem 2.
For all and ,
Then, is an -Nash equilibrium for the game .
Consider the game . By Nash’s theorem there always exists a Nash equilibrium. Let be such an equilibrium where every player uses a mixed strategy selecting strategy with probability . The distribution of vectors which give the number of players which select each of the strategies is an -PMD.
To get an efficient algorithm, we need to search over a restricted set of strategies for each player. To be able to do that we must show that an -Nash equilibrium exists in a more restricted space. To argue that, we begin by a Nash equilibrium and perform a series of operations that maintain the property that the resulting mixed strategy profile is an -equilibrium.
We first proceed by rounding the probabilities so that they are either or at least as done in Lemma 3. This gives a PMD that is -close in total variation distance to . Moreover, if we consider the PMD , which is the -PMD obtained by removing the -th component from the rounded PMD , this is also -close in total variation to , i.e. the PMD obtained after removing the -th component from the original PMD . The proof of this statement is almost identical to the proof in [DKT15] and is omitted. That proof uses Poisson approximations to bound the total variation between the rounded and the unrounded PMDs and uses the fact that the means of the two PMDs can differ by at most in each coordinate. The only difference is that here, the means of the two PMDs can differ by at most in each coordinate which results in the same asymptotic bound for total variation distance. Moreover, note the rounding procedure doesn’t change any probabilities that were originally 0, i.e. .
By the structural theorem of [DKT15], the components of the PMD can be partitioned into two PMDs:
For every player in the sparse PMD, his view about the aggregate strategy of the others is approximately the same as if the large PMD was replaced by the Gaussian, i.e.
With those observations in hand, we proceed to give the algorithm for computing -equilibria for anonymous games. To do this, we first guess the mean and covariance of the Gaussian component as well as all the power-sums of the sparse PMD. We then try to construct CRVs for every player so that the overall mean and covariance as well as the power-sums match those that we guessed and moreover every player’s CRV assigns positive probability mass only to approximately optimal strategies. If we are able to do so, Lemma 4 implies that this gives an approximate Nash equilibrium. In more detail, the algorithm performs the following steps:
Guess the mean and covariance of the Gaussian component and the power sums of the sparse PMD. For every guess, we repeat the next steps until a feasible solution is found.
We need to guess the powersums for different PMDs since CRVs are first clustered according to their value in every coordinate. Since the parameters of the sparse PMD are all multiples of , this results in at most distinct power-sum vectors in totalThis upper bound was derived in [DKT15]. For the gaussian component all entries of the mean and covariance are multiples of which requires guesses in total.
For every player, we need to compute the contribution of his mixed strategy (CRV) to the overall distribution. If that player is to be assigned in the sparse component, its probabilities are all multiples of and we can compute its contribution to the power-sums . Similarly, if that player is to be assigned in the gaussian component its probabilities are all multiples of and we can easily compute its contribution to the mean and covariance.
However, not all assignments are feasible. We need to consider only CRVs for that player that assign positive probability mass to coordinates that are approximately best responses to the strategy of other players. Even though we don’t know the strategies of the others exactly, we can compute a good approximate description of the players view by subtracting from the power sums the players contribution (if any) and computing any that matches those power-sums. Similarly, if the player is mapped to the gaussian component we subtract the players mean and covariance from the overall mean and covariance and compute a discretized Gaussian with the resulting mean and covariance instead. We say that an assignment of a player to a component (sparse or Gaussian) and a specific distribution over strategies is feasible if it approximately maximizes the player’s utility with respect to his approximate view about the strategies of others.
To find if there exists a set of feasible strategies that matches the guessed statistic , we use dynamic programming. The states of our dynamic program are the following: For any prefix of players, we keep track of the remaining power-sums, mean and covariance we need to account for. We iteratively process players one by one keeping track of which states are reachable. Our estimation is feasible if after processing all players we have accounted for all the power-sums, mean and covariance in our original guess. If we find such a solution, we output the assignment of players to mixed strategies that resulted in this solution.
Applying Lemma 4 directly shows that this is indeed an -Nash equilibrium.
The total runtime of the algorithm is polynomial on the number of states of the above dynamic program. Since there are Gaussian parameters in total as well as power sums in total, the overall runtime is and the theorem follows. ∎
On the road to getting the proper cover described by Theorem 3, we first show Theorem 7. This constructs a non-proper cover of the same size. The main theorem of this section is the following:
Moreover, we can efficiently enumerate this cover in time polynomial in its size.
This theorem should be contrasted with Theorem 3, which provides a proper cover of similar size. It should also be contrasted to Theorem 2 of [DKT15], which provides a cover with a leading factor of , so the cover presented here improves the exponent of from quadratic to linear in the dimension. This is the correct order of exponential dependence on , as simply counting the number of -PMDs with deterministic summands gives a lower bound of . We also show in Section 7 that the quasi-polynomial dependence on with an exponent of cannot be avoided, as we provide an essentially matching lower bound on the cover size.
The starting point for our cover will be Theorem 6, stating that every -PMD is -close to the sum of an appropriately discretized Gaussian and a -PMD. We generate an -cover for each and combine them by triangle inequality.
We cover the sparse PMD component using the same methods as in [DKT15]. The first, naive way of covering this component involves gridding over all parameters with granularity. This results in a cover size of .
The more sophisticated way of covering this component uses a “moment matching” technique. A result by Roos [Roo02] shows that the probability mass function can be written as the weighted sum of partial derivatives of a standard multinomial distribution. When analyzed carefully, his result implies that the lower order moments of the distribution are sufficient to characterize the PMD. In other words, any two PMDs with identical “moment profiles” (which describe these lower order moments) are close in total variation distance, and it suffices to keep only one representative for each moment profile. This method results in a cover of size . Combining this with the other approach gives a cover of size
For more details, see the proof of Theorem 2 of [DKT15].
To cover the Gaussian component, [DKT15] grid over all parameters of the Gaussian component, arguing the effectiveness of the gridding using Proposition 2. This gridding results in the leading factor of in the size of the cover. In contrast, we use a spectral covering approach: instead of trying to grid over all parameters of the covariance matrix, we first sparsify it and then match the magnitude of its projection in every direction. In particular, we establish a cover of the following nature:
Let be the set of all Gaussians with structure preserving roundings which may arise as a consequence of Theorem 6 when applied to -Poisson multinomial random vectors with parameter . Then there exists a set of Gaussians with structure preserving roundings of size at most with the following properties:
For any , there exists a , such that and have the same block structure (i.e., the partition of coordinates), and within each block, have the same pivot coordinate and sum for the mean vector coordinates. Furthermore, for each block , letting and be the mean and covariance for the block (excluding the pivot coordinate), we have that for all unit vectors ,
;
;
where .
This lemma statement is slightly technical due to the nature of the Gaussians with structure preserving roundings. It essentially says that we cover the set of Gaussians arising from the structural theorem by matching their block structure exactly, and within each block, matching the moments spectrally. Plugging these guarantees into Lemma 2 and applying the data processing inequality for total variation distance (Lemma 1) gives the desired closeness.
For simplicity of exposition, for the remainder of this overview section, we assume that the Gaussian’s structure preserving rounding consists of a single block, an assumption we do not make in the full proof (described in Section 5.1). By the guarantees of the structural result, in this case, the minimum eigenvalue of the covariance matrix is at least some . So the goal of our exposition in this section is to produce a cover of Gaussians that may result from Theorem 6 and whose covariance matrices have minimum eigenvalue at least .
Since the mean vector only has parameters, we can grid over the entries. Though we require a spectral guarantee, this naive gridding is sufficient. This gives a set of size , such that, for any Gaussian which may arise from Theorem 6, its mean vector is approximated by a mean vector in our set with the approximation guarantees required by Lemma 2.
Covering the covariance matrix takes more care. At a high level, our approach views PMDs through the lens of spectral graph theory and exploits the existence of spectral sparsifiers. Recall the definition of the Laplacian matrix of a graph:
Given an undirected weighted graph on vertices, its Laplacian matrix is an matrix where
To see the connection to PMDs, we observe that the covariance matrix of a PMD is the Laplacian matrix of a graph defined by the parameters. For a single -CRV with parameter vector , it can be shown that the variance of is and the covariance of and is . Since , the covariance matrix is equal to the Laplacian matrix of a graph on nodes with . This can be extended to -PMDs by observing that the sum of random variables has a covariance matrix equal to the sum of the individual covariance matrices, and a similar statement holds for graphs and the corresponding Laplacian matrices. We summarize this connection in the following observation:
At the core of our approach, we use the following celebrated result of Batson, Spielman, and Srivastava [BSS12], which says that the Laplacian matrix of a graph on vertices can be spectrally approximated by the Laplacian matrix of a graph with only edges:
where is the Laplacian matrix of the graph .
Using this tool, the approach will proceed as follows. This theorem implies that, for every true covariance matrix , there exists a matrix with only entries which preserves every projection up to a multiplicative factor of . We can obtain a matrix with the same sparsity pattern as by guessing which subset of entries is non-zero, requiring guesses. Furthermore, we can grid over the non-zero entries of to ensure that it approximates every projection of up to a multiplicative factor of . Since has minimum eigenvalue and maximum entry , gridding requires only guesses, and we get that gives a multiplicative spectral approximation to . To make our approximation finer, we will -cover the set of PSD matrices within a -neighborhood of . We first recall the definition of a cover in this context:
Let S be a set of symmetric PSD matrices. An -cover of the set , denoted by , is a set of PSD matrices such that for any matrix , there exists a matrix such that for all vectors : .
Now, if we could -cover the set of all matrices -close to , we would obtain an -approximation to . We do so using the following lemma, which provides a method to generate such a cover. A slight generalization of this statement appeared as Lemma 9 in [DKT15], but we give a slightly simpler proof in Section 5.2 for completeness.
Let be a symmetric PSD matrix with minimum eigenvalue at least and let be the set of all matrices such that for all vectors , where . Then, there exists an -cover of that has size .
Combining the above, we obtain a set of covariance matrices of size such that, for any Gaussian which may arise in Theorem 6, its covariance matrix is approximated by a covariance matrix in our cover as required by Lemma 2.
Combining the guarantees obtained for the mean and the covariance matrix, we find that they satisfy both conditions of Lemma 2. Therefore, we have described a cover of size for all possible Gaussian components. The proof of Theorem 7 is completed by taking the Cartesian product of this Gaussian cover with the cover for the -PMD component.
For more details on covering the Gaussian component, see Section 5.1.
1 Details on Covering the Gaussian Component
Recall that the Gaussian component will have a structure preserving rounding. The first step in designing our cover will be to guess the partitioning into blocks. There are dimensions, resulting in at most different block structures. In what follows, we will describe how to cover a single block up to accuracy , taking the Cartesian product of the resulting sets will give an -cover of the entire Gaussian at the additional cost of in the exponent.
For a single block which consists of dimensions , we must first guess the size parameter and which dimension is to be used as the pivot. The former is an integer between and , and guessing it comes at a cost of in our cover size. Guessing the latter comes at a cost in our cover size.
Recall that our strategy will be to spectrally match the parameters of the true Gaussian. We will conclude the two distributions are close using the guarantees provided by Lemma 2. We describe how to obtain such guarantees for both the mean and covariance matrix separately.
1.2 Covering the Covariance Matrix of a Block
We will use the characterization provided by Observation 1, which tells us that the covariance matrix of an -PMD is the Laplacian matrix of a graph defined by the parameters of the distribution. Recall from the proof of Theorem 6 (which appears in [DKT15]), the covariance matrix of the Gaussian we are attempting to match is also the covariance matrix of an -Generalized Multinomial Distribution. For the remainder of this proof, we let be the graph defined by this characterization for the covariance matrix of the corresponding -Poisson Multinomial Distribution.
As a starting point, we use Theorem 8, which shows the existence of spectral sparsifiers. In particular it implies that, if given on nodes and we want a subgraph such that , there exists an with at most edges which gives this approximation. The first step in covering the covariance matrix is to guess which edges are present in the graph. Since there are possible edges in the graph, this requires at most
Further, recall that our structural result implies that , so it suffices to obtain a graph such that
For a unit vector and PSD matrices and ,
Suppose we guess the edge weights of such that they are at most away from those of . This tells us , and since the diagonal entries of are the sums of the off-diagonal entries, . This implies that it suffices to additively estimate the edge weights up to accuracy . Since the maximum entry of is at most , the spectral guarantee implies that the maximum entry of is at most , and similarly, the maximum edge weight. Therefore, gridding over all non-zero edge weights, we define a set with at most
At this point, we have a PSD matrix which, when projected onto the subspace orthogonal to , is -spectrally close to the target covariance matrix. We wish to -cover the space of all PSD matrices which are -spectrally close to this matrix. We will use Lemma 6, which we instantiate with parameter “” set to , allowing us to generate a -cover of a -neighborhood of a given PSD matrix with candidates. Since we knew one of the previous candidates was -close to the target, this gives us a matrix which satisfies the second condition of Lemma 2 to approximate this block up to accuracy. The size of this cover is at most
1.3 Putting the Guarantees Together
At this point, to cover a single block up to accuracy , we have a set of size at most
Taking the Cartesian product of sets and multiplying by the number of guesses for the block structure of the Gaussian, we get an overall cover of size
Combining with the cover for the -PMD component, we obtain an overall cover for -PMDs of size
2 Proof of Lemma 6
To construct the cover, we will make use of the eigenvalues and eigenvectors of the matrix . We first show that for any matrix , its eigenvalues are close to the eigenvalues of .
Let be two symmetric PSD matrices such that for all vectors with , for some constant . Then for the eigenvalues of , and the eigenvalues of , it holds that:
From Courant’s minimax principle, we have that the -th eigenvalue of is equal to:
where is an matrix. For the matrix , we have that
Similarly, we have that , so the result follows. ∎
By computing the eigenvalues of , we have estimates of the eigenvalues of within a multiplicative factor of . We can improve our estimates to a better multiplicative factor by gridding multiplicatively around each eigenvalue. This requires another guesses per eigenvalue. So in total, we require guesses for obtaining accurate estimates of the eigenvalues of .
Once we know (approximately) the eigenvalues of , we will try to guess also its eigenvectors . We will do this by performing a careful gridding around the eigenvectors of which we can assume, without loss of generality (by rotating), to be the standard basis vectors . So for each eigenvector of , we will try to approximate it by guessing its projections to the eigenvectors of .
We now bound the projections of eigenvectors of to eigenvectors of . Since we know that , we get that which implies that . Moreover, since , we know that the projection of to will be smaller than . An additional bound for the projection of to can be obtained by considering the variance of the matrices and in the direction . Since we know that , we get that which implies that .
We will now show that the covariance matrix satisfies the property that it is close in all directions to . To do this we will make use of the following lemma from [DKT15]. This roughly states that two PSD matrices spectrally approximate each other in particular directions, then they spectrally approximate each other in every direction.
For all , \Big{|}\Big{(}\frac{v_{i}}{\sqrt{\lambda}_{i}}\Big{)}^{T}\Big{(}\hat{\Sigma}-\Sigma\Big{)}\Big{(}\frac{v_{i}}{\sqrt{\lambda}_{i}}\Big{)}\Big{|}\leq\varepsilon,
For all , \Big{|}\Big{(}\frac{v_{i}}{\sqrt{\lambda}_{i}}+\frac{v_{j}}{\sqrt{\lambda}_{j}}\Big{)}^{T}\Big{(}\hat{\Sigma}-\Sigma\Big{)}\Big{(}\frac{v_{i}}{\sqrt{\lambda}_{i}}+\frac{v_{j}}{\sqrt{\lambda}_{j}}\Big{)}\Big{|}\leq 4\varepsilon.
We will only consider directions for and for .
We first consider direction . We have that:
The first term is in the range , which for , becomes . The rest of the terms can be bounded as follows:
for . This means that . The proof is similar for directions for .
Overall, we can get an estimate of any matrix by making at most guesses, which implies an -cover of this size.
A Proper Cover for PMDs
We show how to turn the non-proper cover of Section 5 into a proper one as described by Theorem 3, using Theorem 1. We note that a non-constructive proper cover follows immediately from Theorem 7, since for each element of an improper -cover that lies within of a PMD, we can match it with such a PMD. The resulting set of PMDs defines then a proper -cover. Our focus in this section is to provide an efficient construction of a proper cover.
Our approach will be to enumerate the improper cover of Theorem 7 and convert each distribution to a nearby -PMD. This cover consists of distributions which are the sum of a Gaussian with a structure preserving rounding and a -PMD. Since the -PMD component is already a collection of -CRVs, this part of the cover is already proper, and it suffices to convert the Gaussian component into a nearby -PMD.
The main technical lemma we prove is the following, which states that if a discretized Gaussian is spectrally close to a GMD , we can obtain a new GMD which is spectrally close to :
Let be a discretized Gaussian and suppose there exists a -GMD with mean and covariance such that for all vectors it holds that and .
Then, it is possible to compute in time a -GMD with mean and covariance such that for all vectors it holds that and .
We prove this lemma using the Shapley-Folkman lemma [Sta69], which states that the Minkowski sum of a large number of sets is approximately convex:
With this lemma in hand, the proof of Lemma 8 proceeds as follows. Let be the set of all possible mean and covariances for a single CRV, and be the Minkowski sum of copies of . Given a discretized Gaussian with mean and covariance , we would ideally like to find such that . However, since this set is not convex, this optimization problem is not obviously tractable. Instead, we convert to a spectrally close which lies on the convex hull of , which can be done using a linear program. At this point, we exploit the “almost convex” characterization provided the Shapley-Folkman lemma, and we will iteratively “peel off” plausible CRVs. More specifically, noting that the moment profile is at most dimensional and applying Lemma 9, we can use a linear program to find the parameters of a single CRV such that subtracting its moments gives a moment profile which lies on the convex hull of . We repeat times until we are left with a point on the convex hull of , at which point we may pick the last CRVs arbitrarily. The proof is completed by arguing that the resulting GMD satisfies the theorem conditions. For the full proof of Lemma 8, see Section 6.1.
We convert the same block of to a discretized Gaussian in the same way, incurring the same cost. Finally, we relate the two discretized Gaussians in total variation distance. As mentioned in the previous paragraph, the means and covariances are spectrally close up to relative accuracy and . We plug this guarantee into Lemma 2 and apply the data processing inequality (Lemma 1) to conclude that the two distributions are -close. The proof is concluded by setting and rescaling by a constant factor.
We first argue that rounding all constituent probability vectors in the -GMD so that all their coordinates are integer multiples of to obtain a -GMD approximately preserves the spectral closeness guarantees with the discretized Gaussian. More specifically, for all vectors it holds that:
We know that and thus , so
Similarly we have that which implies that for all vectors . Thus,
At this point, we have shown that there exists a -GMD with mean and covariance close to that of the discretized Gaussian such that all its constituent probability vectors have coordinates that are integer multiples of . Now, for every probability vector with probabilities that are multiples of , consider its moment profile , where and are the mean and covariance of the -CRV with probabilities . Let be the set of all possible moment profiles generated by such probability vectors . Since there are at most probability vectors the set has size at most . Moreover, it is easy to see that for the rounded GMD , it holds that where denotes the Minkowski addition of with itself times. This is because the mean and covariance of the GMD is equal to the sum of the means and covariances of its constituent CRVs, which are all in since each CRV has probabilities that are integer multiples of .
Naively searching over for a GMD that satisfies the guarantees of is not easy since it would require time that is exponential in . To get a computationally efficient algorithm, we search instead in the set where, for a set , denotes its convex closure, and the equality is a basic property of Minkowski sums. The reason this is easy is that it is solvable by a linear program as follows:
For and , we assign the variables that denote whether we want to pick the moment profile for the -th CRV.
For all , we need that . This ensures that for all , .
We need that the aggregate moment profile satisfies the closeness constraints with . For all we require that:
These are all linear constraints so a solution , can be computed by solving the linear program using the Ellipsoid method. Note that the constraints of the third bullet are infinitely many but can be verified efficiently using a separation oracle. To check the first set of constraints, we can check whether the optimization problem
has a negative solution. This is a convex optimization problem which can be solved in polynomial time. To check the second set of constraints, we note that . By setting and , we can rewrite the constraints as:
This is equivalent to checking whether the maximum eigenvalue of the matrix is greater than .
For any , it holds that . Moreover, and . This implies that and . We have that:
Similarly,
A Lower Bound for Covers of PMDs
In this section, we discuss Theorem 4, the lower bound on the size of any -cover of PMDs. This theorem shows that it is not possible to get significant improvement on the cover size obtained in Theorem 3. In particular, the dependence of the size of the cover on is tight up to a difference of in the exponent of .
It turns out that it is easy to prove a dependence of on the size of any -cover and most of the work is involved in showing a lower bound of on the cover size. Thus, in this overview we only focus on the machinery required to show the lower bound of on the -cover size. We remark that prior to our work, for (i.e. PBDs), Diakonikolas, Kane, and Stewart obtained a lower bound of [DKS16b].
We provide the proof of Theorem 4. The proof will use algebraic geometric tools to argue this fact. In particular, the main theorem we will prove will be the following:
We will now prove Theorem 4 using Theorem 9.
is a multisymmetric polynomial of degree in the variables where and . By multisymmetric, we mean the polynomial is invariant under permuting the rows of .
where the sum is taken over all surjections from where for , if , then .
The definition above might look a little involved, so to illustrate the concept, we will consider a simple example. For example, if and , then
Likewise, if and , then
Another family of polynomials which will be very useful in our reasoning will be the family of power sum multisymmetric polynomials.
As we have already observed, is a multisymmetric polynomial in entries of the matrix at degree is at most . To prove that is a linear combination of for , it suffices to make the following observation: Note that is
The next lemma implies bounds on the coefficients of in expressing . Let us now assume that . It is easy to see the following claim.
For with , we have .
The next claim is also fairly easy to prove.
.
In other words, is obtained by a pointwise product of and the indicator vector of the singleton set . Now, assume that for ,
where . Thus, to prove our claim, it suffices to show that for any particular ,
Note that is just at the position and zero everywhere else. We introduce the following notation: For any integer , we let denote the set of its partitions i.e. a tuple of strictly positive integers summing to ordered in decreasing sequence. For example, for , we have 6 distinct partitions . For any partition , we use to denote the number of summands in . For example, for the partition , . Further, if is a partition of , then
With these notations in place, it is easy to see that
Now, it is easy to see that for any integer , . If denotes the number of in the partition . With this, we have
Now, note that the total number of partitions of with ones in it is upper bounded by . However, it is a well-known fact in number theory, that . Thus,
Let and be two -PMDs such that . Then, there exists such that
where is the constant appearing in Claim 2.
Let be the constant appearing in Claim 2. Let and be the smallest integer such that there exists a with and
Note that by assumption, there exists such a . Next,
Applying Claim 1 and Claim 2, we get that
Again applying the hypothesis on , we have
The strategy for the rest of the proof is as follows: Instead of showing Theorem 9, we will show the following lemma.
To see why it suffices to prove Lemma 11, we have the following claim.
Assume towards a contradiction that . By definition, this means that there is a coupling such that the marginal is distributed as , the marginal is distributed as and . As the support of both and is confined in the box , it easily follows that \big{|}M_{\alpha}(Z_{1})-M_{\alpha}(Z_{2})\big{|}<\Pr[Z^{\prime}_{1}\not=Z^{\prime}_{2}]\cdot m^{\|\alpha\|_{1}}<\delta. This results in a contradiction, thus completing the proof. ∎
In light of Lemma 10, it instead suffices to prove the following lemma.
It is easy to see that this operation defines legitimate PMD matrices. Further, note that is a homogenous polynomial of degree . Thus, it is easy to see that if , then
For us, the utility of Jacobian will come from its role in the following theorem.
As a consequence, we have the following corollary.
To show a lower bound on the entropy of , we will apply Corollary 2. To apply this, we need to prove that . It is not clear how to show this, so we introduce an intermediate map.
The family is usually referred to as power-sum multisymmetric polynomials.
The idea here will be that we will relate the family and and then argue about the Jacobian of a map defined in terms of . The following relation between and was established in Dalbec [Dal99].
Using induction, the following lemma is immediate.
Immediately, we have that . Thus, to show a lower bound on , it suffices to show a lower bound on . We next show the following claim.
This implies that . This means that we can choose a square submatrix of size of of full rank. This means there is a subset of of size (call it ) such that
A Fourier-Based Learning Algorithm for PMDs
We believe the application of Fourier analysis to learn such structured distributions is interesting in its own right and might have application in the future towards obtaining learning algorithms for other interesting classes of distributions. In particular, the recent work on the population recovery problem [WY12, MS13, LZ15] may also be viewed as an example of use of Fourier analysis towards learning of structured distributions.
Refer to Corollary 3 for the precise bounds. Similar exponential decay of Fourier transform is also true around the other points of the form . Let us use where are the eigenvalues of . It is not difficult to show that all but an -fraction of the mass of falls on a set of size (Lemma 18). On the other hand, using the exponential decay of the Fourier transform, we have the following crucial claim: We identify a region of volume such that
Refer to Claim 7 for the precise bounds. Also, in this informal description, we use to hide the dependence on as well as the polylogarithmic factors of . This implies that if is another function such that inside and outside , then
The main idea behind learning PMDs is to look at the Fourier spectrum of PMDs. Specifically, we will prove two structural results about PMDs. One is that the Fourier spectrum of PMDs (roughly) has an exponential decay around the origin. The second result we will prove is the Fourier spectrum is a Lipschitz function and thus to estimate the Fourier spectrum in the entire domain, it suffices to compute it at a few points. Combining these two results along with standard statements on Fourier inversion show that if we construct a hypothesis distribution which approximates the Fourier spectrum of the target PMD at the chosen points and also exhbits a similar exponential decay in the Fourier spectrum, then the hypothesis distribution is close to the target PMD. While the condition on Fourier decay is not algorithmically easy to impose, we show that using some ideas from [DKT15], the problem of imposing these constraints reduces to linear programming. We will first quickly review the notion of Fourier spectrum of integer valued distributions.
Let us now recall the setting: where are independent random variables supported on . Also for and , let . To specify the next lemma, for any , we will need to define an associated vector . For any , define the associated as follows:
Let be a CRV with covariance matrix . Then, .
To prove this, we do a simple case analysis, and use the inequality for :
If both and are of the same type, then note that which gives the required inequality.
If and are type and , then note that . This implies that
Noting that gives the required inequality.
If is of type and is of type 2, then note that the maximum value that can take is . On the other hand, notice that . These two facts immediately imply that
The exact same situation holds if is of type and is of type 3.
Having shown (11), see that this implies that
For any -PMD with covariance matrix , we have that for any :
This follows simply by noticing that for a PMD ,
where is the covariance matrix of . Using the inequality, (for ), we have
We also have the following variant of the above lemma which will be useful for us.
where and are the centered random variables obtained by centering and . Likewise,
Applying the same for the and applying triangle inequality, we get the claim. ∎
We now state the Plancherel identity in this setting. In particular, we have the following easy claim (which can be found in any standard text on Fourier analysis).
The above corollary demonstrates that to learn the PMD to error , it suffices to produce another distribution whose Fourier spectrum is very close to the Fourier spectrum of (the “very small” is quantified by the effective support of ).
Given sample access to a -PMD with mean and covariance matrix , there exists an algorithm which can produce estimates and such that with probability at least for every vector :
The sample and time complexity are .
The following is guaranteed by the multidimensional Chernoff bound.
Let be a -PMD with mean and covariance matrix . Let . For ,
Let the eigenvectors of be with the corresponding eigenvalues . Consider any two distinct . Since and are distinct, hence there must be some such that the projection of and along is separated by . Let us denote the projection of along by . Then the condition of lying in implies that . It is then easy to see that if the number of integer points in is more than , then there must be points and and some , . ∎
For a point , define as
Note that can be partitioned into the regions (for ). In other words,
Let and let . Then,
We now individually bound each of the summands. Fix any particular . Using Corollary 3
The last inequality uses that . This integral is now easily seen to be bounded by
This is exactly the same bound as stated in the claim. ∎
Let . Then,
Doing the exact same calculation as in the proof of Claim 7,
2 Learning algorithm for PMDs
Theorem 6, the structure theorem from [DKT15], allows us to assume that the PMD is essentially a discretized Gaussian convolved with a sparse PMD where the sparse PMD is supported on only summands.
By setting ‘’ from the Theorem statement to be , we get that . Because our subsequent learning algorithm will take samples, we assume that we are getting samples from instead of and that . Furthermore, using the following claim from [DKT15], we can get a spectral estimate with accuracy of the mean and covariance of the Gaussian by guessing the partition of coordinates in the covariance matrix of the Gaussian and going through all elements of the spectral cover of PSD matrices around a fine estimate for obtained using samples from Lemma 17.
Let A be a symmetric PSD matrix with minimum eigenvalue 1 and let be the set of all matrices such that where and . Then, there exists a cover of size such that any is -spectrally close to some element in the cover.
The spectral closeness translates to closeness in total variation distance between Gaussians (Lemma 2) and again since we will be taking samples in the learning algorithm, we can assume that the gaussian has exactly the mean and covariance we guessed.
Similarly, we can assume that the sparse-PMD has known mean and covariance and . This is because any PMD with summands is -close in total variation to a PMD where all the probabilities are rounded to multiples of . This fact follows from union-bounding all the errors of the individual summands. Since for the sparse PMD, all coordinates are multiples of , which implies that the mean and covariance coordinates are also multiples of and we can guess them exactly using guesses. Again, since this sparse PMD is close and we will be getting much fewer samples, we can assume that the sparse PMD has exactly the mean and covariance we guessed.
At this point, we have argued the following:
The PMD is equal to the sum of a discretized Gaussian and a sparse PMD with summands. The mean and covariance of the Gaussian and of the sparse PMD are known, which implies that the mean and covariance of the overall PMD is equal to .
Our learning algorithm attempts to recover the sparse PMD in order to learn the overall distribution . However, imposing the condition that the distribution we are trying to estimate is a sparse PMD will involve solving non linear equations making the computation intractable. Rather, we will seek to learn a sparse distribution supported on where .
To learn this distribution, we will attempt to estimate its Fourier Transform. We will be mostly interested in points on the grid:
where are the eigenvector, eigenvalue pairs of the matrix . From Corollary 3, we know that the Fourier transform decays exponentially as we move away from , and in particular Claim 7 bounds the total mass contained at a distance at least from all the points. For our purposes, we set and perform the following steps to learn the sparse distribution .
Create variables for every with the constraints and .
Let . Let be the points of the grid that lie in . For each of those points, get an estimate of such that and then impose linear constraints on so that and .
Let be the eigenvalues and eigenvectors of We note that, since we may have eigenvalues which are both large and small in magnitude, a naive eigendecomposition algorithm would incur a cost which depends on . However, as we only require the eigenvalues and eigenvectors approximately, this cost can be avoided by applying an appropriate power-iteration method. The cost in terms of and is dominated by the other steps in our algorithm. and consider the set:
Construct a grid of points in with a spacing of in every direction. Let be the subset of these points which fall in . For all these points impose the conditionss that and that follow from Corollary 3.
Finally, add the constraints and
Note that in Step 2, has size at most . If we naively estimated every Fourier coefficient in the number of samples would be too high because every Fourier coefficient requires samples to learn with accuracy and probability of failure . However, we can instead take samples and reuse the same samples to compute all the required Fourier coefficients. Since the probability of error is very small a simple union bound among all of the coefficients, shows that with at least constant probability all of them can be estimated within .
To complete the learning algorithm, we repeat the steps above for each of the guessed mean and covariance matrices . We then perform a hypothesis selection algorithm to choose a distribution within from each of the distributions we obtain. We made guesses, and thus obtained candidate hypotheses. Applying the following tournament theorem for hypothesis selection from [DK14], we can select a good estimate in samples in runtime.
We first show that there is a solution to which satisfies all the constraints. Indeed, if we set the sparse distribution to be equal to the distribution of the sparse PMD we defined above, we get:
since is a probability distribution supported on .
The constraint is satisfied since for ,
The derivation for the constraint on the imaginary part is identical.
From Corollary 3, the sparse PMD satisfies everywhere in . This condition implies the imposed constraints which are only evaluated in few points.
The distribution has mean and covariance , so the last constraint is satisfied.
Consider any point in . Then, note that there is some such that for , . Applying Lemma 16, we get that
We bound the volume of the set . To do this, we again apply Claim 8, and get that
Note that for any point , we there is a point such that and that . Since the variance of is at most in every direction, we get that
By applying Claim 8 to bound the volume of the set and using the fact that is at most , we get that the first integral is at most
The last inequality uses the fact that whenever , it must imply that all the variance comes from and thus . By plugging the value of , we get that
The calculation for the second integral is similar.
Here the first inequality follows by exactly the same calculation we did for the first integral whereas the second inequality uses that . Now, that we had derived that
However, (because the variance of is at most in any direction. This implies that
Note that . Thus, . Applying Claim 7 and noting that
Again using the fact that the variance of in any direction is at most ,
Plugging in the value of , we get that
Combining Claim 11, Claim 12 and Claim 13, we get that
This is at most . Setting to be , we complete the proof of Theorem 5.
Open Problems
A number of interesting questions regarding Poisson Multinomial distributions are left open by this work and [DKS16a]. We outline a few of them here.
The complexity of learning Poisson Multinomials. This work and [DKS16a] both give algorithms for learning PMDs. The sample and time complexities are polynomial in and exponential in . Meanwhile, [DKT15] gives an algorithm with a sample complexity polynomial in both parameters, but the time complexity is exponential in and . Is there an algorithm for learning PMDs with sample and time complexities both polynomial in and ?
Exploring the connection between Poisson Multinomials and Laplacian matrices. In this work, we described a cover for the set of -PMDs of size . Our construction relied crucially on Observation 1 (which states that the covariance matrix of a PMD is Laplacian) and spectral sparsification results for Laplacian matrices. With this connection in hand, can one derive other results for PMDs using the wealth of literature on Laplacian matrices?
A tighter central limit theorem. [VV11] proves a central limit theorem between an -GMD and a discretized Gaussian with the same mean and covariance, upper bounding their total variation distance by , where is the smallest eigenvector of the covariance matrix of the GMD. Both this paper and [DKS16a] qualitatively improve this bound by removing the dependence on , while keeping the dependence on and still polynomial. How well can a GMD be approximated by a discretized Gaussian? In one dimension, the answer is [CGS10], which implies a the answer for multiple dimensions is at least . [DKS16a] achieves this dependence on (up to log factors), but the optimal dependence on is currently unknown.
References
Appendix A Proof of Lemma 2
Without loss of generality, assume that and are full rank. If not, the guarantees in the statement ensure that their nullspace is identical, and we can project to a lower dimension such that the resulting matrices are full rank.
First, we note that the assumptions in the lemma statement can be converted to be in terms of the minimum of the two variances, instead of the maximum. Define . The second assumption can be rearranged to see that . Plugging this back into the second assumption gives that
where the last inequality holds for (otherwise, the lemma’s conclusion is trivial). Similarly, the second assumption also implies , when plugged into the first assumption gives
For the remainder of the proof, we will use these guarantees instead of the ones in the lemma statement.
We recall the standard formula for KL-divergence between two Gaussian distributions. Let be the eigenvalues of .
We bound the divergence induced by differences in the means and covariances separately. We start with the means. Note that
Now we bound the divergence induced by differences in the covariances. We bound the eigenvalues of . Note that
Substituting makes the latter condition equivalent to
The Courant-Fischer Theorem implies that for all .
At this point, we note that for all . This implies