Publications · Conference

Towards convergence rates for parameter estimation in Gaussian-gated mixture of experts

Huy Nguyen, TrungTin Nguyen⋆†, Khai Nguyen, Nhat Ho

⋆ Equal contribution, † Corresponding author.

AISTATS 2024 · PMLR AISTATS 2024. Proceedings of Machine Learning Research, Vol. 238 (2024).

Abstract

Originally introduced as a neural network for ensemble learning, mixture of experts (MoE) has recently become a fundamental building block of highly successful modern deep neural networks for heterogeneous data analysis in several applications of machine learning and statistics. Despite its popularity in practice, a satisfactory level of theoretical understanding of the MoE model is far from complete. To shed new light on this problem, we provide a convergence analysis for maximum likelihood estimation (MLE) in the Gaussian-gated MoE model. The main challenge of that analysis comes from the inclusion of covariates in the Gaussian gating functions and expert networks, which leads to their intrinsic interaction via some partial differential equations with respect to their parameters. We tackle these issues by designing novel Voronoi loss functions among parameters to accurately capture the heterogeneity of parameter estimation rates. Our findings reveal that the MLE has distinct behaviors under two complement settings of location parameters of the Gaussian gating functions, namely when all these parameters are non-zero versus when at least one among them vanishes. Notably, these behaviors can be characterized by the solvability of two different systems of polynomial equations. Finally, we conduct a simulation study to empirically verify our theoretical results.

1 INTRODUCTION

Mixture of experts (MoE) [19, 23] is a popular statistical machine learning model where experts are either regression functions or classifiers, while the input-dependent weights (also called gating functions) softly partition the input space into different regions and define which regions each expert is responsible for (see [56, 32, 8] for further details). In regression analysis with heterogeneous data, softmax-gated MoE [19, 23] and Gaussian-gated MoE (GMoE) [54] models are the most popular choices. One of the main drawbacks of the softmax-gated MoE models is the difficulty of applying an expectation-maximization (EM) algorithm [5], which requires an internal iterative numerical optimization procedure, e.g., Newton-Raphson algorithm, to update the softmax parameters in the maximization step. On the other hand, parameters of the GMoE models can be updated analytically, which helps reduce the computational complexity of the estimation routine. For those reasons, GMoE has become a fundamental component of modern deep neural networks in various fields, including speech recognition [12, 55], computer vision [29, 50], natural language processing [52, 9, 35, 7, 49], medical images [14], robot dynamics [51, 34], remote sensing [4, 25, 10, 11], and econometrics [48, 47, 6]. However, there is a paucity of work aiming at theoretically understanding the density estimation and parameter estimation in the GMoE models, which has remained poorly understood in the literature to the best of our knowledge.

Related literature. In the GMoE setting, early classical research focused on identifiability issues [22] and parameter estimation in the exact-fitted setting, assuming the true number of components k0 is known [20]. For most applications, it is a too strong presumption as the true number of components is seldom known. To deal with this problem, there are three common practical approaches. The first approach is based on model selection, most importantly the Bayesian information criterion from asymptotic theory [11, 3, 24] and the slope heuristic [1, 2] in a non-asymptotic framework [45, 44, 46, 42]. In particular, the bias term can be substantially reduced with a sufficiently large model collection w.r.t. the number of mixture components k by well-studied universal approximations theorems [41, 33, 21]. However, since we have to search for the optimal k over all possible values, this approach is computationally expensive. The second approach is to design a tractable Bayesian nonparametric GMoE model. For example, [43] avoided any commitment to an arbitrary k with posterior consistency guarantee thanks to the merge-truncate-merge post-processing in [13]. However, this approach still depends on a tuning parameter, which prevents the direct application of this approach to real data sets. The last approach is to use prior knowledge to over-specify the true model, i.e. specifying more mixture components than necessary, where most existing work is limited to its particular case, including mixture models [15, 16, 17, 13, 30] and mixture of experts [18, 40, 37, 38, 36]. It is worth noting that the convergence behavior of parameter estimations in the GMoE model has remained an open question, which we aim to answer in this paper. Before going into further details, we first formally introduce an affine instance of the GMoE model. This is a simplified but standard setting where we use linear functions for Gaussian mean experts.

GMoE setting. GMoE models are used to capture the non-linear and heterogeneous relationship between the response Y𝒴 and the set of covariates X𝒳d, d. In the affine GMoE model, the response Y is approximated by a k0 local affine:

Y=j=1k0𝕀(Z=j)[(aj0)X+bj0+ej0]. (1)

Here 𝕀 is an indicator function and Z is a latent variable that captures a cluster relationship, such that Z=j if Y comes from cluster j[k0]:={1,2,,k0}. Vectors aj0d and scalars bj0 define cluster-specific affine transformations. In addition, ej0 are error terms that capture both the reconstruction error (due to the local affine approximations) and the observation noise in . Let d:={f(|ψ,Σ):ψd,Σ𝒮d+} be the family of d-dimensional Gaussian density functions with mean ψ and positive-definite covariance matrix Σ, where 𝒮d+ indicates the set of all symmetric positive-definite matrices on d×d. Following the usual assumption that ej0 is a zero-mean Gaussian variable with variance νj0+, it follows that

p(Y|X,Z=j)=f𝒟(Y|(aj0)X+bj0,νj0),f𝒟1.

To enforce the affine transformations to be local, X is defined as a mixture of k0 Gaussian components:

p(X|Z=j)=f(X|cj0,Γj0),p(Z=j)=πj0, (2)

where fd. Here, we refer to f𝒟 and f as the data density and the local density, respectively. Additionally, πj0>0 are called mixing proportions (or weights), satisfying j=1k0πj0=1. Via the law of total probability, we obtain the GMoE model of order k0 whose joint density function pG0(X,Y) is given by:

j=1k0πj0f(X|cj0,Γj0)f𝒟(Y|(aj0)X+bj0,νj0). (3)

Here, G0:=j=1k0πj0δ(cj0,Γj0,aj0,bj0,νj0) denotes a true but unknown probability mixing measure, where δ is the Dirac measure and for j[k0], (cj0,Γj0,aj0,bj0,νj0)Θd×𝒮d+×d××+ are called components of G0. We assume that {(Xi,Yi)}i[n] are i.i.d.  samples of random variable (X,Y), coming from the GMoE model of order k0. To facilitate our theoretical guarantee, we assume that Θ is compact and 𝒳 is bounded.

Maximum likelihood estimation. We propose a general theoretical framework for analyzing the statistical performance of maximum likelihood estimation (MLE) for parameters under the setting of the GMoE model. Since the true order k0 is generally unknown in practice, it is necessary to over-specify the number of components of mixing measures to at most k, where k>k0. In particular, we consider

G^nargmaxG𝒪k(Θ)i=1nlog(pG(Xi,Yi)), (4)

where 𝒪k(Θ):={G=i=1kπiδ(ci,Γi,ai,bi,νi):1kk,i=1kπi=1,(ci,Γi,ai,bi,νi)Θ} denotes the set of all mixing measures with at most k components.

Theoretical challenges. For the purpose of deriving parameter estimation rates in the GMoE model, we first use the Taylor expansion to decompose the term pG^n(X,Y)pG0(X,Y) into a linear combination of elements which belong to a linearly independent set and associate with coefficients involving the discrepancies between parameter estimations and true parameters. By doing so, when the density estimation pG^n converges to the true density pG0, those parameter discrepancies also go to zero and we then obtain our desired parameter estimation rates. Nevertheless, the density decomposition is challenging due to a number of linearly dependent derivative terms in the Taylor expansion. In particular, we find out two interactions among the parameters of either function f𝒟 or f via the following partial differential equations (PDEs):

2f𝒟b2=2f𝒟ν,2fcc=2fΓ. (5)

We refer to those interactions as interior interactions since each of them involves either parameters b,ν of function f𝒟 or parameters c,Γ of function f. Furthermore, we also figure out an interaction between the parameters of functions f𝒟 and f. More specifically, let us denote F(X,Y|θ):=f(X|c,Γ)f𝒟(Y|aX+b,ν) where θ:=(c,Γ,a,b,ν). Then, by taking the derivatives of F with respect to its parameters as follows:

2Fcb(X,Y|θj0) =Γ1(Xcj0)ff𝒟b;
Fa(X,Y|θj0) =Xff𝒟b,

it can be seen that the following PDE holds true when the location parameter of f vanishes, i.e. cj0=0:

2Fcb(X,Y|θj0)=Γ1Fa(X,Y|θj0). (6)

We refer to the interaction among parameters c,b,a in equation (6) as the exterior interaction. Back to the density decomposition, it is necessary to aggregate linearly dependent derivative terms in equations (5) and (6) by taking the summation of their associated coefficients. As a result, we achieve our desired linear combination of linearly independent terms. However, the structure of associated coefficients in that combination becomes complex owing to the previous aggregation. Thus, when those coefficients converge to zero, we have to cope with two complex systems of polynomial equations given in equations (9) and (12).

Overall contributions. In this paper, we characterize the convergence behavior of maximum likelihood estimation in the GMoE model. Firstly, we demonstrate that the density estimation pG^n converges to the true density pG0 under the Total Variation distance V at the parametric rate V(pG^n,pG0)=𝒪(n1/2). Regarding the parameter estimation problem, given the above challenge discussion, we consider two complement settings of the location parameters c10,c20,,ck00 based on the validity of the PDE in equation (6) as follows (see also Table 1):

1. Type I setting: all the values of c10,c20,,ck00 are different from zero. Since the PDE (6) does not hold under this setting, we have to deal with only the interior interactions in equation (5). Thus, we propose a novel Voronoi loss function D¯(G,G0) defined in equation (10) to capture those interactions, and then establish the Total Variation lower bound D¯(G^n,G0)V(pG^n,pG0)=𝒪(n1/2). This result together with the formulation of D¯(G^n,G0) indicate that exact-fitted parameters cj0,Γj0,aj0,bj0,νj0, which are approximated by exactly one component, share the same estimation rate of order 𝒪(n1/2). By contrast, the rates for estimating over-fitted parameters cj0,Γj0,bj0,νj0, which are fitted by at least two components, depend on the solvability of the system of polynomial equations (9) and become no faster than 𝒪(n1/4). These slow rates are due to the interior interactions among those parameters in equation (5). As over-fitted parameters aj0 are not involved in those interactions, their estimation rates keep unchanged of order 𝒪(n1/4).

2. Type II setting: at least one among the values of c10,c20,,ck00 is equal to zero. Without loss of generality, we assume that c10,c20,,ck~0 equal zero, where 1k~k0, while other cj’s are non-zero. Since the PDE (6) holds true under this setting, we have to confront both interior and exterior interactions among parameters. For that purpose, we construct another novel Voronoi loss function D~(G,G0) in equation (3.2) to handle those interactions, and then derive the Total Variation lower bound D~(G^n,G0)V(pG^n,pG0)=𝒪(n1/2). Due to the occurrence of both interior and exterior interactions, the rates for estimating over-fitted parameters cj0,Γj0,aj0,bj0,νj0 are now determined by the solvability of both systems of polynomial equations (9) and (12). Meanwhile, the estimation rates for their exact-fitted counterparts remain the same of order 𝒪(n1/2).

Setting Exact-fitted cj0,Γj0,aj0,bj0,νj0 Over-fitted aj0 Over-fitted cj0,bj0 Over-fitted Γj0,νj0
j[k~] j[k0][k~] j[k~] j[k0][k~] j[k~] j[k0][k~]
Type I 𝒪(n1/2) 𝒪(n1/4) 𝒪(n1/2r¯j) 𝒪(n1/r¯j)
Type II 𝒪(n1/2) 𝒪(n1/r~j) 𝒪(n1/4) 𝒪(n1/2r~j) 𝒪(n1/2r¯j) 𝒪(n1/r~j) 𝒪(n1/r¯j)
Table 1: Summary of parameter estimation rates in the GMoE model under the Type I and Type II settings. Recall that the cardinality of Voronoi cells 𝒜j (see Section 2) generated by true components (cj0,Γj0,aj0,bj0,νj0) indicates the number of components fitting them. When |𝒜j|=1, we call them exact-fitted parameters, but when |𝒜j|>1, they are referred to as over-fitted parameters. Additionally, the notations r¯j:=r¯(|𝒜j|) and r~j:=r~(|𝒜j|) stand for the solvability of two polynomial equation systems (9) and (12), respectively. For example, if |𝒜j|=2, then we have r¯j=r~j=4. Meanwhile, we get r¯j=r~j=6 if |𝒜j|=3.

Practical implication. In practice, the parameters specific to each mixing component may carry useful information about the heterogeneity of the underlying (latent) subpopulations. Since in reality there is a tendency to “over-fit” the mixture generously by adding many more mixing components, our theory warns against this because, as we have shown, the convergence rate via standard methods such as MLE for subpopulation-specific parameters deteriorates rapidly with the number of redundant components. Hopefully, the theoretical results will suggest practical ways to identify benign scenarios and impose helpful constraints when GMoE models have favourable convergence rates, and detect pathological scenarios that practitioners would do well to avoid. In particular, practitioners can consistently estimate the true number of components based on our important threshold on the convergence rates of the MLE using the merge-truncate-merge procedure [13] or Group-Sort-Fuse [31].

Paper organization.

