我们证明样本对于学习￼中的k高斯混合是必要且充分的，直到整体偏差距离误差ε。这改善了该问题已知的上限和下限。对于轴对齐高斯分布的混合，我们证明样本足够，匹配已知的下界。上限是基于一种基于样本压缩概念的分布式学习新技术。任何一类允许这种样本压缩方案的分布也可以通过很少的样本来学习。我们的主要结果是证明了￼中的高斯类具有有效的样本压缩。
- 1071
- 0
- 1
注脚
展开查看详情
1. Nearly tight sample complexity bounds for learning mixtures of Gaussians via sample compression schemes∗ Hassan Ashtiani Shai Ben-David Department of Computing and Software School of Computer Science McMaster University, and University of Waterloo, Vector Institute, ON, Canada Waterloo, ON, Canada zokaeiam@mcmaster.ca shai@uwaterloo.ca Nicholas J. A. Harvey Christopher Liaw Department of Computer Science Department of Computer Science University of British Columbia University of British Columbia Vancouver, BC, Canada Vancouver, BC, Canada nickhar@cs.ubc.ca cvliaw@cs.ubc.ca Abbas Mehrabian Yaniv Plan School of Computer Science Department of Mathematics McGill University University of British Columbia Montréal, QC, Canada Vancouver, BC, Canada abbasmehrabian@gmail.com yaniv@math.ubc.ca Abstract We prove that Θ(kd2 /ε2 ) samples are necessary and sufficient for learning a mixture of k Gaussians in Rd , up to error ε in total variation distance. This improves both the known upper bounds and lower bounds for this problem. For mixtures of axis-aligned Gaussians, we show that O(kd/ε2 ) samples suffice, matching a known lower bound. The upper bound is based on a novel technique for distribution learning based on a notion of sample compression. Any class of distributions that allows such a sample compression scheme can also be learned with few samples. Moreover, if a class of distributions has such a compression scheme, then so do the classes of products and mixtures of those distributions. The core of our main result is showing that the class of Gaussians in Rd has an efficient sample compression. 1 Introduction Estimating distributions from observed data is a fundamental task in statistics that has been studied for over a century. This task frequently arises in applied machine learning and it is common to assume that the distribution can be modeled using a mixture of Gaussians. Popular software packages have implemented heuristics, such as the EM algorithm, for learning a mixture of Gaussians. The theoretical machine learning community also has a rich literature on distribution learning; the recent survey [9] considers learning of structured distributions, and the survey [13] focuses on mixtures of Gaussians. ∗ For the full version of this paper see [2]. 32nd Conference on Neural Information Processing Systems (NeurIPS 2018), Montréal, Canada.
2.This paper develops a general technique for distribution learning, then employs this technique in the important setting of mixtures of Gaussians. The theoretical model we adopt is density estimation: given i.i.d. samples from an unknown target distribution, find a distribution that is close to the target distribution in total variation (TV) distance. Our focus is on sample complexity bounds: using as few samples as possible to obtain a good estimate of the target distribution. For background on this model see, e.g., [7, Chapter 5] and [9]. Our new technique for proving upper bounds on the sample complexity involves a form of sample compression. If it is possible to “encode” members of a class of distributions using a carefully chosen subset of the samples, then this yields an upper bound on the sample complexity of distribution learning for that class. In particular, by constructing compression schemes for mixtures of axis- aligned Gaussians and general Gaussians, we obtain new upper bounds on the sample complexity of learning with respect to these classes, which we prove to be optimal up to logarithmic factors. 1.1 Main results In this section, all learning results refer to the problem of producing a distribution within total variation distance ε from the target distribution. Our first main result is an upper bound for learning mixtures of multivariate Gaussians. This bound is tight up to logarithmic factors. Theorem 1.1 The class of k-mixtures of d-dimensional Gaussians can be learned using O(kd2 /ε2 ) samples. We emphasize that the O(·) notation hides a factor polylog(kd/ε), but has no dependence whatsoever on the condition number or scaling of the distribution. Previously, the best known upper bounds on the sample complexity of this problem were O(kd2 /ε4 ), due to [3], and O(k 4 d4 /ε2 ), based on a VC-dimension bound that we discuss below. For the case of a single Gaussian (i.e., k = 1), a sample complexity bound of O(d2 /ε2 ) is well known, again using a VC-dimension bound discussed below. Our second main result is a minimax lower bound matching Theorem 1.1 up to logarithmic factors. Theorem 1.2 Any method for learning the class of k-mixtures of d-dimensional Gaussians has sample complexity Ω(kd2 /ε2 log3 (1/ε)) = Ω(kd2 /ε2 ). Here and below Ω (and O) allow for poly-logarithmic factors. Previously, the best known lower bound on the sample complexity was Ω(kd/ε2 ) [20]. Even for a single Gaussian (i.e., k = 1), an Ω(d2 /ε2 ) lower bound was not known prior to this work. Our third main result is an upper bound for learning mixtures of axis-aligned Gaussians, i.e., Gaussians with diagonal covariance matrices. This bound is tight up to logarithmic factors. Theorem 1.3 The class of k-mixtures of axis-aligned d-dimensional Gaussians can be learned using O(kd/ε2 ) samples. A matching lower bound of Ω(kd/ε2 ) was proved in [20]. Previously, the best known upper bounds were O(kd/ε4 ), due to [3], and O((k 4 d2 + k 3 d3 )/ε2 ), based on a VC-dimension bound that we discuss below. Computational efficiency. Although our approach for proving sample complexity upper bounds is algorithmic, our focus is not on computational efficiency. The resulting algorithms are efficient in terms of number of needed samples, but their running times are exponential in the dimension d 2 and the number of mixture components k. More precisely, the running time is 2kd polylog(d,k,1/ε) for mixtures of general Gaussians, and 2kd polylog(d,k,1/ε) for mixtures of axis-aligned Gaussians. The existence of a polynomial time algorithm for density estimation is unknown even for the class of mixtures of axis-aligned Gaussians, see [10, Question 1.1]. Even for the case of a single Gaussian, the published proofs of the O(d2 /ε2 ) bound (of which we are aware) are not algorithmically efficient. Using ideas from our proof of Theorem 1.1, we can show that an algorithmically efficient proof for single Gaussians can be obtained by computing the empirical mean and a careful modification of the sample covariance matrix of O(d2 /ε2 ) samples. 2
3.1.2 Related work Distribution learning is a vast topic and many approaches have been considered in the literature; here we only review approaches that are most relevant to our problem. For parametric families of distributions, a common approach is to use the samples to estimate the parameters of the distribution, possibly in a maximum likelihood sense, or possibly aiming to approximate the true parameters. For the specific case of mixtures of Gaussians, there is a substantial theoretical literature on algorithms that approximate the mixing weights, means and covariances; see [13] for a recent survey of this literature. The strictness of this objective cuts both ways. On the one hand, a successful learner uncovers substantial structure of the target distribution. On the other hand, this objective is clearly impossible when the means and covariances are extremely close. Thus, algorithms for parameter estimation of mixtures necessarily require some “separability” assumptions on the target parameters. Density estimation has a long history in the statistics literature, where the focus is on the sample complexity question; see [6, 7, 19] for general background. It was first studied in the computational learning theory community under the name PAC learning of distributions by [14], whose focus is on the computational complexity of the learning problem. For density estimation there are various possible measures of distance between distributions, the most popular ones being the TV distance and the Kullback-Leibler (KL) divergence. Here we focus on the TV distance since it has several appealing properties, such as being a metric and having a natural probabilistic interpretation. In contrast, KL divergence is not even symmetric and can be unbounded even for intuitively close distributions. For a detailed discussion on why TV is a natural choice, see [7, Chapter 5]. A popular method for distribution learning in practice is kernel density estimation (see, e.g., [7, Chapter 9]). The few rigorously proven sample complexity bounds for this method require either smoothness assumptions (e.g., [7, Theorem 9.5]) or boundedness assumptions (e.g., [12, Theo- rem 2.2]) on the class of densities. The class of Gaussians is not universally Lipschitz or universally bounded, so those results do not apply to the problems we consider. Another elementary method for density estimation is using histogram estimators (see, e.g., [7, Section 10.3]). Straightforward calculations show that histogram estimators for mixtures of Gaussians result in a sample complexity that is exponential in the dimension. The same is true for estimators based on piecewise polynomials. The minimum distance estimate [7, Section 6.8] is another approach for deriving sample complexity upper bounds for distribution learning. This approach is based on uniform convergence theory. In particular, an upper bound for any class of distributions can be achieved by bounding the VC- dimension of an associated set system, called the Yatracos class (see [7, page 58] for the definition). For example, [11] used this approach to bound the sample complexity of learning high-dimensional log-concave distributions. For the class of single Gaussians in d dimensions, this approach leads to the optimal sample complexity upper bound of O(d2 /ε2 ). However, for mixtures of Gaussians and axis-aligned Gaussians in Rd , the best known VC-dimension bounds [1, Theorem 8.14] result in loose upper bounds of O(k 4 d4 /ε2 ) and O((k 4 d2 + k 3 d3 )/ε2 ), respectively. Another approach is to first approximate the mixture class using a more manageable class such as piecewise polynomials, and then study the associated Yatracos class, see, e.g., [5]. However, piecewise polynomials do a poor job in approximating d-dimensional Gaussians, resulting in an exponential dependence on d. For density estimation of mixtures of Gaussians, the current best sample complexity upper bounds (in terms of k and d) are O(kd2 /ε4 ) for general Gaussians and O(kd/ε4 ) for axis-aligned Gaussians, both due to [3]. For the general Gaussian case, their method takes an i.i.d. sample of size O(kd2 /ε2 ) 2 2 and partitions this sample in every possible way into k subsets. Based on those partitions, k O(kd /ε ) “candidate distributions” are generated. The problem is then reduced to learning with respect to that finite class of candidates. Their sample complexity has a suboptimal factor of 1/ε4 , of which 1/ε2 arises in their approach for choosing the best candidate, and another factor 1/ε2 is due to the exponent in the number of candidates. 3
4.Our approach via compression schemes also ultimately reduces the problem to learning with respect to finite classes. However, our compression technique leads to a more refined bound. In the case of mixtures of Gaussians, one factor of 1/ε2 is again incurred due to learning with respect to finite classes. The key is that the number of compressed samples has no additional factor of 1/ε2 , so the overall sample complexity bound has only a O(1/ε2 ) dependence on ε. As for lower bounds on the sample complexity, much fewer results are known for learning mixtures of Gaussians. The only lower bound of which we are aware is due to [20], which shows a bound of Ω(kd/ε2 ) for learning mixtures of spherical Gaussians (and hence for general Gaussians as well). This bound is tight for the axis-aligned case, as we show in Theorem 1.3, but loose in the general case, as we show in Theorem 1.2. 1.2.1 Comparison to parameter estimation In this section we observe that neither our upper bound (Theorem 1.1) nor our lower bound (Theo- rem 1.2) can directly follow from bounds on parameter estimation for Gaussian distributions. Recall that our sample complexity upper bound in Theorem 1.1 has no dependence on the condition number of the distribution. We now show that, if a learning algorithm with entrywise approximation guarantee is used to learn the distribution in KL divergence or TV distance, then the approximation parameter must depend on the condition number. Let κ(Σ) be the condition number of the covariance matrix Σ, i.e., the ratio of the maximum and minimum eigenvalues; refer to Section 2 for other relevant definitions. 2 Proposition 1.4 Set ε = κ(Σ)+1 ˆ that are good . There exist two covariance matrices Σ and Σ entrywise approximations: ˆ i,j | ≤ ε |Σi,j − Σ and ˆ i,j ∈ [1, 1 + 2ε] · Σi,j Σ ∀i, j, but the corresponding distributions are as far as they can get, i.e., KL N (0, Σ) ˆ =∞ N (0, Σ) and ˆ = 1. TV N (0, Σ), N (0, Σ) Thus, given a black-box algorithm that provides an entrywise approximation to the true covariance 2 matrix Σ, if ε ≥ κ(Σ)+1 ˆ which does not approximate Σ in KL divergence or total , it might output Σ, variation distance. One might imagine that lower bounds on the sample complexity of parameter estimation readily imply lower bounds on distribution learning. The following proposition shows this is not the case. Proposition 1.5 For any ε ∈ (0, 1/2] there exist two covariance matrices Σ and Σ ˆ such that ˆ ≤ ε, but there exist i, j such that, for any c ≥ 1, Σ TV N (0, Σ), N (0, Σ) ˆ i,j ∈ [1/c, c] · Σi,j . 1.3 Our techniques We introduce a novel method for learning distributions via a form of sample compression. Given a class of distributions, suppose there is a method for “compressing” the samples generated by any distribution in the class. Further, suppose there exists a fixed decoder for the class, such that given the compressed set of samples and a sequence of bits, it approximately recovers the original distribution. In this case, if the size of the compressed set and the number of bits is guaranteed to be small, we show that the sample complexity of learning that class is small as well. More precisely, we say a class of distributions admits (τ, t, m) compression if there exists a decoder function such that upon generating m i.i.d. samples from any distribution in the class, we are guaranteed, with reasonable probability, to have a subset of size at most τ of that sample, and a sequence of at most t bits, on which the decoder outputs an approximation to the original distribution. Note that τ, t, and m can be functions of ε, the accuracy parameter. We prove that compression implies learning. In particular, if a class admits (τ, t, m) compression, then the sample complexity of learning with respect to this class is bounded by O(m + (τ + t)/ε2 ) (Theorem 3.5). 4
5.An attractive property of compression is that it enjoys two closure properties. Specifically, if a base class admits compression, then the class of mixtures of that base class, as well as the class of products of the base class, are compressible (Lemmas 3.6 and 3.7). Consequently, it suffices to provide a compression scheme for the class of single Gaussian distributions in order to obtain a compression scheme for classes of mixtures of Gaussians (and therefore, to be able to bound their sample complexity). We prove that the class of d-dimensional Gaussian distributions admits (O(d), O(d2 ), O(d)) compression (Lemma 4.1). The high level idea is that by generating O(d) samples from a Gaussian, one can get some rough sketch of the geometry of the Gaussian. In particular, the convex hull of the points drawn from the Gaussian enclose an ellipsoid centered at the mean and whose principal axes are the eigenvectors of the covariance matrix. Using ideas from convex geometry and random matrix theory, we show one can in fact encode the center of the ellipsoid and the principal axes using a linear combination of these samples. Then we discretize the coefficients and obtain an approximate encoding. The above results together imply tight (up to logarithmic factors) upper bounds of O(kd2 /ε2 ) for mixtures of k Gaussians, and O(kd/ε2 ) for mixtures of k axis-aligned Gaussians over Rd . The compression framework we introduce is quite flexible, and can be used to prove sample complexity upper bounds for other distribution classes as well. This is left for future work. In this paper we assume the target belongs to the known class of distributions (this is called the realizable setting in the learning theory literature). In the full version of this paper [2] we relax this requirement and give similar sample complexity bounds for the setting where the target is close (in TV distance) to some distribution in the class (known as agnostic learning). Lower bound. For proving our lower bound for mixtures of Gaussians, we first prove a lower bound of Ω(d2 /ε2 ) for learning a single Gaussian. Although the approach is quite intuitive, the details are intricate and much care is required to make a formal proof. The main step is to construct a 2 large family (of size 2Ω(d ) ) of covariance matrices such that the associated Gaussian distributions are well-separated in terms of their TV distance while simultaneously ensuring that their relative KL divergences are small. Once this is established, we can then apply a generalized version of Fano’s inequality to complete the proof. 2 To construct this family of covariance matrices, we sample 2Ω(d ) matrices from the following probabilistic process: start with an identity covariance matrix; then choose a random subspace of dimension d/9 and slightly increase the eigenvalues corresponding to this eigenspace. It is easy to bound the KL divergences between the constructed Gaussians. To lower bound the total variation, we show that for every pair of these distributions, there is some subspace for which a vector drawn from one Gaussian will have slightly larger projection than a vector drawn from the other Gaussian. Quantifying this gap will then give us the desired lower bound on the total variation distance. Paper outline. We set up our formal framework and notations in Section 2. In Section 3, we define compression schemes for distributions, prove their closure properties, and show their connection with density estimation. Theorem 1.1 and Theorem 1.3 are proved in Section 4. The proof of Theorem 1.2 as well as all other omitted proofs can be found in the full version [2]. 2 Preliminaries A distribution learning method or density estimation method is an algorithm that takes as input a sequence of i.i.d. samples generated from a distribution g, and outputs (a description of) a distribution gˆ as an estimation for g. We work with continuous distributions in this paper, and so we identify a probability distribution by its probability density function. Let f1 and f2 be two probability distributions defined over Rd . Their total variation (TV) distance is defined by 1 TV (f1 , f2 ) := sup (f1 (x) − f2 (x))dx = f1 − f2 1 , B⊆Rd B 2 5
6.where f 1 := Rd |f (x)|dx is the L1 norm of f . The Kullback-Leibler (KL) divergence between f1 and f2 is defined by f1 (x) KL (f1 f2 ) := f1 (x) log dx. Rd f2 (x) In the following definitions, F is a class of probability distributions, and g is a distribution (not necessarily in F). Definition 2.1 (ε-approximation) A distribution gˆ is an ε-approximation for g if gˆ − g 1 ≤ ε. Definition 2.2 (PAC-learning distributions) A distribution learning method is called a (realizable) PAC-learner for F with sample complexity mF (ε, δ) if, for all distributions g ∈ F and all ε, δ ∈ (0, 1), given ε, δ, and an i.i.d. sample of size mF (ε, δ) from g, with probability at least 1 − δ (over the samples) the method outputs an ε-approximation of g. Let ∆n := { (w1 , . . . , wn ) : wi ≥ 0, wi = 1 } denote the n-dimensional simplex. Definition 2.3 (k-mix(F)) Let F be a class of probability distributions. Then the class of k-mixtures of F, written k-mix(F), is defined as k k-mix(F) := i=1 wi fi : (w1 , . . . , wk ) ∈ ∆k , f1 , . . . , fk ∈ F . Let d denote the dimension. A Gaussian distribution with mean µ ∈ Rd and covariance matrix Σ ∈ Rd×d is denoted by N (µ, Σ). If Σ is a diagonal matrix, then N (µ, Σ) is called an axis-aligned Gaussian. For a distribution g, we write X ∼ g to mean X is a random variable with distribution g, and we write S ∼ g m to mean that S is an i.i.d. sample of size m generated from g. Definition 2.4 Let A, B be symmetric, positive definite matrices of the same size. The log-det divergence of A and B is defined as LD (A, B) := Tr(B −1 A − I) − log det(B −1 A). We will use v or v 2 to denote the Euclidean norm of a vector v, A or A 2 to denote the operator norm of a matrix A, and A F := Tr(AT A) to denote the Frobenius norm of a matrix A. For x ∈ R, we will write (x)+ := max{0, x}. All logarithms are in the natural base. 3 Compression schemes and their connection with learning Let F be a class of distributions over a domain Z. Definition 3.1 (distribution decoder) A distribution decoder for F is a deterministic function J : ∞ n ∞ n n=0 Z × n=0 {0, 1} → F, which takes a finite sequence of elements of Z and a finite sequence of bits, and outputs a member of F. Definition 3.2 (distribution compression schemes) Let τ, t, m : (0, 1) → Z≥0 be functions. We say F admits (τ, t, m) compression if there exists a decoder J for F such that for any distribution g ∈ F, the following holds: For any ε ∈ (0, 1), if a sample S is drawn from g m(ε) , then with probability at least 2/3, there exists a sequence L of at most τ (ε) elements of S, and a sequence B of at most t(ε) bits, such that J (L, B) − g 1 ≤ ε. Note that S and L are sequences rather than sets; in particular, they can contain repetitions. Also note that in this definition, m(ε) is a lower bound on the number of samples needed, whereas τ (ε), t(ε) are upper bounds on the size of compression and the number of bits. Essentially, the definition asserts that with reasonable probability, there is a (short) sequence consisting of elements S and some (small number of) additional bits, from which g can be approximately reconstructed. We say that the distribution g is “encoded” with L and B, and in general we would like to have a compression scheme of a small size. Remark 3.3 In the definition above we required the probability of existence of L and B to be at least 2/3, but one can boost this probability to 1 − δ by generating a sample of size m(ε) log(1/δ). 6
7.Next we show that if a class of distributions can be compressed, then it can be learned; thus we build the connection between compression and learning. We will need the following useful result about PAC-learning of finite classes of distributions, which immediately follows from [7, Theorem 6.3] and a standard Chernoff bound. It states that a finite class of size M can be learned using O(log(M/δ)/ε2 ) samples. Denote by [M ] the set {1, 2, ..., M }. Throughout the paper, a/bc always means a/(bc). Theorem 3.4 ([7]) There exists a deterministic algorithm that, given candidate distributions f1 , . . . , fM , a parameter ε > 0, and log(3M 2 /δ)/2ε2 i.i.d. samples from an unknown distribu- tion g, outputs an index j ∈ [M ] such that fj − g 1 ≤ 3 min fi − g 1 + 4ε, i∈[M ] with probability at least 1 − δ/3. The proof of the following theorem appears in the full version [2]. Theorem 3.5 (compressibility implies learnability) Suppose F admits (τ, t, m) compression. Let τ (ε) := τ (ε/6) + t(ε/6). Then F can be learned using ε 1 τ (ε) log(m( 6ε ) log(1/δ)) + log(1/δ) ε τ (ε) O m log + =O m + 2 samples. 6 δ ε2 6 ε We next prove two closure properties of compression schemes. First, Lemma 3.6 below states that if a class F of distributions can be compressed, then the class of distributions that are formed by taking products of members of F can also be compressed. If p1 , . . . , pd are distributions over domains d d Z1 , . . . , Zd , then i=1 pi denotes the standard product distribution over i=1 Zi . For a class F of d distributions, define F d := i=1 pi : p1 , . . . , pd ∈ F . Lemma 3.6 (compressing product distributions) If F admits (τ (ε), t(ε), m(ε)) compression, then F d admits (dτ (ε/d), dt(ε/d), m(ε/d) log(3d)) compression. Our next lemma states that if a class F of distributions can be compressed, then the class of distributions that are formed by taking mixtures of members of F can also be compressed. Lemma 3.7 (compressing mixtures) If F admits (τ (ε), t(ε), m(ε)) compression, then k-mix(F) admits (kτ (ε/3), kt(ε/3) + k log2 (4k/ε)), 48m(ε/3)k log(6k)/ε) compression. 4 Upper bound: learning mixtures of Gaussians by compression schemes In this section we prove an upper bound of O(kd2 /ε2 ) for the sample complexity of learning mixtures of k Gaussians in d dimensions, and an upper bound of O(kd/ε2 ) for the sample complexity of learning mixtures of k axis-aligned Gaussians. The heart of the proof is to show that Gaussians have compression schemes in any dimension. Lemma 4.1 For any positive integer d, the class of d-dimensional Gaussians admits an O(d log(2d)), O(d2 log(2d) log(d/ε)), O(d log(2d)) compression scheme. Remark 4.2 In the special case d = 1, there also exists a (2, 0, O(1/ε)) (i.e. constant size) com- pression scheme: if we draw C/ε samples from N (µ, σ 2 ), for a sufficiently large constant C, with probability at least 2/3 there exist two points in the sample such that one of them is within distance σε/2 of µ − σ and the other one is within distance σε/2 of µ + σ. Given these two points, the decoder can estimate µ and σ up to additive precision εσ/2, which results in an ε-approximation of N (µ, σ 2 ) by Lemma ??. Remarkably, this compression scheme has constant size, as the value of τ + t is independent of ε (unlike Lemma 4.1). This scheme could be used instead of Lemma 4.1 in the proof of Theorem 1.3, although it would not improve the sample complexity bound asymptotically. Proof of Theorem 1.1. Combining Lemma 4.1 and Lemma 3.7 implies that the class of k-mixtures of d-dimensional Gaussians admits a O(kd log(2d)), O(kd2 log(2d) log(d/ε) + k log(k/ε)), O(dk log k log(2d)/ε) 7
8.compression scheme. Applying Theorem 3.5 with m(ε) = O(dk/ε) and τ (ε) = O(d2 k) shows that the sample complexity of learning this class is O(kd2 /ε2 ). This proves Theorem 1.1. Proof of Theorem 1.3. Let G denote the class of 1-dimensional Gaussian distribu- tions. By Lemma 4.1, G admits an (O(1), O(log(1/ε)), O(1)) compression scheme. Com- bining Lemma 3.6 and Lemma 3.7 gives the class k-mix(G d ) admits (O(kd), O(kd log(d/ε) + k log(k/ε)), O(k log(k) log(3d)/ε)) compression. Applying Theorem 3.5 implies that the class of k-mixtures of axis-aligned Gaussians in Rd can be learned using O(kd/ε2 ) samples. 4.1 Proof of Lemma 4.1 Let N (µ, Σ) denote the target distribution, which we are to encode. Remark 4.3 The case of rank-deficient Σ can easily be reduced to the case of full-rank Σ. If the rank of Σ is r < d, then any X ∼ N (µ, Σ) lies in some affine subspace S of dimension r. With high probability, the first d samples from N (µ, Σ) uniquely identify S. We encode S using these samples, and for the rest of the process we work in this affine subspace. Hence, we may assume Σ has full rank d. To prove Lemma 4.1, we will need the following result from the random matrix theory literature [cf. 16, Corollary 4.1]. Let S d−1 := y ∈ Rd : y = 1 and B2d := y ∈ Rd : y ≤ 1 . We use 1 the notation 20 B2d to denote the set of d-dimensional vectors with Euclidean norm at most 1/20. The convex hull of a set T is denoted by conv(T ). √ Lemma 4.4 Let q1 , . . . , qm be i.i.d. samples from N (0, Id ), and let T := { ±qi : qi ≤ 4 d }. Then for a large enough constant C > 0, if m ≥ Cd(1 + log d) then 1 d Pr B ⊆ conv(T ) ≤ 1/6. 20 2 Note that the lemma can be improved to require only m ≥ Cd samples [see 16, Corollary 4.1], but this would not improve our final bound. The remainder of the proof amounts to showing that with only a small number of additional bits, we can approximate the mean and each eigenvector of the covariance matrix as a linear combination of a subset of the drawn samples. d d Suppose Σ = i=1 vi viT , where the vi vectors are orthogonal. Let Ψ := i=1 vi viT / vi . Note that both Σ and Ψ are positive definite, and that Σ = Ψ . Moreover, it is easy to see that Σ−1 = 2 d d T i=1 vi vi / vi 4 and Ψ−1 = i=1 vi viT / vi 3 . Lemma 4.5 Let C > 0 be a sufficiently large constant. Given m = 2Cd(1 + log d) samples S from N (µ, Σ), with probability at least 2/3, one can encode vectors v1 , . . . , vd , µ ∈ Rd satisfying Ψ−1 (vj − vj ) ≤ ε/24d2 ∀j ∈ [d], and Ψ−1 (µ − µ) ≤ ε/2, using O(d2 log(2d) log(d/ε)) bits and the points in S. Lemma 4.1 now follows immediately from the following lemma Lemma 4.6 Suppose Σ = Ψ2 = i∈[d] vi viT , where the vi are orthogonal and Σ is full rank, and that Ψ−1 (µ − µ) ≤ ζ, and that Ψ−1 (vj − vj ) ≤ ρ ≤ 1 holds for all j ∈ [d]. Then, TV N µ, vi viT , N µ, vi viT ≤ 9d3 ρ2 + ζ 2 /2. i∈[d] i∈[d] 5 Discussion A central open problem in distribution learning and density estimation is characterizing the sample complexity of learning a distribution class. An insight from supervised learning theory is that 8
9.the sample complexity of learning a class (of concepts, functions, or distributions) is typically proportional to (some notion of) intrinsic dimension of the class divided by ε2 , where ε is the error tolerance. For the case of agnostic binary classification, the intrinsic dimension is captured by the VC-dimension of the concept class (see [21, 4]). For the case of distribution learning with respect to ‘natural’ parametric classes, we expect this dimension to be equal to the number of parameters. This is indeed true for the class of Gaussians (which have d2 parameters) and axis-aligned Gaussians (which have d parameters), and we showed in this paper that it holds for their mixtures as well (which have kd2 and kd parameters, respectively). In binary classification, the combinatorial notion of Littlestone-Warmuth compression has been shown to be sufficient [15] and necessary [18] for learning. In this work, we showed that the new but related notion of distribution compression is sufficient for distribution learning. Whether the existence of compression schemes is necessary for learning an arbitrary class of distributions remains an intriguing open problem. It is worth mentioning that while it may first seem that the VC-dimension of the Yatracos set associated with a class of distributions can characterize its sample complexity, it is not hard to come up with examples where this VC-dimension is infinite while the class can be learned with finite samples. Covering numbers do not characterize the sample complexity either: for instance the class of Gaussians does not have a finite covering number in the TV metric, nevertheless it is learnable with finitely many samples. A concept related to compression is that of core-sets. In a sense, core-sets can be viewed as a special case of compression, where the decoder is required to be the empirical error minimizer. See [17] for using core-sets in maximum likelihood estimation. Acknowledgments We thank Yaoliang Yu for pointing out a mistake in an earlier version of this paper, and Luc Devroye for fruitful discussions. Abbas Mehrabian was supported by a CRM-ISM postdoctoral fellowship and an IVADO-Apogée-CFREF postdoctoral fellowship. Nicholas Harvey was supported by an NSERC Discovery Grant. Christopher Liaw was supported by an NSERC graduate award. Yaniv Plan was supported by NSERC grant 22R23068. Addendum The lower bound of Theorem 1.2 was recently improved in a subsequent work [8] from Ω(kd2 /ε2 log3 (1/ε)) to Ω(kd2 /ε2 log(1/ε)) using a different construction. References [1] Martin Anthony and Peter Bartlett. Neural network learning: theoretical foundations. Cam- bridge University Press, 1999. [2] Hassan Ashtiani, Shai Ben-David, Nicholas J. A. Harvey, Christopher Liaw, Abbas Mehrabian, and Yaniv Plan. Near-optimal sample complexity bounds for robust learning of Gaussians mixtures via compression schemes. arXiv preprint. URL https://arxiv.org/abs/1710. 05209. [3] Hassan Ashtiani, Shai Ben-David, and Abbas Mehrabian. Sample-efficient learning of mixtures. In Proceedings of the Thirty-Second AAAI Conference on Artificial Intelligence, AAAI’18, pages 2679–2686. AAAI Publications, 2018. URL https://arxiv.org/abs/1706.01596. [4] Anselm Blumer, Andrzej Ehrenfeucht, David Haussler, and Manfred K. Warmuth. Learnability and the Vapnik-Chervonenkis dimension. J. ACM, 36(4):929–965, October 1989. ISSN 0004- 5411. doi: 10.1145/76359.76371. URL http://doi.acm.org/10.1145/76359.76371. [5] Siu-On Chan, Ilias Diakonikolas, Rocco A. Servedio, and Xiaorui Sun. Efficient density estimation via piecewise polynomial approximation. In Proceedings of the Forty-sixth Annual ACM Symposium on Theory of Computing, STOC ’14, pages 604–613, New York, NY, USA, 2014. ACM. ISBN 978-1-4503-2710-7. doi: 10.1145/2591796.2591848. 9
10. [6] Luc Devroye. A course in density estimation, volume 14 of Progress in Probability and Statistics. Birkhäuser Boston, Inc., Boston, MA, 1987. ISBN 0-8176-3365-0. [7] Luc Devroye and Gábor Lugosi. Combinatorial methods in density estimation. Springer Series in Statistics. Springer-Verlag, New York, 2001. ISBN 0-387-95117-2. doi: 10.1007/ 978-1-4613-0125-7. [8] Luc Devroye, Abbas Mehrabian, and Tommy Reddad. The minimax learning rate of normal and Ising undirected graphical models. arXiv preprint arXiv:1806.06887, 2018. [9] Ilias Diakonikolas. Learning Structured Distributions. In Peter Bühlmann, Petros Drineas, Michael Kane, and Mark van der Laan, editors, Handbook of Big Data, chapter 15, pages 267– 283. Chapman and Hall/CRC, 2016. URL http://www.crcnetbase.com/doi/pdfplus/ 10.1201/b19567-21. [10] Ilias Diakonikolas, Daniel M. Kane, and Alistair Stewart. Statistical query lower bounds for robust estimation of high-dimensional Gaussians and Gaussian mixtures. In 2017 IEEE 58th Annual Symposium on Foundations of Computer Science (FOCS), pages 73–84, Oct 2017. doi: 10.1109/FOCS.2017.16. Available on arXiv:1611.03473 [cs.LG]. [11] Ilias Diakonikolas, Daniel M. Kane, and Alistair Stewart. Learning multivariate log-concave distributions. In Proceedings of Machine Learning Research, volume 65 of COLT’17, pages 1—-17, 2017. ISBN 3-540-35294-5, 978-3-540-35294-5. URL http://proceedings.mlr. press/v65/diakonikolas17a/diakonikolas17a.pdf. [12] Ildar Ibragimov. Estimation of analytic functions. In State of the art in probability and statistics (Leiden, 1999), volume 36 of IMS Lecture Notes Monogr. Ser., pages 359–383. Inst. Math. Statist., Beachwood, OH, 2001. doi: 10.1214/lnms/1215090078. URL https: //doi.org/10.1214/lnms/1215090078. [13] Adam Kalai, Ankur Moitra, and Gregory Valiant. Disentangling Gaussians. Communications of the ACM, 55(2), 2012. [14] Michael Kearns, Yishay Mansour, Dana Ron, Ronitt Rubinfeld, Robert E. Schapire, and Linda Sellie. On the learnability of discrete distributions. In Proceedings of the Twenty- sixth Annual ACM Symposium on Theory of Computing, STOC ’94, pages 273–282, New York, NY, USA, 1994. ACM. ISBN 0-89791-663-8. doi: 10.1145/195058.195155. URL http://doi.acm.org/10.1145/195058.195155. [15] Nick Littlestone and Manfred Warmuth. Relating data compression and learnability. Technical report, Technical report, University of California, Santa Cruz, 1986. [16] Alexander E. Litvak, Alain Pajor, Mark Rudelson, and Nicole Tomczak-Jaegermann. Smallest singular value of random matrices and geometry of random polytopes. Adv. Math., 195(2): 491–523, 2005. ISSN 0001-8708. URL https://doi.org/10.1016/j.aim.2004.08.004. [17] Mario Lucic, Matthew Faulkner, Andreas Krause, and Dan Feldman. Training Gaussian mixture models at scale via coresets. Journal of Machine Learning Research, 18(160):1–25, 2018. URL http://jmlr.org/papers/v18/15-506.html. [18] Shay Moran and Amir Yehudayoff. Sample compression schemes for VC classes. Journal of the ACM (JACM), 63(3):21, 2016. [19] Bernard W. Silverman. Density estimation for statistics and data analysis. Monographs on Statistics and Applied Probability. Chapman & Hall, London, 1986. ISBN 0-412-24620-1. [20] Ananda Theertha Suresh, Alon Orlitsky, Jayadev Acharya, and Ashkan Jafar- pour. Near-optimal-sample estimators for spherical gaussian mixtures. In Z. Ghahramani, M. Welling, C. Cortes, N. D. Lawrence, and K. Q. Weinberger, editors, Advances in Neural Information Processing Systems 27, pages 1395– 1403. Curran Associates, Inc., 2014. URL http://papers.nips.cc/paper/ 5251-near-optimal-sample-estimators-for-spherical-gaussian-mixtures. pdf. [21] Vladimir N. Vapnik and Alexey Ya. Chervonenkis. On the uniform convergence of relative frequencies of events to their probabilities. Theory of Probability & Its Applications, 16(2): 264–280, 1971. doi: 10.1137/1116025. 10