The rest of this paper proceeds as follows. In Section 2, we begin with providing some background on the identifiability of the GMoE model and the rate for estimating the joint density function under that model. Next, in Section 3, we establish the convergence rates of parameter estimation under both Type I and Type II settings, which are then empirically verified by simulation studies in Section 4. Finally, we conclude the paper in Section 5 and defer proofs of all theoretical results to the supplementary material.

Notation.

Throughout the paper, {1,2,,n} is abbreviated as [n] for any n. Given any two sequences of positive real numbers {an}n=1 and {bn}n=1, we write an=𝒪(bn) or anbn to indicate that there exists a constant C>0 such that anCbn for all n1. Next, for any vector vd, we denote |v|:=v1+v2++vd, whereas vp stands for its p-norm with a note that v implicitly indicates the 2-norm unless stating otherwise. By abuse of notation, we also denote by A the Frobenius norm of any matrix Ad×d. Additionally, the notation |S| represents for the cardinality of any set S. Finally, given two probability density functions p,q with respect to the Lebesgue measure μ, we define V(p,q):=12|pq|𝑑μ as their Total Variation distance, while h2(p,q):=12(pq)2𝑑μ denotes the squared Hellinger distance between them.

2 PRELIMINARIES

In this section, we first verify the identifiability of the GMoE model, and then establish the density estimation rate under that model. Lastly, we introduce a notion of a Voronoi cells, which will be used to build Voronoi loss functions in Section 3.

Firstly, we demonstrate in the following proposition that the GMoE model is identifiable:

Proposition 1 (Identifiability of the GMoE model).

Let G and G be two mixing measures in 𝒪k(Θ). If the equation pG(X,Y)=pG(X,Y) holds true for almost surely (X,Y)𝒳×𝒴, then we obtain that GG.

The proof of Proposition 1 is deferred to Appendix C.1. Given the above result, we know that two mixing measures G and G are equivalent if and only if they share the same joint density function.

Next, we characterize the convergence rate of the joint density estimation pG^n to its true counterpart pG0 under the Total Variation distance.

Proposition 2 (Joint Density Estimation Rate).

With the MLE G^n defined in equation (4), the following bound indicates that the density estimation pG^n converges to the true density pG0 under the Total Variation distance at the parametric rate of order 𝒪(n1/2) (up to a logarithmic term):

(V(pG^n,pG0)>C1log(n)/n)nC2,

where C1 and C2 are universal constants.

The proof of Proposition 2 can be found in Appendix C.2. This result is a key ingredient to study the parameter estimation problem in the GMoE model in subsequent sections. In particular, if we are able to establish the lower bound of the Total Variation distance in terms of some loss function D between two mixing measures, i.e., V(pG,pG0)D(G,G0) for any mixing measure G𝒪k(Θ), then the MLE G^n also converges to the true mixing measure G0 at the parametric rate of 𝒪(n1/2). Based on this result, we then achieve the parameter estimation rates through the formulation of the loss function D(G^n,G0). For that purpose, let us introduce a notion of Voronoi cells which are essential to construct Voronoi loss functions later in Section 3.

Voronoi cells. In general, true parameters which are fitted by exactly one component should enjoy faster estimation rates than those approximated by more than one component. Therefore, in order to capture the convergence behavior of parameter estimations accurately, we define k0 different index sets called Voronoi cells to control the number of fitted components approaching each of the k0 true components. More formally, for any G𝒪k(Θ), the Voronoi cell 𝒜j:=𝒜j(G) generated by θj0:=(cj0,Γj0,aj0,bj0,νj0) is defined as

𝒜j :={i[k]:θiθj0θiθ0,j}, (7)

for any j[k0], where θi:=(ci,Γi,ai,bi,νi). An illustration of Voronoi cells is given in Appendix A. Notably, the cardinality of each Voronoi cell 𝒜j is exactly the number of fitted components approximating the true component θj0.

3 PARAMETER ESTIMATION RATES

In this section, we conduct the convergence analysis for parameter estimation in the GMoE model under the Type I and Type II settings in Section 3.1 and Section 3.2, respectively. Then, we sketch the proof for main results in both settings in Section 3.3.

3.1 Type I Setting

Let us recall that under this setting, all the values of c10,c20,,ck00 are non-zero. Although the exterior interaction between the parameters of two functions f and f𝒟 mentioned in equation (6) does not hold in this scenario, we encounter two interior interactions among parameters b,ν and c,Γ via the following PDEs:

2f𝒟b2=2f𝒟ν,2fcc=2fΓ. (8)

System of polynomial equations. To precisely characterize the estimation rates for those parameters, we need to consider the solvability of a system of polynomial equations which was previously studied in [15]. In particular, for each m2, let r¯(m) be the smallest positive integer r such that the system:

l=1mn1,n2:n1+2n2=spl2q1ln1q2ln2n1!n2!=0,s=1,2,,r, (9)

does not have any non-trivial solutions for the unknown variables {pl,q1l,q2l}l=1m. Here, a solution is called non-trivial if all the values of pl are different from zero, whereas at least one among q1l is non-zero. The following lemma gives us the values of r¯(m) at some specific points m.

Lemma 1 (Proposition 2.1, [15]).

When m=2, we have that r¯(m)=4, while for m=3, we get r¯(m)=6. If m4, then r¯(m)7.

Proof of Lemma 1 is in [15]. Now, we are ready to introduce a Voronoi loss function used for this setting.

Voronoi loss function. For simplicity, we denote Δcij:=cicj0, ΔΓij:=ΓiΓj0, Δaij:=aiaj0, Δbij:=bibj0, Δνij:=νiνj0 and r¯j:=r¯(|𝒜j|). Additionally, we also define mappings Kij:5 such that Kij(κ1,κ2,κ3,κ4,κ5):=Δcijκ1+ΔΓijκ2+Δaijκ3+|Δbij|κ4+|Δνij|κ5, for any j[k0] and i𝒜j. Then, the Voronoi loss function D¯(G,G0) of interest in this setting is given by:

D¯(G,G0):=j:|𝒜j|>1,i𝒜jπiKij(r¯j,r¯j2,2,r¯j,r¯j2)+
j:|𝒜j|=1,i𝒜jπiKij(1,1,1,1,1)+j=1k0|i𝒜jπiπj0|. (10)

Given this loss function, we capture parameter estimation rates in the GMoE model in the following theorem.

Theorem 1.

Under the Type I setting, the Total Variation lower bound V(pG,pG0)D¯(G,G0) holds for any G𝒪k(Θ), which implies that there exists a universal constant C3>0 depending on G0 and Θ satisfying

(D¯(G^n,G0)>C3log(n)/n)nC4,

where C4>0 is a constant that depends only on Θ.

Proof of Theorem 1 is in Appendix B.1. It follows from Theorem 1 that the discrepancy D¯(G^n,G0) vanishes at a rate of order 𝒪(n1/2) up to a logarithmic constant, which leads to following observations: (i) True parameters cj0,Γj0,aj0,bj0,νj0, which are fitted by exactly one component, share the same estimation rate of order 𝒪(n1/2); (ii) By contrast, the rates for estimating parameters fitted by more than one element are significantly slower. In particular, the estimation rates for cj0,bj0 are of order 𝒪(n1/2r¯(|𝒜jn|)), whereas those for Γj0,νj0 are of order 𝒪(n1/r¯(|𝒜jn|)) in which 𝒜jn:=𝒜j(G^n). For instance, if we have |𝒜jn|=3, then Lemma 1 indicates that the previous two rates become 𝒪(n1/12) and 𝒪(n1/6), respectively. These slow rates are owing to the interior interactions among those parameters in equation (8). Meanwhile, aj0 admits a much faster rate of order 𝒪(n1/4) as it does not interact with other parameters.

3.2 Type II Setting

Next, we consider the Type II setting, namely when at least one among c10,c20,,ck00 is equal to vector 𝟎d. Without loss of generality, we assume that c10=c20==ck~0=𝟎d, while ck~+10,ck~+20,,ck00 are different from 𝟎d. Under this setting, we encounter not only the two interior interactions in equation (8) but also the exterior interaction expressed by the following PDE:

2Fcb(X,Y|θj0)=Γ1Fa(X,Y|θj0), (11)

where F(X,Y|θ):=f(X|c,Γ)f𝒟(Y|aX+b,ν) and θ:=(c,Γ,a,b,ν). This phenomenon poses a lot of challenges in the parameter estimation problem. Therefore, we will only present the results when d=1 for simplicity, while those for the setting d>1 can be argued in a similar fashion but with more complex notations.

System of polynomial equations. Due to the emergence of the exterior interaction, we need to control the solvability of a totally new system of polynomial equations, which is given by

l=1mα𝒥1,2pl2q1lα1q2lα2q3lα3q4lα4q5lα5α1!α2!α3!α4!α5!=0, (12)

for all 1,20 satisfying 11+2r, where 𝒥1,2:={α=(αl)l=155:α1+2α2+α3=1,α3+α4+2α5=2}. Now, we define r~(m) as the smallest natural number r such that the system in equation (12) does not have any non-trivial solutions for the unknown variables {pl,q1l,q2l,q3l,q4l,q5l}l=1m, namely, all of pl are non-zero, whereas at least one among q4l is different from zero. The following lemma establishes a connection between r~(m) and r¯(m) as well as provides the values of r~(m) given some specific choices of m.

Lemma 2.

In general, we have r~(m)r¯(m) for all m. Furthermore, the equality occurs when m=2 and m=3, meaning that r~(2)=4 and r~(3)=6.

Proof of Lemma 2 is in Appendix C.3. Next, we introduce a Voronoi loss function tailored to this setting.

Voronoi loss function. Firstly, let us reformulate the mappings Kij defined in Section 3.1 for d=1 as Kij(κ1,κ2,κ3,κ4,κ5):=|Δcij|κ1+|ΔΓij|κ2+|Δaij|κ3+|Δbij|κ4+|Δνij|κ5. In addition, we denote r~j:=r~(|𝒜j|) and r¯j:=r¯(|𝒜j|), for any j[k0]. Then, the Voronoi loss of interest D~(G,G0) is defined as follows:

D~(G,G0):=j[k~]:|𝒜j|>1,i𝒜jπiKij(r~j,r~j2,r~j2,r~j,r~j2)
+j[k0][k~]:|𝒜j|>1,i𝒜jπiKij(r¯j,r¯j2,2,r¯j,r¯j2)
+j[k0]:|𝒜j|=1,i𝒜jπiKij(1,1,1,1,1)+j=1k0|i𝒜jπiπj0|. (13)

Given the above loss function, we derive the rates for estimating parameters under the Type II setting in the following theorem.

Theorem 2.

Under the Type II setting, the Total Variation lower bound V(pG,pG0)D~(G,G0) holds for any G𝒪k(Θ), which indicates that we can find a constant C5>0 depending on G0,Θ such that

(D~(G^n,G0)>C5log(n)/n)nC6,

where C6>0 is a constant that depends only on Θ.

Proof of Theorem 2 is in Appendix B.2. Similar to Theorem 1, the Voronoi loss D~(G^n,G0) also converges to zero at a rate of order 𝒪(n1/2) (up to a logarithmic term) under the Type II setting. Moreover, true parameters cj0,Γj0,aj0,bj0,νj0 enjoy the same estimation rates as their counterparts in Section 3.1 for any j[k0]:|𝒜jn|=1 and j[k0][k~]:|𝒜jn|>1. However, the difference in the convergence behavior occurs when j[k~]:|𝒜jn|>1. In particular, the rates for estimating parameters aj0 now drop substantially to 𝒪(n1/r~(|𝒜j|)) in comparison with 𝒪(n1/4) under the Type I settings. This phenomenon happens due to the interaction of aj0 with parameters cj0,bj0 via the PDE in equation (11).

3.3 Proof Sketch

Since arguments used for the proof of Theorem 1 are included in that of Theorem 2, we will present the former proof sketch implicitly inside the latter. In particular, we focus on establishing the bound infG𝒪k(Θ)V(pG,pG0)/D~(G,G0)>0 under the Type II setting when d=1. For that purpose, we will respectively demonstrate its local and global versions by contradiction as follows:

Local bound: We wish to prove that

limε>0infG𝒪k(Θ),D~(G,G0)εV(pG,pG0)/D~(G,G0)>0.

Assume that this bound does not hold, then we can find a sequence Gn=i=1knπinδθin𝒪k(Θ), where θin:=(cin,Γin,ain,bin,νin), such that V(pGn,pG0)/D~(Gn,G0) and D~(Gn,G0) both vanish as n. Now, we decompose Ξn:=pGn(X,Y)pG0(X,Y) as

Ξn =j=1k0i𝒜jπin[F(X,Y|θin)F(X,Y|θj0)]
+j=1k0(i𝒜jπinπj0)F(X,Y|θj0),

where θj0:=(cj0,Γj0,aj0,bj0,νj0). Let us denote h1(X,a,b)=aX+b for any ad,b. Then, for i𝒜j and i𝒜j where j[k~] and j[k0][k~], we invoke the Taylor expansion up to some orders r1j and r2j (we will choose later) for F(X,Y|θin) and F(X,Y|θin), respectively, as follows:

F(X,Y|θin)F(X,Y|θj0)
=1+2=12r1jQ1,2n(j)X1f(X|cj0,Γj0)
×2f𝒟h12(Y|aj0X+bj0,νj0)+R1ij(X,Y),
F(X,Y|θin)F(X,Y|θj0)
=R2ij(X,Y)+α3=0r2jτ1+τ2=02(r2jα3)Tα3,τ1,τ2n(j)Xα3
×τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0).

Here R1ij(X,Y) and R2ij(X,Y) are Taylor remainders such that their ratios to D~(Gn,G0) vanishes as n. Thus, we can treat Ξn/D~(Gn,G0) as a linear combination of linearly independent terms

X1f(X|cj0,Γj0)2f𝒟h12(Y|aj0X+bj0,νj0),
Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0)

associated with coefficients Q1,2n(j) and Tα3,τ1,τ2n(j), respectively. Moreover, it follows from Fatou’s lemma that Ξn/D~(Gn,G0) approaches zero when n. Consequently, all the coefficients in the representation of Ξn/D~(Gn,G0), i.e. Q1,2n(j)/D~(Gn,G0) and Tα3,τ1,τ2n(j)/D~(Gn,G0), go to zero as n. Therefore, in order to point out a contradiction, we need to choose the values of r1j and r2j such that at least one among these coefficients does not vanish. As a result, we achieve the aforementioned local bound. Now, we will show how to determine such values of r1j and r2j. It is worth noting that if we set k~=0, then Type II settings reduces to Type I settings and we only need to deal with r2j as follows:

Type I setting: We will specify an appropriate of r2j during proving by contradiction that not all the coefficients Tα3,τ1,τ2n(j)/D~(Gn,G0) tend to zero. Assume that these coefficients all vanish, then we extract some useful limits among them for our arguments and end up with the following system of polynomial equations:

i𝒜jn1+2n2=spl2q1ln1q2ln2n1!n2!=0,s=1,2,,r2j.

By construction, this system must have at least one non-trivial solution. Thus, to contradict this condition, we set r2j=r¯(|𝒜j|), which makes the above system has no non-trivial solutions.

Type II setting: When k~>0, i.e. there exist some zero-valued parameter cj, we will keep r2j=r¯(|𝒜j|) for all j[k0][k~] and find the desired values of r1j for j[k~] by showing by contradiction that not all the coefficients Q1,2n(j)/D~(Gn,G0) go to zero. En route to pointing out a contradiction to the hypothesis, we come across a more complex system of polynomial equations than its counterpart in the previous setting, specifically

i𝒜jα𝒥1,2pi2q1iα1q2iα2q3iα3q4iα4q5iα5α1!α2!α3!α4!α5!=0,

for all 1,20 such that 11+2r1j, where 𝒥1,2:={α=(αi)i=155:α1+2α2+α3=1,α3+α4+2α5=2}. Since this system necessarily has a non-trivial solution, we choose r1j=r~(|𝒜j|) so that it admits only trivial solutions, which contradicts the previous claim. Consequently, we can find a constant ε>0 such that

infG𝒪k(Θ),D~(G,G0)εV(pG,pG0)/D~(G,G0)>0.

Global bound: Thus, to complete the proof, it is sufficient to demonstrate the global bound

infG𝒪k(Θ),D~(G,G0)>εV(pG,pG0)/D~(G,G0)>0.

If this bound did not hold, there would be a mixing measure G𝒪k(Θ) that satisfies pG(X,Y)=pG0(X,Y) for almost surely (X,Y), which leads to GG0 by Proposition 1. As a result, we obtain D~(G,G0)=0, which contradicts the constraint that D~(G,G0)>ε. Hence, the proof sketch is completed.

Refer to caption
(a) Model I, k=4
Refer to caption
(b) Model I, k=5
Refer to caption
(c) Model II, k=4
Refer to caption
(d) Model II, k=5
Figure 1: Log-log scaled plots of the empirical mean of discrepancies D¯(G^n,G0) and D~(G^n,G0) and G0 (orange lines with error bars) and least-squares fitted linear regression (black dash-dotted lines) when d=1 and k0=3.

4 EXPERIMENTS

In this section, we empirically validate the convergence rates of parameter estimation in four GMoE models which satisfy the assumptions of Type I and Type II settings, respectively, when k0=3. Note that for simplicity, we only perform a simulation study to illustrate the convergence rates of Theorems 1 and 2 for the GMoE model when X lies in one- and two-dimensional space with unknown location and scale parameters. All code to reproduce our simulation study is publicly available111https://github.com/Trung-TinNGUYEN/CRPE-GMoE and all simulations below were performed in Python 3.9.13 on a standard Unix machine.

Numerical schemes. In Model I, we set G0 as follows:

j=13πj0δ(cj0,Γj0,aj0,bj0,νj0)=0.3δ(0.1,0.04,0.40,0.34,0.01)
+0.4δ(0.1,0.02,0.71,0.33,0.03)+0.3δ(0.5,0.01,0,0.2,0.02).

For Model II, we consider the same setting as in Model I but with c10=0 and b10=0.3. To demonstrate the claim that the empirical convergence rates of parameter estimation under the Type I (Model III) and Type II (Model IV) settings also hold in higher dimensions, we conduct a numerical simulation for d=2 and k0=3. In Model III, we set G0 as

j=13πj0δ(cj0,Γj0,aj0,bj0,νj0)=0.3δ(0.1𝟏d,0.04𝐈d,0.4𝟏d,0.34,0.01)
+0.4δ(0.1𝟏d,0.02𝐈d,0.71𝟏d,0.33,0.03)
+0.3δ(0.5𝟏d,0.01𝐈d,𝟎d,0.2,0.02),

where 𝟏d=(1,1), 𝟎d=(0,0) and 𝐈d is the identity matrix of size d. In Model IV, we consider the same setting of G0 as in Model III but with c10=𝟎d and b10=0.3.

Refer to caption
(a) Model III, k=4
Refer to caption
(b) Model III, k=5
Refer to caption
(c) Model IV, k=4
Refer to caption
(d) Model IV, k=5
Figure 2: Log-log scaled plots of the empirical mean of discrepancies D¯(G^n,G0) and D~(G^n,G0) (orange lines with error bars) and least-squares fitted linear regression (black dash-dotted lines) when d=2 and k0=3.

Numerical details. In accordance with the hierarchical GMoE setting of (2), we generate 20 samples (Xi,Yi)i[n] of size n for each setting, given 100 different choices of sample size n between 102 and 105. Then, we compute the MLE G^n w.r.t. a number of components k for each sample. For both of these settings, we choose k{k0+1,k0+2} with corresponding r¯,r~{4,6} using Lemmas 1 and 2. Here we implement the MLE using the EM algorithm for GMoE. This is a simplification of a general hybrid GMoE-EM from [4, Section 5]. We choose the convergence criteria ϵ=105 and 2000 maximum EM iterations. Our goal is to illustrate the theoretical properties of the estimator G^n. Therefore, we have initialized the EM algorithm in a favourable way. More specifically, we first randomly partitioned the set {1,,k} into k0 index sets J1,,Jk0, each containing at least one point, for any given k and k0 and for each replication. Finally, we sampled cj0 (resp. Γj0,aj0,bj0,νj0) from a unique Gaussian distribution centered on ct0 (resp. Γt0,at0,bt0,νt0), with vanishing covariance so that jJt.

Empirical convergence rates. The empirical mean of discrepancies D¯ and D~ between G^n and G0, and the choice of k for Models I-II are reported in Figure 1. It can be observed from Figure 1 that those average discrepancies vanish at a rate of order 𝒪(n1/2), which matches the results of Theorems 1 and 2, where the only theoretical assumption that can be violated is the global convergence of the MLE. Note that the use of the joint density function allows the GMoE to be linked to a hierarchical mixture model, which guarantees global convergence for parameter estimation for arbitrary dimensions, see recent advances, e.g.,  [27, 26, 28]. We can therefore guarantee that the rates in Theorems 1 and 2 also hold in higher dimensions. Indeed, it can be observed from Figure 2 that the average discrepancies D¯(G^n,G0) and D~(G^n,G0) also approach zero at the rate of order 𝒪(n1/2) for d=2, confirming the empirical behaviour of Theorems 1 and 2 under the high dimensional settings.

5 CONCLUSION

In this paper, we conduct a convergence analysis for density estimation and parameter estimation in the Gaussian-gated mixture of experts (GMoE) under two complement settings of location parameters of the gating function. We demonstrate that the density estimation rate remains parametric on the sample size under both settings. On the other hand, due to several challenges induced by the interior and exterior interactions among parameters arising in those settings, we have to solve two complex systems of polynomial equations and then propose two corresponding novel Voronoi loss functions among parameters. We show that these Voronoi losses are able to capture the dependence of parameter estimation rates on the number of fitted components, which are more accurate than those characterized by the generalized Wasserstein loss used in previous works. We believe that our current techniques can be extended to the GMoE model with general experts in [18] and to the hierarchical MoE for exponential family models in [20]. In addition, understanding the convergence behavior of least squares estimation under the deterministic MoE model [39] with Gaussian gate is also a potential direction. However, we leave such non-trivial developments for future work.

Acknowledgements

NH acknowledges support from the NSF IFML 2019844 and the NSF AI Institute for Foundations of Machine Learning.

References

  • [1] J.-P. Baudry, C. Maugis, and B. Michel. Slope heuristics: overview and implementation. Statistics and Computing, 22(2):455–470, 2012. a
  • [2] L. Birgé and P. Massart. Minimal penalties for Gaussian model selection. Probability Theory and Related Fields, 138(1):33–73, 2007. Publisher: Springer. a
  • [3] F. Chamroukhi and B.-T. Huynh. Regularized Maximum Likelihood Estimation and Feature Selection in Mixtures-of-Experts Models. Journal de la Société Française de Statistique, 160(1):57–85, 2019. a
  • [4] A. Deleforge, F. Forbes, and R. Horaud. High-dimensional regression with gaussian mixtures and partially-latent response variables. Statistics and Computing, 25(5):893–911, 2015. a b
  • [5] A. P. Dempster, N. M. Laird, and D. B. Rubin. Maximum Likelihood from Incomplete Data Via the EM Algorithm. Journal of the Royal Statistical Society: Series B (Methodological), 39(1):1–22, Sept. 1977. Publisher: John Wiley & Sons, Ltd. a
  • [6] C. Diani, G. Galimberti, and G. Soffritti. Multivariate cluster-weighted models based on seemingly unrelated linear regression. Computational Statistics & Data Analysis, page 107451, Feb. 2022. a
  • [7] T. G. Do, H. K. Le, T. Nguyen, Q. Pham, B. T. Nguyen, T.-N. Doan, C. Liu, S. Ramasamy, X. Li, and S. HOI. HyperRouter: Towards Efficient Training and Inference of Sparse Mixture of Experts. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, Singapore, Dec. 2023. Association for Computational Linguistics. a
  • [8] W. Fedus, J. Dean, and B. Zoph. A review of sparse expert models in deep learning. arXiv preprint arXiv:2209.01667, 2022. a
  • [9] W. Fedus, B. Zoph, and N. Shazeer. Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity. Journal of Machine Learning Research, 23(120):1–39, 2022. a
  • [10] F. Forbes, H. D. Nguyen, T. Nguyen, and J. Arbel. Mixture of expert posterior surrogates for approximate Bayesian computation. In JDS 2022 - 53èmes Journées de Statistique de la Société Française de Statistique (SFdS), Lyon, France, June 2022. a
  • [11] F. Forbes, H. D. Nguyen, T. Nguyen, and J. Arbel. Summary statistics and discrepancy measures for approximate Bayesian computation via surrogate posteriors. Statistics and Computing, 32(5):85, Oct. 2022. a b
  • [12] J. Fritsch, M. Finke, and A. Waibel. Adaptively growing hierarchical mixtures of experts. In Advances in Neural Information Processing Systems, volume 9, 1996. a
  • [13] A. Guha, N. Ho, and X. Nguyen. On posterior contraction of parameters and interpretability in Bayesian mixture modeling. Bernoulli, 27(4):2159 – 2188, 2021. Publisher: Bernoulli Society for Mathematical Statistics and Probability. a b c
  • [14] X. Han, H. Nguyen, C. Harris, N. Ho, and S. Saria. Fusemoe: Mixture-of-experts transformers for fleximodal fusion. arXiv preprint arXiv:2402.03226, 2024. a
  • [15] N. Ho and X. Nguyen. Convergence rates of parameter estimation for some weakly identifiable finite mixtures. The Annals of Statistics, 44(6):2726 – 2755, 2016. Publisher: Institute of Mathematical Statistics and Bernoulli Society. a b c d
  • [16] N. Ho and X. Nguyen. On strong identifiability and convergence rates of parameter estimation in finite mixtures. Electronic Journal of Statistics, 10(1):271–307, 2016. Publisher: The Institute of Mathematical Statistics and the Bernoulli Society. a
  • [17] N. Ho and X. Nguyen. Singularity Structures and Impacts on Parameter Estimation in Finite Mixtures of Distributions. SIAM Journal on Mathematics of Data Science, 1(4):730–758, Jan. 2019. Publisher: Society for Industrial and Applied Mathematics. a
  • [18] N. Ho, C.-Y. Yang, and M. I. Jordan. Convergence Rates for Gaussian Mixtures of Experts. Journal of Machine Learning Research, 23(323):1–81, 2022. a b
  • [19] R. A. Jacobs, M. I. Jordan, S. J. Nowlan, and G. E. Hinton. Adaptive mixtures of local experts. Neural computation, 3(1):79–87, 1991. Publisher: MIT Press. a b
  • [20] W. Jiang and M. A. Tanner. Hierarchical mixtures-of-experts for exponential family regression models: approximation and maximum likelihood estimation. Annals of Statistics, pages 987–1011, 1999. a b
  • [21] W. Jiang and M. A. Tanner. Hierarchical Mixtures-of-Experts for Generalized Linear Models: Some Results on Denseness and Consistency. In D. Heckerman and J. Whittaker, editors, Proceedings of the Seventh International Workshop on Artificial Intelligence and Statistics, volume R2 of Proceedings of Machine Learning Research. PMLR, Jan. 1999. a
  • [22] W. Jiang and M. A. Tanner. On the identifiability of mixtures-of-experts. Neural Networks, 12(9):1253–1258, 1999. a
  • [23] M. I. Jordan and R. A. Jacobs. Hierarchical mixtures of experts and the EM algorithm. Neural computation, 6(2):181–214, 1994. Publisher: MIT Press. a b
  • [24] A. Khalili. New estimation and feature selection methods in mixture-of-experts models. Canadian Journal of Statistics, 38(4):519–539, 2010. Publisher: Wiley Online Library. a
  • [25] B. Kugler, F. Forbes, and S. Douté. Fast Bayesian inversion for high dimensional inverse problems. Statistics and Computing, 32(2):31, Mar. 2022. a
  • [26] J. Kwon and C. Caramanis. EM Converges for a Mixture of Many Linear Regressions. In S. Chiappa and R. Calandra, editors, Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, volume 108 of Proceedings of Machine Learning Research, pages 1727–1736. PMLR, Aug. 2020. a
  • [27] J. Kwon, N. Ho, and C. Caramanis. On the Minimax Optimality of the EM Algorithm for Learning Two-Component Mixed Linear Regression. In A. Banerjee and K. Fukumizu, editors, Proceedings of The 24th International Conference on Artificial Intelligence and Statistics, volume 130 of Proceedings of Machine Learning Research, pages 1405–1413. PMLR, Apr. 2021. a
  • [28] J. Kwon, W. Qian, C. Caramanis, Y. Chen, and D. Davis. Global Convergence of the EM Algorithm for Mixtures of Two Component Linear Regression. In A. Beygelzimer and D. Hsu, editors, Proceedings of the Thirty-Second Conference on Learning Theory, volume 99 of Proceedings of Machine Learning Research, pages 2055–2110. PMLR, June 2019. a
  • [29] S. Lathuilière, R. Juge, P. Mesejo, R. Muñoz-Salinas, and R. Horaud. Deep mixture of linear inverse regressions applied to head-pose estimation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 4817–4825, 2017. a
  • [30] T. Manole and N. Ho. Refined Convergence Rates for Maximum Likelihood Estimation under Finite Mixture Models. In K. Chaudhuri, S. Jegelka, L. Song, C. Szepesvari, G. Niu, and S. Sabato, editors, Proceedings of the 39th International Conference on Machine Learning, volume 162 of Proceedings of Machine Learning Research, pages 14979–15006. PMLR, July 2022. a
  • [31] T. Manole and A. Khalili. Estimating the number of components in finite mixture models via the Group-Sort-Fuse procedure. The Annals of Statistics, 49(6):3043 – 3069, 2021. Publisher: Institute of Mathematical Statistics. a
  • [32] S. Masoudnia and R. Ebrahimpour. Mixture of experts: a literature survey. Artificial Intelligence Review, 42(2):275–293, 2014. a
  • [33] E. F. Mendes and W. Jiang. On convergence rates of mixtures of polynomial experts. Neural computation, 24(11):3025–3051, 2012. Publisher: MIT Press. a
  • [34] J. Moody and C. J. Darken. Fast Learning in Networks of Locally-Tuned Processing Units. Neural Computation, 1(2):281–294, 1989. a
  • [35] B. Mustafa, C. R. Ruiz, J. Puigcerver, R. Jenatton, and N. Houlsby. Multimodal Contrastive Learning with LIMoE: the Language-Image Mixture of Experts. In A. H. Oh, A. Agarwal, D. Belgrave, and K. Cho, editors, Advances in Neural Information Processing Systems, 2022. a
  • [36] H. Nguyen, P. Akbarian, and N. Ho. Is temperature sample efficient for softmax Gaussian mixture of experts? arXiv preprint arXiv:2401.13875, 2024. a
  • [37] H. Nguyen, P. Akbarian, T. Nguyen, and N. Ho. A general theory for softmax gating multinomial logistic mixture of experts. arXiv preprint arXiv:2310.14188, 2023. a
  • [38] H. Nguyen, P. Akbarian, F. Yan, and N. Ho. Statistical perspective of top-k sparse softmax gating mixture of experts. In International Conference on Learning Representations, 2024. a
  • [39] H. Nguyen, N. Ho, and A. Rinaldo. On least squares estimation in softmax gating mixture of experts. arXiv preprint arXiv:2402.02952, 2024. a
  • [40] H. Nguyen, T. Nguyen, and N. Ho. Demystifying softmax gating function in Gaussian mixture of experts. In Advances in Neural Information Processing Systems, 2023. a
  • [41] H. D. Nguyen, T. Nguyen, F. Chamroukhi, and G. J. McLachlan. Approximations of conditional probability density functions in Lebesgue spaces via mixture of experts models. Journal of Statistical Distributions and Applications, 8(1):13, Aug. 2021. a
  • [42] T. Nguyen, F. Chamroukhi, H. D. Nguyen, and F. Forbes. Model selection by penalization in mixture of experts models with a non-asymptotic approach. In JDS 2022 - 53èmes Journées de Statistique de la Société Française de Statistique (SFdS), Lyon, France, June 2022. a
  • [43] T. Nguyen, F. Forbes, J. Arbel, and H. D. Nguyen. Bayesian nonparametric mixture of experts for high-dimensional inverse problems. hal-04015203, Mar. 2023. a
  • [44] T. Nguyen, D. N. Nguyen, H. D. Nguyen, and F. Chamroukhi. A non-asymptotic theory for model selection in high-dimensional mixture of experts via joint rank and variable selection. Preprint hal-03984011, Feb. 2023. a
  • [45] T. Nguyen, H. D. Nguyen, F. Chamroukhi, and F. Forbes. A non-asymptotic approach for model selection via penalization in high-dimensional mixture of experts models. Electronic Journal of Statistics, 16(2):4742 – 4822, 2022. Publisher: Institute of Mathematical Statistics and Bernoulli Society. a
  • [46] T. Nguyen, H. D. Nguyen, F. Chamroukhi, and G. J. McLachlan. Non-asymptotic oracle inequalities for the Lasso in high-dimensional mixture of experts. arXiv:2009.10622, Feb. 2023. a
  • [47] A. Norets and D. Pati. Adaptive Bayesian estimation of conditional densities. Econometric Theory, 33(4):980–1012, 2017. Publisher: Cambridge University Press. a
  • [48] A. Norets and J. Pelenis. Adaptive Bayesian estimation of conditional discrete-continuous distributions with an application to stock market trading activity. Journal of Econometrics, 2021. a
  • [49] Q. Pham, G. Do, H. Nguyen, T. Nguyen, C. Liu, M. Sartipi, B. T. Nguyen, S. Ramasamy, X. Li, S. Hoi, and N. Ho. Competesmoe – effective training of sparse mixture of experts via competition, 2024. a
  • [50] J. Puigcerver, C. R. Ruiz, B. Mustafa, C. Renggli, A. S. Pinto, S. Gelly, D. Keysers, and N. Houlsby. Scalable Transfer Learning with Expert Models. In International Conference on Learning Representations, 2021. a
  • [51] M. Sato and S. Ishii. On-line EM algorithm for the normalized gaussian network. Neural computation, 12(2):407–432, feb 2000. a
  • [52] N. Shazeer, A. Mirhoseini, K. Maziarz, A. Davis, Q. Le, G. Hinton, and J. Dean. Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer. In International Conference on Learning Representations, 2017. a
  • [53] S. van de Geer. Empirical Processes in M-estimation. Cambridge University Press, 2000. a b c d e f
  • [54] L. Xu, M. Jordan, and G. E. Hinton. An Alternative Model for Mixtures of Experts. In G. Tesauro, D. Touretzky, and T. Leen, editors, Advances in Neural Information Processing Systems, volume 7. MIT Press, 1995. a
  • [55] Z. You, S. Feng, D. Su, and D. Yu. Speechmoe2: Mixture-of-Experts Model with Improved Routing. In ICASSP 2022 - 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), pages 7217–7221, 2022. a
  • [56] S. E. Yuksel, J. N. Wilson, and P. D. Gader. Twenty Years of Mixture of Experts. IEEE Transactions on Neural Networks and Learning Systems, 23(8):1177–1193, 2012. a

In this supplementary material, we first include an illustration of Voronoi cells in Appendix A to help the readers understand this concept better. Then, we provide the proof of Theorem 1 and Theorem 2 in Appendix B. Finally, proofs for the remaining results are presented in Appendix C.

Appendix A ILLUSTRATION OF VORONOI CELLS

In this appendix, we aim to illustrate the Voronoi cells defined in Section 2. For that purpose, let us recall the definition of that concept here. In particular, for any mixing measure G𝒪k(Θ), the Voronoi cell 𝒜j:=𝒜j(G) generated by a true component θj0:=(cj0,Γj0,aj0,bj0,νj0) of G0 is given by

𝒜j:={i[k]:θiθj0θiθ0,j}, (14)

for any j[k0], where θi:=(ci,Γi,ai,bi,νi) is a component of G. Now, we provide an illustration of the above Voronoi cells under the setting when k0=6 and k=10 in Figure 3.

Refer to caption
Figure 3: Illustration of Voronoi cells defined in equation (14) when k0=6 and k=10. In this figure, red squares represent for true components (i.e. components of G0), while blue circles indicate fitted components (i.e. components of G). By definition, each Voronoi cell is generated by one true component, and its cardinality is exactly the number of corresponding fitted components. For example, the square in cell 4 is approximated by two rounds, which means that the cardinality of cell 4 is two.

Connection to Theorem 1. Under the Type I setting, parameters of the true components (cj0,Γj0,aj0,bj0,νj0) in cells 3, 5 and 6, which are fitted by one component, enjoy a parametric estimation rate of order 𝒪(n1/2). Next, the rates for estimating parameters c10,b10 of the true component in cell 1, which are approximated by three components, stand at order 𝒪(n1/2r¯(3))=𝒪(n1/12), while those for Γ10,ν10 are of order 𝒪(n1/r¯(3))=𝒪(n1/6). Meanwhile, the estimation rate for a10 is independent of the cardinality of its corresponding Voronoi cell and remains stable at order 𝒪(n1/4).

Connection to Theorem 2. Parameter estimation rates under the Type II setting share the same behavior as those in Theorem 1 except for the rates of estimating aj0. More specifically, if c10=0, the estimation rate for a10 now depends on the cardinality of cell 1, and experiences a drop to order 𝒪(n1/r~(3))=𝒪(n1/6).

Appendix B PROOF OF MAIN RESULTS

Before going to the proofs for Theorems 1 and 2 in Appendices B.1 and B.2, respectively, let us define some necessary notations used throughout this appendix. Firstly, for any vector vd, either vi or v(i) represents the i-th entry of v, while the sum of its entries is abbreviated as |v|:=v1+v2++vd. Next, for any vector pd, we denote vp:=v1p1v2p2vdpd and p!:=p1!p2!pd!. Additionally, we sometimes use the notation h1 and h2 to denote the expert functions considered in this work. In particular, we define h1(X,a,b)=aX+b as the mean expert function for any X𝒳d, ad and b, whereas h2(X,ν)=ν stands for the variance expert function for any ν+. Finally, since parameters in the proofs for Theorem 1 and Theorem 2 belong to various high-dimensional spaces, we summarize their domains in Table 2 and Table 3, respectively, to help readers keep track of them.

c Γ a b ν τ1 τ2 α1 α2 α3 α4 α5 1 2
Thm 1 d 𝒮d+ d + d d d×d d d N/A N/A
Table 2: Domains for parameters used in the proof of Theorem 1
c Γ a b ν τ1 τ2 α1 α2 α3 α4 α5 1 2
Thm 2 + +
Table 3: Domains for parameters used in the proof of Theorem 2

B.1 Proof of Theorem 1

Our goal is to show the following inequality:

infG𝒪k,β(Θ)V(pG,pG0)/D¯(G,G0)>0, (15)

which implies the desired Total Variation lower bound V(pG^n,pG0)D¯(G^n,G0). Given this bound, the joint density estimation rate in Proposition 2 then leads to the convergence rate of the MLE G^n to G0 under the loss D¯ as follows:

(D¯(G^n,G0)>C3log(n)/n)nC4,

for some universal constants C3 and C4. Note that the infimum in equation (15) is subject to all the mixing measures in the set 𝒪k,β(Θ):={G=i=1kπiδ(ci,Γi,ai,bi,νi):1kk,i=1kπi=1,πiβ,(ci,Γi,ai,bi,νi)Θ}, for some positive constant β. Now, we divide the proof of inequality (15) into two parts which we refer to as local bound and global bound.

Local bound: Firstly, we will prove the local version of inequality (15):

limε0infG𝒪k,β(Θ)D¯(G,G0)εV(pG,pG0)/D¯(G,G0)>0. (16)

Assume by contrary that the claim in equation (16) does not hold. Then, there exists a sequence of mixing measures Gn=i=1knπinδ(cin,Γin,ain,bin,νin)𝒪k,β(Θ) such that D¯(Gn,G0)0 and V(pGn,pG0)/D¯(Gn,G0)0 as n. Moreover, since knk for all n, we can replace (Gn) by its subsequence that admits a fixed number of atoms kn=kk. Additionally, 𝒜j=𝒜jn does not change with n for all j[k0].

Step 1 - Taylor expansion for density decomposition: Now, we consider the quantity

pGn(X,Y)pG0(X,Y)
= j:|𝒜j|>1i𝒜jπin[f(X|cin,Γin)f𝒟(Y|(ain)X+bin,νin)f(X|cj0,Γj0)f𝒟(Y|(aj0)X+bj0,νj0)]
+ j:|𝒜j|=1i𝒜jπin[f(X|cin,Γin)f𝒟(Y|(ain)X+bin,νin)f(X|cj0,Γj0)f𝒟(Y|(aj0)X+bj0,νj0)]
+ j=1k0(i𝒜jπinπj0)f(X|cj0,Γj0)f𝒟(Y|(aj0)X+bj0,νj0)
: =An+Bn+En.

For each j[k0]:|𝒜j|>1, we perform a Taylor expansion up to the r¯(|𝒜j|)-th order, and then rewrite An with a note that α=(α1,α2,α3,α4,α5)d×d×d×d×× as follows:

An =j:|𝒜j|>1i𝒜jπin|α|=1r¯(|𝒜j|)1α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5
×|α1|+|α2|fcα1Γα2(X|cj0,Γj0)|α3|+α4+α5f𝒟aα3bα4να5(Y|(aj0)X+bj0,νj0)+R1(X,Y)

where R1(X,Y) is a remainder term such that R1(X,Y)/D¯(Gn,G0)0 as n, which is due to the uniform Holder continuity of a location-scale Gaussian family. Since f d-dimensional Gaussian density functions, we have the following partial differential equation (PDE):

|α1|+|α2|fcα1Γα2(X|cj0,Γj0)=12|α2||α1|+2|α2|fcτ(α1,α2)(X|cj0,Γj0),

where τ(α1,α2):=(α1(v)+u=1d(α2(uv)+α2(vu)))v=1d=(α1(v)+2u=1dα2(uv))v=1dd. Similarly, as f is an univariate Gaussian density function, then

|α3|+α4+α5f𝒟aα3bα4να5(Y|(aj0)X+bj0,νj0) =Xα32α5|α3|+α4+2α5f𝒟h1|α3|+α4+2α5(Y|(aj0)X+bj0,νj0),

where h1(X,a,b)=aX+b is the mean expert function. Combine these results together, An can be represented as follows:

An =j:|𝒜j|>1i𝒜jπin|α|=1r¯(|𝒜j|)1α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5
×12|α2||α1|+2|α2|fcτ(α1,α2)(X|cj0,Γj0)Xα32α5|α3|+α4+2α5f𝒟h1|α3|+α4+2α5(Y|(aj0)X+bj0,νj0)+R1(X,Y),

Let τ1=τ(α1,α2)d and τ2=α4+2α5, we can rewrite An as

An =j:|𝒜j|>1|α3|=0r¯(|𝒜j|)|τ1|+τ2=02(r¯(|𝒜j|)|α3|)τ(α1,α2)=τ1α4+2α5=τ2i𝒜jπin2|α2|+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0)+R1(X,Y),

Analogously, for each j[k0]:|𝒜j|=1, by means of Taylor expansion up to the first order, Bn is rewritten as follows:

Bn =j:|𝒜j|=1|α3|=01|τ1|+τ2=02(1|α3|)τ(α1,α2)=τ1α4+2α5=τ2i𝒜jπin2|α2|+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0)+R2(X,Y), (17)

where R2(X,Y) is a remainder such that R2(X,Y)/D¯(Gn,G0)0 as n.

It is worth noting that An, Bn and En can be treated as linear combinations of elements of the following set:

:={Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y| (aj0)X+bj0,νj0):j[k0],0|α3|r¯(|𝒜j|),
0|τ1|+τ22(r¯(|𝒜j|)|α3|)}. (18)

Let Tα3,τ1,τ2n(j) be the coefficients of

Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0)

in the representations of An, Bn and En.

Step 2 - Proof of non-vanishing coefficients by contradiction: Assume that all the coefficients in the representations of An/D¯(Gn,G0), Bn/D¯(Gn,G0) and En/D¯(Gn,G0) go to 0 as n. Then, by taking the summation of the absolute values of coefficients in En/D¯(Gn,G0), which are |T𝟎d,𝟎d,0(j)|/D¯(Gn,G0) for all j[k0], we get that

1D¯(Gn,G0)j=1k0|i𝒜jπinπj0|0. (19)

Subsequently, from the formulation of Bn in equation (B.1), we have

1D¯(Gn,G0)j:|𝒜j|=1i𝒜jπin(Δcijn1+ΔΓijn1+Δaijn1+|Δbijn|+|Δνijn|)0.

It follows from the topological equivalence of 1-norm and 2-norm that

1D¯(Gn,G0)j:|𝒜j|=1i𝒜jπin(Δcijn+ΔΓijn+Δaijn+|Δbijn|+|Δνijn|)0. (20)

Next, from the formulation of An, by combining all terms of the form |Tα3,𝟎d,0(j)|/D¯(Gn,G0) where j[k0]:|𝒜j|>1 and α3{2e1,2e2,,2ed} with eu:=(0,,0,1u-th,0,,0) being a one-hot vector in d for all u[d], we obtain that

1D¯(Gn,G0)j:|𝒜j|>1i𝒜jπinΔaijn20. (21)

Putting the results in equations (19), (20) and (21) together with the formulation of D¯(Gn,G0) in equation (10), we deduce that

j:|𝒜j|>1i𝒜jπin(Δcijnr¯(|𝒜j|)+ΔΓijnr¯(|𝒜j|)/2+|Δbijn|r¯(|𝒜j|)+|Δνijn|r¯(|𝒜j|)/2)D¯(Gn,G0)1.

As a result, we can find an index j[k0] such that |𝒜j|>1 and

i𝒜jπin(Δcijnr¯(𝒜j)+ΔΓijnr¯(𝒜j)/2+|Δbijn|r¯(𝒜j)+|Δνijn|r¯(𝒜j)/2)D¯(Gn,G0)↛0. (22)

Without loss of generality (WLOG), we may assume that j=1. Now, we divide our arguments into two main cases as follows:

Case 11D¯(Gn,G0)i𝒜1πin(Δci1nr¯(|𝒜1|)+ΔΓi1nr¯(|𝒜1|)/2)↛0.

Here, we continue to split this case into two possibilities:

Case 1.11D¯(Gn,G0)i𝒜1πin(Δci1nr¯(|𝒜1|)+((ΔΓi1n)(uu))u=1dr¯(|𝒜1|)/2)↛0.

In this case, it must hold for some index u[d] that

1D¯(Gn,G0)i𝒜1πin(|(Δci1n)(u)|r¯(|𝒜1|)+|(ΔΓi1n)(uu)|r¯(|𝒜1|)/2)↛0. (23)

WLOG, we assume that u=1 throughout case 1.1. In the representation of An, we consider the following coefficient:

T𝟎d,τ1,0(1)=i𝒜1α1,α2:τ(α1,α2)=τ1πin2|α2|α1!α2!(Δci1n)α1(ΔΓi1n)α2, (24)

where τ1d such that τ1(u)=0 for all u=2,,d. Thus, the constraint τ(α1,α2)=τ1 holds if and only if α1(u)=α2(u1)=α2(1v)=α2(uv)=0 for all u,v=2,,d. Therefore, by assumption, we have

T𝟎d,τ1,0(1)D¯(Gn,G0)=1D¯(Gn,G0)i𝒜1α1(1)+2α2(11)=τ1(1)πin2α2(11)α1(1)!α2(11)!(Δci1n)α1(1)(ΔΓi1n)α2(11)0. (25)

Collect results in equations (23) and (25), we obtain that

i𝒜1α1(1)+2α2(11)=τ1(1)πin2α2(11)α1(1)!α2(11)!(Δci1n)α1(1)(ΔΓi1n)α2(11)i𝒜1πin(|(Δci1n)(1)|r¯(|𝒜1|)+|(ΔΓi1n)(11)|r¯(|𝒜1|)/2)0. (26)

Next, we define M¯n=max{|(Δci1n)(1)|,|(ΔΓi1n)(11)|1/2:i𝒜1} and π¯n=maxi𝒜1πin. For any i𝒜1, it is clear that the sequence of positive real numbers (πin/π¯n) is bounded, therefore, we can replace it by its subsequence that admits a non-negative limit denoted by pi2=limnπin/π¯n. In addition, let us denote (Δci1n)(1)/M¯nηi and (ΔΓi1n)(11)/2M¯n2γi. From the formulation of 𝒪k,β(Θ), since πinβ, the real numbers pi will not vanish, and at least one of them is equal to 1. Analogously, at least one of the ηi and γi is equal to either 1 or 1.

Note that i𝒜1πin(|(Δci1n)(1)|r¯(|𝒜1|)+|(ΔΓi1n)(11)|r¯(|𝒜1|)/2)/(π¯nM¯nτ1(1))↛0 for all τ1(1)[r¯(|𝒜1|)]. Thus, we are able to divide both the numerator and the denominator in equation (26) by π¯nM¯nτ1(1) and let n in order to achieve the following system of polynomial equations:

i𝒜1α1(1)+2α2(11)=τ1(1)pi2ηiα1(1)γiα2(11)α1(1)!α2(11)!=0,τ1(1)[r¯(|𝒜1|)].

However, by the definition of r¯(|𝒜1|), the above system cannot admit any non-trivial solutions, which is a contradiction. Thus, case 1.1 cannot happen.

Case 1.21D¯(Gn,G0)i𝒜1πin(((ΔΓi1n)(uv))1uvdr¯(|𝒜1|)/2)↛0.

In this case, it must hold for some indices uv that

1D¯(Gn,G0)i𝒜1πin|(ΔΓi1n)(uv)|r¯(|𝒜1|)/2↛0.

Recall that |𝒜1|>1, or equivalently, |𝒜1|2, we have that r¯(|𝒜1|)4. Therefore, the above equation leads to

1D¯(Gn,G0)i𝒜1πin|(ΔΓi1n)(uv)|2↛0. (27)

WLOG, we assume that u=1 and v=2 throughout case 1.2. We continue to consider the coefficient T𝟎d,τ1,0 in equation (24) with τ1=(2,2,0,,0)d. By assumption, we have T𝟎d,τ1,0/D¯(Gn,G0)0, which together with equation (27) imply that

i𝒜1α1,α2:τ(α1,α2)=τ1πin2|α2|α1!α2!(Δci1n)α1(ΔΓi1n)α2i𝒜1πin|(ΔΓi1n)(12)|20. (28)

Similarly, by combining the fact that case 1.1 does not hold and the result in equation (27), we get

i𝒜1πin(Δci1nr¯(|𝒜1|)+((ΔΓi1n)(uu))u=1dr¯(|𝒜1|)/2)i𝒜1πin|(ΔΓi1n)(12)|20.

Since r¯(|𝒜1|)4, the above limit indicates that any terms in equation (28) with α1(u)>0 and α2(uu)>0 for u{1,2} will vanish. Consequently, we deduce from equation (28) that

1=i𝒜1πin|(ΔΓi1n)(12)|2i𝒜1πin|(ΔΓi1n)(12)|20,

which is a contradiction. Thus, case 1.2 cannot happen.

Case 21D¯(Gn,G0)i𝒜1πin(|Δbi1n|r¯(|𝒜1|)+|Δνi1n|r¯(|𝒜1|)/2)↛0.

In this case, we consider the coefficient T𝟎d,𝟎d,0(1) in the formulation of An. By assumption,

T𝟎d,𝟎d,0(1)D¯(Gn,G0)=1D¯(Gn,G0)i𝒜1πinα4,α5:α4+2α5=τ2(Δbi1n)α4(Δνi1n)α52α5α4!α5!0.

Consequently, we obtain that

i𝒜1πinα4,α5:α4+2α5=τ2(Δbi1n)α4(Δνi1n)α52α5α4!α5!i𝒜1πin(|Δbi1n|r¯(|𝒜1|)+|Δνi1n|r¯(|𝒜1|)/2)0.

By employing the same arguments for showing that the equation (26) does not hold in case 1.1, we obtain that the above limit does not hold, either. Thus, case 2 cannot happen.

From the above results of the two main cases, we conclude that not all the coefficients in the representations of An/D¯(Gn,G0), Bn/D¯(Gn,G0) and En/D¯(Gn,G0) vanish as n.

Step 3 - Application of Fatou’s lemma: Subsequently, we denote by mn the maximum of the absolute values of the coefficients in the representations of An/D¯(Gn,G0), Bn/D¯(Gn,G0) and En/D¯(Gn,G0), that is,

mn:=max(α3,τ1,τ2,j)𝒮|Tα3,τ1,τ2n(j)|/D¯(Gn,G0),

where the constraint set 𝒮 is defined as

𝒮:={(α3,τ1,τ2,j)d×d××[k0]:0|α3|r¯(|𝒜j|),0|τ1|,τ22(r¯(|𝒜j|)|α3|)}.

Additionally, we define Tα3,τ1,τ2n(j)/mnξα3,τ1,τ2(j) as n for all (α3,τ1,τ2,j)𝒮. Since not all the coefficients in the representations of An/D¯(Gn,G0), Bn/D¯(Gn,G0) and En/D¯(Gn,G0) vanish as n, at least one among ξα3,τ1,τ2(j) is different from zero and mn↛0. Then, by applying the Fatou’s lemma, we get that

0=limn1mn2V(pGn,pG)D¯(Gn,G0)lim infn1mn|pGn(X,Y)pG(X,Y)|D¯(Gn,G0)d(X,Y)0.

Moreover, by definition, we have

1mn pGn(X,Y)pG(X,Y)D¯(Gn,G0)
(α3,τ1,τ2,j)𝒮ξα3,τ1,τ2(j)Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0).

As a consequence, we achieve that

(α3,τ1,τ2,j)𝒮ξα3,τ1,τ2(j)Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0)=0,

for almost surely (X,Y). Since elements of the set defined in equation (18) are linearly independent (proof of this claim is deferred to the end of this proof), the above equation implies that ξα3,τ1,τ2(j)=0 for all (α3,τ1,τ2,j)𝒮, which contradicts the fact that at least one among ξα3,τ1,τ2(j) is different from zero. Hence, we reach the conclusion in equation (16), which indicates that there exists some ε0>0 such that

infG𝒪k,β(Θ)D¯(G,G0)ε0V(pG,pG0)/D¯(G,G0)>0.

Global bound: Given the above result, in order to achieve the inequality in equation (15), we only need to prove its following global version:

infG𝒪k,β(Θ)D¯(G,G0)>ε0V(pG,pG0)/D¯(G,G0)>0.

Assume by contrary that the above claim is not true. Then, there exists a sequence Gn𝒪k,β(Θ) such that V(pGn,pG0)/D¯(Gn,G0)0 and D¯(Gn,G0)>ε0 for all n. Since the set Θ is compact, we can replace Gn by its subsequence that converges to some mixing measure G𝒪k,β(Θ). Consequently, we deduce that D¯(G,G0)=limnD¯(Gn,G0)ε0. This result together with the fact that V(pGn,pG0)/D¯(Gn,G0)0 lead to the limit V(pGn,pG0)0 as n. Again, by applying the Fatou’s lemma, we obtain that

0=limn2V(pGn,pG0) lim infn|pGn(X,Y)pG0(X,Y)|d(X,Y)
=|pG(X,Y)pG0(X,Y)|d(X,Y)0.

As a consequence, we have that pG(X,Y)=pG0(X,Y) for almost surely (X,Y). Due to the identifiability of the model, this equality leads to GG0, which contradicts the bound D¯(G,G0)ε0>0. Hence, we achieve the conclusion in equation (15).

Linear independence of elements in : For completion, we will demonstrate elements of the set defined in equation (18) are linearly independent by definition. In particular, assume that there exist real numbers ξα3,τ1,τ2(j), where (α3,τ1,τ2,j)𝒮, such that the following equation holds for almost surely (X,Y):

(α3,τ1,τ2,j)𝒮ξα3,τ1,τ2(j)Xα3|τ1|fcτ1(X|cj0,Γj0)|α3|+τ2f𝒟h1|α3|+τ2(Y|(aj0)X+bj0,νj0)=0.

Now, we rewrite the above equation as follows:

j=1k0ω=02r¯(|𝒜j|)(|α3|+τ2=ω|τ1|=02(r¯(|𝒜j|)α3)τ2 ξα3,τ1,τ2(j)Xα3|τ1|fcτ1(X|cj0,Γj0))
×ωf𝒟h1ω(Y|(aj0)X+bj0,νj0)=0, (29)

for almost surely (X,Y). As (aj0,bj0,νj0) for j[k0] are k0 distinct tuples, we deduce that ((aj0)X+bj0,νj0) for j[k0] are also k0 distinct tuples for almost surely X. Thus, for almost surely X, one has ωf𝒟h1ω(Y|(aj0)X+bj0,νj0) for j[k0] and 0ω2r¯(|𝒜j|) are linearly independent with respect to Y. Given that result, the equation (B.1) indicates that for almost surely X,

|α3|+τ2=ω|τ1|=02(r¯(|𝒜j|)ω)ξα3,τ1,τ2(j)Xα3|τ1|fcτ1(X|cj0,Γj0)=0,

for all j[k0] and 0ω2r¯(|𝒜j|). Note that for each j[k0] and 0ωr¯(|𝒜j|), the left hand side of the above equation can be viewed as a high-dimensional polynomial of two random vectors X and Xcj0 (cj0𝟎d) in 𝒳, which is a compact set in d. As a result, the above equation holds when ξα3,τ1,τ2(j)=0 for all j[k0], 0ω2r¯(|𝒜j|), |α3|+τ2=ω and |τ1|2(r¯(|𝒜j|)α3)τ2. This is equivalent to ξα3,τ1,τ2(j)=0 for all (α3,τ1,τ2,j)𝒮.

Hence, we conclude that the elements of are linearly independent.

B.2 Proof of Theorem 2

In order to reach the conclusion in Theorem 2, we only need to demonstrate the following inequality:

infG𝒪k,β(Θ)V(pG,pG0)/D~(G,G0)>0. (30)

In this proof, we will only prove the following local version of inequality (30) while the global version can be argued in the same fashion as in Appendix (B.1):

limε0infG𝒪k,β(Θ)D~(G,G0)εV(pG,pG0)/D~(G,G0)>0. (31)

Assume that the claim in equation (31) is not true. This indicates that we can find a sequence of mixing measures Gn=i=1knπinδ(ci,Γin,ain,bin,νin)𝒪k,β(Θ) that satisfies: D~(Gn,G0)0 and V(pGn,pG0)/D~(Gn,G0)0 as n. Additionally, since knk for all n, we are able to replace (Gn) by its subsequence which admits a fixed number of atoms knkk and 𝒜j=𝒜jn is independent of n for all j[k0].

Step 1 - Taylor expansion for density decomposition: Next, we take into account the quantity

pGn(X,Y)pG0(X,Y)
=j:|𝒜j|>1i𝒜jπin[f(X|cin,Γin)f𝒟(Y|ainX+bin,νin)f(X|cj0,Γj0)f𝒟(Y|aj0X+bj0,νj0)]
+j:|𝒜j|=1i𝒜jπin[f(X|cin,Γin)f𝒟(Y|ainX+bin,νin)f(X|cj0,Γj0)f𝒟(Y|aj0X+bj0,νj0)]
+j=1k0(i𝒜jπinπj0)f(X|cj0,Γj0)f𝒟(Y|aj0X+bj0,νj0)
:=An+Bn+En.

For each j[k0]:|𝒜j|>1, by means of Taylor expansion up to the r~(|𝒜j|)-th order, An can be rewritten as follows with a note that α=(α1,α2,α3,α4,α5)5:

An =j:|𝒜j|>1i𝒜jπin|α|=1r~(|𝒜j|)1α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5
×α1+α2fcα1Γα2(X|cj0,Γj0)α3+α4+α5faα3bα4να5(Y|aj0X+bj0,νj0)+R1(X,Y)
=j:|𝒜j|>1i𝒜jπin|α|=1r~(|𝒜j|)1α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5
×12α2α1+2α2fcα1+2α2(X|cj0,Γj0)Xα32α5α3+α4+2α5f𝒟h1α3+α4+2α5(Y|aj0X+bj0,νj0)+R3(X,Y),

where R3(X,Y) is Taylor remainder such that R3(X,Y)/D~(Gn,G0)0. Since cj0 is equal to zero when j[k~] and different from zero otherwise, the formulation of α1+2α2fcα1+2α2(X|cj0,Γj0) will vary when j[k~] compared to k~+1jk0. Thus, we will consider these two cases of j separately.

For j[k~], when α1 is an even integer, we have

α1+2α2fcα1+2α2(X|cj0,Γj0)={w=0α1/2+α2t2w,α1+2α2X2w,j[k~]w=0α1/2+α2s2w,α1+2α2(Xcj0)2w,k~+1jk0.

On the other hand, when α1 is an odd integer, we get

α1+2α2fcα1+2α2(X|cj0,Γj0)={w=0(α11)/2+α2t2w+1,α1+2α2X2w+1,j[k~]w=0(α11)/2+α2s2w+1,α1+2α2(Xcj0)2w+1,k~+1jk0.

By combining both cases, we rewrite An as follows:

An =j:|𝒜j|>1,j[k~]i𝒜j1+2=12r~(|𝒜j|)α1,2πin2α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5t1α3,α1+2α2X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)
+ j:|𝒜j|>1k~+1jk0i𝒜jα3=0r~(|𝒜j|)τ1+τ2=02(r~(|𝒜j|)α3)α1+2α2=τ1α4+2α5=τ2πin2α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5×Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0)+R3(X,Y), (32)

where for any 012r¯(|𝒜j|) and 022r¯(|𝒜j|)1, we define

1,2:={α=(αi)i=155: α1+2α2+α31,α3+α4+2α5=2,
1α1+α2++α5r~(|𝒜j|)}.

Regarding the formulation of Bn, for each j[k0]:|𝒜j|=1, we perform a Taylor expansion up to the first order and obtain that

Bn =j:|𝒜j|=1,j[k~]i𝒜j1+2=12α1,2πin2α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5t1α3,α1+2α2X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)
+j:|𝒜j|=1k~+1jk0i𝒜jα3=02τ1+τ2=02(1α3)α1,α2:α1+2α2=τ1α4,α5:α4+2α5=τ2πin2α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3
×(Δbijn)α4(Δνijn)α5×Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0)+R4(X,Y), (33)

where R4(X,Y) is a Taylor remainder such that R4(X,Y)/D~(Gn,G0)0 as n.

From equations (32) and (B.2), we can treat An/D~(Gn,G0), Bn/D~(Gn,G0) and En/D~(Gn,G0) as linear combinations of elements of the following set:

: ={X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0):j[k~],01+22r~(|𝒜j|)}
{Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0):k~+1jk0, 0α3r~(|𝒜j|),
0τ1+τ22(r~(|𝒜j|)α3)}. (34)

For any (j,1,2)𝒬:={(j,1,2)3:j[k~],01+22r~(|𝒜j|)}, let Q1,2n(j) be the coefficient of

X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)

in the representations of An, Bn and En. It follows from equations (32) and (B.2) that Q1,2n(j) is given by

Q1,2n(j)={α1,2i𝒜jπint1α3,α1+2α22α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5(1,2)(0,0),i𝒜jπinπj0,(1,2)=(0,0).

Meanwhile, we denote by Tα3,τ1,τ2n(j) the coefficient of

Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0),

for all (j,α3,τ1,τ2)𝒯:={(j,α3,τ1,τ2)3:k~+1jk0,0α3r~(|𝒜j|),0τ1+τ22(r~(|𝒜j|)α3)}. Thus, Tα3,τ1,τ2n(j) is represented as

Tα3,τ1,τ2n(j)={α1+2α2=τ1,α4+2α5=τ2i𝒜jπin2α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5(α3,τ1,τ2)(0,0,0),i𝒜jπinπj0,(α3,τ1,τ2)=(0,0,0).

Step 2 - Proof of non-vanishing coefficients by contradiction: Assume by contrary that all the coefficients of elements in the set in the representations of An/D~(Gn,G0), Bn/D~(Gn,G0) and En/D~(Gn,G0) vanish when n tends to infinity. It is worth noting that for (1,2)(0,0), we have 1+1,21,2 and

𝒥1,2:=1,21+1,2={(α1,,α5)5: α1+2α2+α3=1,α3+α4+2α5=2,
1α1+α2++α5r~(|𝒜j|)}.

Since Q1,2n(j)/D~(Gn,G0)0 for all tuples (j,1,2)𝒬, we achieve that

S1,2n(j)D~(Gn,G0):=Q1,2n(j)Q1+1,2(j)D~(Gn,G0)
=α𝒥1,2i𝒜jπint1α3,α1+2α22α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5D~(Gn,G0)
=α𝒥1,2i𝒜jπin(Γj0)α1+2α22α2+α5α!(Δcijn)α1(ΔΓijn)α2(Δaijn)α3(Δbijn)α4(Δνijn)α5D~(Gn,G0)
0,

where the third inequality follows from the fact that t1α3,α1+2α2=tα1+2α2,α1+2α2=(Γj0)(α1+2α2). Additionally, we also let S0,0(j):=Q0,0(j) for all j[k~].

By assumption, |S0,0(j)|/D~(Gn,G0)0 for all j[k~] and |T0,0,0(j)|/D~(Gn,G0)0 for all k~+1jk0 as n. By taking the summation of all such terms, we get that

1D~(Gn,G0)j=1k0|i𝒜jπinπj0|0. (35)

Next, we consider indices j[k0]:|𝒜j|=1, i.e. those in the formulation of Bn. For j[k~], since |S1,2n(j)|/D~(Gn,G0)0 for all (1,2){(1,0),(0,1),(1,1),(2,0),(0,2)}, we get that

1D~(Gn,G0)j:|𝒜j|=1j[k~]i𝒜jπin(|Δcijn|+|ΔΓijn|+|Δaijn|+|Δbijn|+|Δνijn|)0. (36)

Moreover, for k~+1jk0, as |Tα3,τ1,τ2n(j)|/D~(Gn,G0)0 for all (α3,τ1,τ2){(0,1,0),(0,2,0),(1,0,0),(0,0,1),(0,0,2)}, we deduce that

1D~(Gn,G0)j:|𝒜j|=1k~+1jk0i𝒜jπin(|Δcijn|+|ΔΓijn|+|Δaijn|+|Δbijn|+|Δνijn|)0. (37)

Let us denote

Kijn(κ1,κ2,κ3,κ4,κ5):=|Δcijn|κ1+|ΔΓijn|κ2+|Δaijn|κ3+|Δbijn|κ4+|Δνijn|κ5.

Then, equations (36) and (37) indicates that

1D~(Gn,G0)j:|𝒜j|=1i𝒜jπinKijn(1,1,1,1,1)0. (38)

Additionally, since |T2,0,0(j)|/D~(Gn,G0)0 for all k~+1jk0, we have that

1D~(Gn,G0)j:|𝒜j|>1i𝒜jπin|Δaijn|20. (39)

Putting the results in equations (35), (38) and (39) together with the formulation of D~(Gn,G0) in equation (3.2), we obtain that

1D~(Gn,G0) [j:|𝒜j|>1j[k~]i𝒜jπinKijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)
+j:|𝒜j|>1k~+1jk0i𝒜jπinK3,ijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)]1, (40)

where K3,ijn(κ1,κ2,κ3,κ4,κ5):=|Δcijn|κ1+|ΔΓijn|κ2+|Δbijn|κ4+|Δνijn|κ5. Now, we will divide our arguments into two main scenarios based on the above limit:

Case 1: j:|𝒜j|>1j[k~]i𝒜jπinKijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)D~(Gn,G0)↛0.

This assumption indicates that we can find an index j[k~]:|𝒜j|>1 such that

1D~(Gn,G0)i𝒜jπinKijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)↛0.

WLOG, we may assume that j=1 throughout this case. Recall that S1,2n(1)/D~(Gn,G0)0 for all pairs (1,2) such that 01+22r~(|𝒜1|). Combine this result with the assumption of case 1, we obtain

S1,2n(1)D1(Gn,G0)=S1,2n(1)D~(Gn,G0)D~(Gn,G0)D1(Gn,G0)0,

where D1(Gn,G0):=i𝒜1πinKi1n(r~(|𝒜1|),r~(|𝒜1|)2,r~(|𝒜1|)2,r~(|𝒜1|),r~(|𝒜1|)2). By expanding the formulations of S1,2n(1) and D1(Gn,G0), we have that

i𝒜1α𝒥1,2πin(Γ10)α1+2α22α2+α5α!(Δci1n)α1(ΔΓi1n)α2(Δai1n)α3(Δbi1n)α4(Δνi1n)α5i𝒜1πin[|Δci1n|r~(|𝒜1|)+|ΔΓi1n|r~(|𝒜1|)2+|Δai1n|r~(|𝒜1|)2+|Δbi1n|r~(|𝒜1|)+|Δνi1n|r~(|𝒜1|)2]0. (41)

Next, we define M¯n:=max{|Δci1n|,|ΔΓi1n|1/2,|Δai1n|1/2,|Δbi1n|,|Δνi1n|1/2:i𝒜1} and π¯n:=maxi𝒜1πin. For any i𝒜1, since the sequence (πin/π¯n)i𝒜1 is bounded, we can substitute it with its subsequence that admits a non-negative limit pi2=limnπin/π¯n.

Additionally, we define (Δci1n)/(Γj0M¯n)q1i, (ΔΓi1n)/[2(Γj0)2M¯n2]q2i, (Δai1n)/M¯n2q3i, (Δbi1n)/M¯nq4i and (Δνi1n)/2M¯n2q5i. It can be seen from the formulation of 𝒪k,β(Θ) that πinδ, therefore, pi’s will not vanish and at least one of them is equal to 1. Similarly, at least one of the limits q1i,q2i,,q5i will be equal to either 1 or 1.

Since

i𝒜1πin[|Δci1n|r~(|𝒜1|)+|ΔΓi1n|r~(|𝒜1|)2+|Δai1n|r~(|𝒜1|)2+|Δbi1n|r~(|𝒜1|)+|Δνi1n|r~(|𝒜1|)2]π¯nM¯n1+2↛0,

for all pairs (1,2) such that 1+2[r~(|𝒜1|)], we can divide both the numerator and the denominator in equation (41) by π¯nM¯n1+2, and then let n to achieve the following system of polynomial equations:

i𝒜1α𝒥1,2pi2q1iα1q2iα2q3iα3q4iα4q5iα5α1!α2!α3!α4!α5!=0,

for all pairs (1,2) such that 01+2r~(|𝒜1|). Nevertheless, according to the definition of r~(|𝒜1|), the above system cannot admit any non-trivial solutions, which is a contradiction. Thus, case 1 does not hold.

Case 2: 1D~(Gn,G0)j:|𝒜j|>1k~+1jk0i𝒜jπinK3,ijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)↛0.

This assumption implies that there exists an index k~+1jk0:|𝒜j|>1 such that

1D~(Gn,G0)i𝒜jπinK3,ijn(r~(|𝒜j|),r~(|𝒜j|)2,r~(|𝒜j|),r~(|𝒜j|)2)↛0. (42)

By applying similar arguments for equation (22) in the proof of Theorem 1 to equation (42), we are able to point out that equation (42) cannot happen, which is a contradiction. As a result, case 2 cannot happen either.

Collect the results of the above two scenarios, we realize that the limit in equation (40) does not hold true, which is a contradiction. As a consequence, not all the coefficients of elements in the set , defined in equation (B.2), in the representations of An/D~(Gn,G0), Bn/D~(Gn,G0) and En/D~(Gn,G0) go to zero as n.

Step 3 - Application of Fatou’s lemma: Next, we denote by mn the maximum of the absolute values of those coefficients, which means that

mn:=max{max(j,1,2)𝒬|Q1,2n(j)|D~(Gn,G0),max(j,α3,τ1,τ2)𝒯|Tα3,τ1,τ2n(j)|D~(Gn,G0)}.

In addition, let us define Q1,2n(j)/mnζ1,2(j) for (j,1,2)𝒬 and Tα3,τ1,τ2n(j)ξα3,τ1,τ2(j) for (j,α3,τ1,τ2)𝒯 as n. As not all the coefficients of elements of in the representations of An/D~(Gn,G0), Bn/D~(Gn,G0) and En/D~(Gn,G0) vanish as n, at least one among ζ1,2(j) and ξα3,τ1,τ2(j) is different from zero and mn↛0. By invoking the Fatou’s lemma, we get that

0=limn1mn2V(pGn,pG)D~(Gn,G0)lim infn1mn|pGn(X,Y)pG(X,Y)|D~(Gn,G0)d(X,Y)0.

Furthermore, we have that

1mn pGn(X,Y)pG(X,Y)D~(Gn,G0)
(j,1,2)𝒬ζ1,2(j)X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)
+(j,α3,τ1,τ2)𝒯ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0).

Consequently, we achieve that

(j,1,2)𝒬ζ1,2(j)X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)
+(j,α3,τ1,τ2)𝒯ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0)=0,

for almost surely (X,Y). Since elements of the set defined in equation (B.2) are linearly independent (proof of this claim is deferred to the end of this proof), the above equation indicates that ζj,1,2(j)=ξα3,τ1,τ2(j)=0 for all (j,1,2)𝒬 and (j,α3,τ1,τ2)𝒯, which contradicts the fact that at least one among ζj,1,2(j), ξα3,τ1,τ2(j) is different from zero. Hence, we reach the conclusion in equation (31).

Linear independence of elements in : For completion, we will show that elements of the set defined in equation (B.2) are linearly independent by definition. In particular, assume that there exist real numbers ζ1,2(j) and ξα3,τ1,τ2(j), where (j,1,2)𝒬 and (j,α3,τ1,τ2)𝒯, such that the following equation holds for almost surely (X,Y):

(j,1,2)𝒬ζ1,2(j)X12f𝒟h12(Y|aj0X+bj0,νj0)f(X|cj0,Γj0)
+(j,α3,τ1,τ2)𝒯ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0)α3+τ2f𝒟h1α3+τ2(Y|aj0X+bj0,νj0)=0,

Now, we rewrite the above equation as follows:

j=1k0ω=02r~(|𝒜j|)[α3+τ2=ωτ1=02(r~(|𝒜j|)α3)τ2ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0)𝟏{k~+1jk0}
+1=02r~(|𝒜j|)ωζ1,ω(j)X1f(X|cj0,Γj0)𝟏{j[k~]}]ωf𝒟h1ω(Y|aj0X+bj0,νj0)=0, (43)

for almost surely (X,Y). As (aj0,bj0,νj0) for j[k0] are k0 distinct tuples, we deduce that ((aj0)X+bj0,νj0) for j[k0] are also k0 distinct tuples for almost surely X. Thus, for almost surely X, one has ωf𝒟h1ω(Y|(aj0)X+bj0,νj0) for j[k0] and 0ω2r~(|𝒜j|) are linearly independent with respect to Y. Given that result, the equation (B.2) indicates that for almost surely X,

α3+τ2=ωτ1=02(r~(|𝒜j|)α3)τ2ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0)𝟏{k~+1jk0}
+1=02r~(|𝒜j|)ωζ1,ω(j)X1f(X|cj0,Γj0)𝟏{j[k~]}=0.

for all j[k0] and 0ω2r~(|𝒜j|). This equation is equivalent to

1=02r~(|𝒜j|)ωζ1,ω(j)X1f(X|cj0,Γj0) =0, (44)
α3+τ2=ωτ1=02(r~(|𝒜j|)α3)τ2ξα3,τ1,τ2(j)Xα3τ1fcτ1(X|cj0,Γj0) =0, (45)

for all j[k~], 0ω2r~(|𝒜j|) and k~+1jk0, 0ω2r~(|𝒜j|). We can treat the left hand side of equation (44) as a polynomial of the random vector X𝒳, which is a compact set in . Meanwhile, the left hand side of equation (45) can be viewed as another polynomial of X and Xcj0, where cj00. As a result, the above equations hold when ζ1,ω(j)=0 for all j[k~], 0ω2r~(|𝒜j|), 012r~(|𝒜j|)ω, and ξα3,τ1,τ2(j)=0 for all k~+1jk0, 0ω2r~(|𝒜j|), α3+τ2=ω and 0τ12(r~(|𝒜j|)α3)τ2. This result is equivalent to ζ1,2(j)=0, for all (j,1,2)𝒬 and ξα3,τ1,τ2(j)=0 for all (j,α3,τ1,τ2)𝒯.

Hence, the elements of are linearly independent, which completes the proof.

Appendix C PROOF OF REMAINING RESULTS

In this appendix, we provide proofs for Proposition 1, Proposition 2 and Lemma 2 in that order.

C.1 Proof of Proposition 1

For any two mixing measures G=i=1kπiδ(ci,Γi,ai,bi,νi) and G=i=1kπiδ(ci,Γi,ai,bi,νi), we assume that pG(X,Y)=pG(X,Y) holds true for almost surely (X,Y)𝒳×𝒴, or equivalently,

i=1kπif(X|ci,Γi)f𝒟(Y|(ai)X+bi,νi)=i=1kπif(X|ci,Γi)f𝒟(Y|(ai)X+bi,νi). (46)

Recall that if Y|X𝒩1(aX+b,ν) and X𝒩d(c,Γ), then

(XY)𝒩d+1((cac+b),(ΓΓaaΓaΓa+ν)).

Let us denote

ψi:=(ci(ai)ci+bi),Σi:=(ΓiΓiai(ai)Γi(ai)Γiai+νi),
ψi:=(ci(ai)ci+bi),Σi:=(ΓiΓiai(ai)Γi(ai)Γiai+νi).

Then, equation (46) can be rewritten as

i=1kπif(X,Y|ψi,Σi)=i=1kπif(X,Y|ψi,Σi), (47)

for almost surely (X,Y), where f belongs to the family of (d+1)-dimensional Gaussian density functions. Since the location-scale Gaussian mixtures are identifiable, it follows from the above equation that k=k and {π1,π2,,πk}{π1,π2,,πk}. WLOG, we may assume that πi=πi for any i[k].

Subsequently, we construct a partition of the set [k], denoted by P1,P2,,Pm that satisfies the following properties:

  • (i)

    πi=πi for any iP and [m];

  • (ii)

    πiπj if i and j are not in the same set P for any [m].

Given this partition, we represent equation (47) as follows:

=1miPπif(X,Y|ψi,Σi)==1miPπif(X,Y|ψi,Σi),

for almost surely (X,Y). Consequently, for each [m], we obtain that

{(ψi,Σi):iP}{(ψi,Σi):iP}.

WLOG, we may assume that (ψi,Σi)=(ψi,Σi) for any iP. Given this result, by some simple algebraic derivations, we achieve that (ci,Γi,ai,bi,νi)=(ci,Γi,ai,bi,νi) for any iP and [m]. As a result, it follows that

G==1miPπiδ(ci,Γi,ai,bi,νi)==1miPπiδ(ci,Γi,ai,bi,νi)=G.

Hence, the proof is completed.

C.2 Proof of Proposition 2

Prior to presenting the proof of Proposition 2, let us review fundamental background on density estimation for M-estimators, which is covered in [53]. First of all, we define 𝒫k,β(Θ):={pG(X,Y):G𝒪k,β(Θ)} as the set of joint densities of all mixing measure in 𝒪k,β(Θ). In addition, we denote

𝒬k,β(Θ) :={p(G+G0)/2(X,Y):G𝒪k,β(Θ)},
𝒬k,β1/2(Θ) :={p(G+G0)/21/2(X,Y):G𝒪k,β(Θ)}.

Subsequently, for any δ>0, the Hellinger ball centered around the density pG0(X,Y) and intersected with the set 𝒬k,β1/2(Θ) is defined as

𝒬k,β1/2(Θ,δ):={g1/2𝒬k,β1/2(Θ):h(g,pG0)δ}.

Additionally, Geer et al. [53] introduce the following quantity to capture the size of the above Hellinger ball:

𝒥B(δ,𝒬k,β1/2(Θ)):=δ2/213δHB1/2(u,𝒬k,β1/2(Θ,u),)duδ, (48)

where HB1/2(u,𝒬k,β1/2(Θ,u),) denotes the bracketing entropy of 𝒬k,β1/2(Θ,u) under the Euclidean distance, and uδ:=max{u,δ}. Given these notations, let us state the result regarding the joint density estimation rate presented in Theorem 7.4 in [53].

Lemma 3 (Theorem 7.4, [53]).

Take Ψ(δ)𝒥B(δ,𝒬k,β1/2(Θ)) such that Ψ(δ)/δ2 is a non-increasing function of δ. Then, for a universal constant c and a sequence (δn) that satisfies nδn2cΨ(δn), we obtain that

(h(pG^n,pG0)>δ)cexp(nδ2c2),

for any δδn.

Proof of Lemma 3 is provided in [53]. Next, we introduce the upper bounds of the covering number (under the sup norm) N(ε,𝒫k,β(Θ),), and the bracketing entropy (under the Hellinger distance) HB(ε,𝒫k,β(Θ),h) of the metric space 𝒫k,β(Θ). For further detail about the definitions of these terms, readers are referred to [53].

Lemma 4.

Given a bounded set Θ, we have for any ε[0,1/2] that

  • (i)

    logN(ε,𝒫k,β(Θ),)log(1/ε);

  • (ii)

    HB(ε,𝒫k,β(Θ),h)log(1/ε).

Proof of Lemma 4 is relegated to Appendix C.2.2. Now, we already have all necessary ingredients to provide the proof for Proposition 2 in Appendix C.2.1

C.2.1 Proof of Proposition 2

Note that for any u>0, we have

HB(u,𝒬k,β1/2(Θ),)HB(u,𝒫k,β(Θ),h)log(1/u),

where the second inequality is induced by part (ii) of Lemma 4. Then, it follows from equation (48) that

𝒥B(δ,𝒬k,β1/2(Θ))δ2/213δlog(1/u)duδ. (49)

By choosing Ψ(δ):=δ[log(1/δ)]1/2, we get that Ψ(δ)/δ2 is a non-increasing function of δ and Ψ(δ)𝒥B(δ,𝒬k,β1/2(Θ)) from equation (49). Let δn:=log(n)/n, we achieve that nδn2cΨ(δn) for some universal constant c. As a result, Lemma 3 gives us that

(h(pG^n,pG0)>C1log(n)/n)exp(C2log(n))=nC2,

where C1 and C2 are some universal constants. Finally, since the Total Variation is upper bounded by the Hellinger distance, we obtain the desired conclusion.

C.2.2 Proof of Lemma 4

Part (i). Given some ε>0, since Θ is a compact set, we can find an ε-cover of Θ, denoted by Θε. Additionally, let Δε be an ε-cover of an (k1)-dimensional simplex. Assume that |Θε|=T and |Δε|=S. Note that Θd×𝒮d+×d××+ is a subspace of d2+4d, then it can be checked that T=𝒪(ε(d2+4d)k) and S=𝒪(ε(k1)). Next, we define

𝒢:={pG𝒫k,β(Θ):(π1,π2,,πk)Δε,(ci,Γi,ai,bi,νi)Θε}.

Given some mixing measure G=i=1kπiδθi𝒪k,β(Θ) with kk and θi:=(ci,Γi,ai,bi,νi)Θ, let us consider G¯=i=1kπiδθ~i where θ~i:=(c~i,Γ~i,a~i,b~i,ν~i)Θε such that θ~iθiε for any i[k]. In addition, we also take into account another mixing measure G~=i=1kπ~iδθ~i where (π~1,π~2,,π~k,0,,0)Δε such that (π~i)i=1k(πi)i=1kε. From the definition of 𝒢, we get that pG~𝒢. Since (π~i)i=1k(πi)i=1kε, we can deduce that

pG¯pG~i=1k|π~iπi|f(X|c~i,Γ~i)f𝒟(Y|(a~i)X+b~i,ν~i)ε.

Next, we consider

pGpG¯i=1kπiF(θi|X,Y)F(θ~i|X,Y),

where we denote F(θ|X,Y):=f(X|c,Γ)f𝒟(Y|aX+b,ν). As F is twice differentiable with respect to θ and 𝒳 is a bounded set, we achieve the following inequality:

i=1kπiF(θi|X,Y)F(θ~i|X,Y)i=1kπθ~iθiε,

which leads to pGpG¯ε. As a consequence, by the triangle inequality, we have

pGpG~pGpG¯+pG¯pG~ε.

Given this result, it follows that 𝒢 is an ε-cover of 𝒫k,β(Θ), therefore,

N(ε,𝒫k,β(Θ),)|𝒢|=S×T=𝒪(ε(d2+4d)k)×𝒪(ε(k1))=𝒪(ε(d2+4d+1)k+1),

which implies that logN(ε,𝒫k,β(Θ),)log(1/ε).

Part (ii). We begin with finding an upper bound for the density f(X|c,Γ)f𝒟(Y|aX+b,ν). Since 𝒳 and Θ are bounded sets, we can find positive constants u,u1,u2,u3,l1,l3 such that cu, l1λmin(Γ)λmax(Γ)u1, u2aX+bu2 and l3νu3, where λmin(Γ) and λmax(Γ) are the smallest and the largest eigenvalues of Γ, respectively. Firstly, it is clear that

f(X|c,Γ)=1(2π)ddet(Γ)exp(12(xc)Γ1(xc))1(2πl1)d/2.

Additionally, note that

(Xc)Γ1(xc)λmin(Γ1)Xc2=1λmax(Γ)Xc2.

Moreover, for any X2u, by the Cauchy-Schwartz inequality, we get

4Xc2X2=3X28Xc+4c23X28Xc+4c20,

which implies that (Xc)Γ1(xc)14u1X2. As a result,

f(X|c,Γ)=1(2π)ddet(Γ)exp(12(xc)Γ1(xc))1(2πl1)d/2exp(X28u1),

for any X2u. Combine this result with the previous bound, we obtain that f(X|c,Γ)G1(X), where

G1(X):={1(2πl1)d/2exp(X28u1),X2u,1(2πl1)d/2,X<2u.

By arguing in a similar fashion, we also have f𝒟(Y|aX+b,ν)G2(X,Y) where

G2(X,Y):={12πl3exp(Y28u3),|Y|2u212πl3,|Y|<2u2.

Consequently, we achieve that f(X|c,Γ)f𝒟(Y|aX+b,ν)G(X,Y):=G1(X)G2(X,Y).

Next, given some η>0 that we will choose later, we consider an η-cover of 𝒫k,β(Θ) which is assumed to have N elements denoted by f1,f2,,fN. For any i[N], we define

Li(X,Y):=max{fi(X,Y)η,0},Ui(X,Y):={fi(X,Y)+η,G(X,Y)}.

Then, we can validate that 𝒫k,β(Θ)i=1N[Li(X,Y),Ui(X,Y)] and Ui(X,Y)Li(X,Y)min{2η,G(X,Y)}. Furthermore, we also deduce that

UiLi1=(Ui(X,Y)Li(X,Y))d(X,Y)
=|Y|<2u2(Ui(X,Y)Li(X,Y))d(X,Y)+|Y|2u2(Ui(X,Y)Li(X,Y))d(X,Y)
c1η+exp(c12/(2u3))c2η,

where c1=max{2u2,8u3}log(1/η) and c2>0 is some universal constant. This means that each bracket [Li(X,Y),Ui(X,Y)] is of size c2η. Recall that the bracketing entropy is the logarithm of the smallest number of brackets to cover 𝒫k,β(Θ), it follows that

HB(c2η,𝒫k,β(Θ),1) logN(η,𝒫k,β(Θ),1)logN(η,𝒫k,β(Θ),)log(1/η),

where the second inequality occurs since 1, while the last inequality is due to the result in part (i). Moreover, as the Hellinger distance is upper bounded by the L1-norm 1, we get that

HB(c2η,𝒫k,β(Θ),h)HB(c2η,𝒫k,β(Θ),1)log(1/η).

Here, if we choose η=ε/c2, we can conclude that HB(ε,𝒫k,β(Θ),h)log(1/ε).

C.3 Proof of Lemma 2

We begin with recalling the system of interest here:

l=1mα𝒥1,2pl2q1lα1q2lα2q3lα3q4lα4q5lα5α1!α2!α3!α4!α5!=0, (50)

with unknown variables {(pl,q1l,q2l,q3l,q4l,q5l)}l=1m5 for all 10 and 20 that satisfy 11+2r, where

𝒥1,2:={α=(αi)i=155:α1+2α2+α3=1,α3+α4+2α5=2}.

Let us consider only a part of the above system when 1=0 as follows:

l=1mα4,α5α4+2α5=2pl2q4lα4q5lα5α4!α5!=0, (51)

for all 12r, which takes the same form as the system in equation (9). Thus, it follows from Lemma 1 that the smallest positive integer r such that the system (51) does not admit any non-trivial solutions is r¯(m). Therefore, we obtain that r~(m)r¯(m).

Next, we will respectively show that r~(2)=4 and r~(3)=6.

When m=2: In this case, it follows from the above result that r~(m)r¯(m)=4. Thus, it is sufficient to demonstrate r~(m)>3, i.e. pointing out a non-trivial solution for the system (50) when r=3, which is given by

l=1mpl2q1l=0,l=1mpl2q4l=0,
l=1mpl2(12!q1l2+q2l)=0,l=1mpl2(q1lq4l+q3l)=0,l=1mpl2(12!q4l2+q5l)=0,
l=1mpl2(13!q1l3+q1lq2l)=0,l=1mpl2(12!q1l2q4l+q1lq3l+q2lq4l)=0,
l=1mpl2(12!q1lq4l2+q1lq5l+q3lq4l)=0,l=1mpl2(13!q4l3+q4lq5l)=0. (52)

We can check that the following is a non-trivial solution of the system (C.3):

pl=0,q1l=q2l=q3l=0,l[m],
q41=1,q42=1,q51=q52=12.

Hence, we conclude that r~(m)=4.

When m=3: Again, according to Lemma 1, we have r~(m)r¯(m)=6. Therefore, it suffices to show a non-trivial solution of the system (50) for r=5, which is a combination of the system (C.3) and the following system:

l=1mpl2(14!q1l4+12!q1l2q2l+12!q2l2)=0,l=1mpl2(14!q4l2+12!q4l2q5l+12!q5l2)=0,
l=1mpl2(13!q1l3q4l+12!q1lq2l2+12!q1l2q2l+q2lq3l)=0,
l=1mpl2(13!q1lq4l3+12!q1lq4lq5l2+12!q3lq4l2+q3lq5l)=0,
l=1mpl2(12!2!q1l2q4l2+12!q2lq4l2+12!q1l2q5l+q2lq5l+q1lq3lq4l+12!q3l2)=0,
l=1mpl2(15!q1l5+13!q1l3q2l+12!q1lq2l2)=0,l=1mpl2(15!q4l5+13!q4l3q5l+12!q4lq5l2)=0,
l=1mpl2(14!q1l4q4l+12!q1l2q2l+12!q2l2q4l+13!q1l3q3l+q1lq2lq3l)=0,
l=1mpl2(14!q1lq4l4+12!q1lq4l2q5l+12!q1lq5l2+13!q3lq4l3+q3lq4lq5l)=0,
l=1mpl2(13!2!q1l3q4l2+13!q1l3q5l+12!q1lq2lq4l2+q1lq2lq4l+12!q1l2q3lq4l+q2lq3lq4l+12!q1lq3l2)=0,
l=1mpl2(12!3!q1l2q4l3+q1l2q4lq5l+13!q2lq3l3+q2lq4lq5l+12!q1lq3lq4l2+q1lq3lq5l+12!q3l2q4l)=0.

It can be verified that the following is a non-trivial of this system:

pl=0,q1l=q2l=q3l=0,l[m],
q41=33,q42=33,q43=0,q51=q52=16,q53=0.

As a consequence, we obtain that r~(m)>5, which implies the desired conclusion that r~(m)=6.

Cite this paper

Please cite the published version. Venue: AISTATS 2024, Proceedings of Machine Learning Research, Vol. 238 (2024). DOI: PMLR 238 (AISTATS 2024). Official record: PMLR.

BibTeX
@inproceedings{nguyen2024convergence,
  title     = {Towards convergence rates for parameter estimation in Gaussian-gated mixture of experts},
  author    = {Nguyen, Huy and Nguyen, TrungTin and Nguyen, Khai and Ho, Nhat},
  booktitle = {Proceedings of The 27th International Conference on Artificial Intelligence and Statistics (AISTATS)},
  series    = {Proceedings of Machine Learning Research}, volume = {238},
  year      = {2024}, publisher = {PMLR},
  url       = {https://openreview.net/forum?id=MSHFiXz8jS},
}