Publications · Journal

Approximations of conditional probability density functions in Lebesgue spaces via mixture of experts models

Hien D. Nguyen, TrungTin Nguyen, Faicel Chamroukhi, Geoffrey J. McLachlan

† Corresponding author.

J. Stat. Distrib. Appl. · Journal Journal of Statistical Distributions and Applications. Open-access journal article (Vol. 8, Art. 13, 2021).

Abstract

Mixture of experts (MoE) models are widely applied for conditional probability density estimation problems. We demonstrate the richness of the class of MoE models by proving denseness results in Lebesgue spaces, when inputs and outputs variables are both compactly supported. We further prove an almost uniform convergence result when the input is univariate. Auxiliary lemmas are proved regarding the richness of the soft-max gating function class, and their relationships to the class of Gaussian gating functions.

1Department of Mathematics and Statistics, La Trobe University, Bundoora Victoria, Australia.
2Normandie Univ, UNICAEN, CNRS, LMNO, 14000 Caen, France.
3School of Mathematics and Physics, University of Queensland, St. Lucia Brisbane, Australia.
Corresponding author—email: h.nguyen5@latrobe.edu.au.

Key words: mixture of experts; conditional probability density functions; approximation theory; mixture models; Lebesgue spaces

1 Introduction

Mixture of experts (MoE) models are a widely applicable class of conditional probability density approximations that have been considered as solution methods across the spectrum of statistical and machine learning problems; see, for example, the reviews of Yuksel et al., (2012), Masoudnia & Ebrahimpour, (2014), and Nguyen & Chamroukhi, (2018).

Let =𝕏×𝕐, where 𝕏d and 𝕐q, for d,q. Suppose that the input and output random variables, 𝑿𝕏 and 𝒀𝕐, are related via the conditional probability density function (PDF) f(𝒚|𝒙) in the functional class:

={f:[0,)|𝕐f(𝒚|𝒙)dλ(𝒚)=1,𝒙𝕏},

where λ denotes the Lebesgue measure. The MoE approach seeks to approximate the unknown target conditional PDF f by a function of the MoE form:

m(𝒚|𝒙)=k=1KGatek(𝒙)Expertk(𝒚),

where 𝐆𝐚𝐭𝐞=(Gatek)k[K]𝒢K ([K]={1,,K}), Expert1,,ExpertK, and K. Here, we say that m is a K-component MoE model with gates arising from the class 𝒢K and experts arising from , where is a class of PDFs with support 𝕐.

The most popular choices for 𝒢K are the parametric soft-max and Gaussian gating classes:

𝒢SK={𝐆𝐚𝐭𝐞=(Gatek(;𝜸))k[K]|k[K],Gatek(;𝜸)=exp(ak+𝒃k)l=1Kexp(al+𝒃l),𝜸𝔾SK}

and

𝒢GK={𝐆𝐚𝐭𝐞=(Gatek(;𝜸))k[K]|k[K],Gatek(;𝜸)=πkϕ(;𝝂k,𝚺k)l=1Kπlϕ(;𝝂l,𝚺l),𝜸𝔾GK},

respectively, where

𝔾SK={𝜸=(a1,,aK,𝒃1,,𝒃K)K×(d)K}

and

𝔾GK={𝜸=(𝝅,𝝂1,,𝝂K,𝚺1,,𝚺K)ΠK1×(d)K×𝕊dK}.

Here,

ϕ(;𝝂,𝚺)=|2π𝚺|1/2exp[12(𝝂)𝚺1(𝝂)]

is the multivariate normal density function with mean vector 𝝂 and covariance matrix 𝚺, 𝝅=(π1,,πK) is a vector of weights in the simplex:

ΠK1={𝝅=(πk)k[K]|k[K],πk>0,k=1Kπk=1},

and 𝕊d is the class of d×d symmetric positive definite matrices. The soft-max and Gaussian gating classes were first introduced by Jacobs et al., (1991) and Jordan & Xu, (1995), respectively. Typically, one chooses experts that arise from some location-scale class:

ψ={gψ(;𝝁,σ):𝕐[0,)|gψ(;𝝁,σ)=1σqψ(𝝁σ),𝝁q,σ(0,)},

where ψ is a PDF, with respect to q in the sense that ψ:q[0,) and qψ(𝒚)dλ(𝒚)=1.

We shall say that fp() for any p[1,) if

fp,=(|𝟏f|pdλ(𝒛))1/p<,

where 𝟏 is the indicator function that takes value 1 when 𝒛, and 0 otherwise. Further, we say that f() if

f,=inf{a0|λ({𝒛||f(𝒛)|>a})=0}<.

We shall refer to p, as the p norm on , for p[0,], and where the context is obvious, we shall drop the reference to .

Suppose that the target conditional PDF f is in the class p=p. We address the problem of approximating f, with respect to the p norm, using MoE models in the soft-max and Gaussian gated classes,

Sψ = {mKψ:[0,)|mKψ(𝒚|𝒙)=k=1KGatek(𝒙)gψ(𝒚;𝝁k,σk),
gψψ,𝐆𝐚𝐭𝐞𝒢SK,𝝁k𝕐,σk(0,),k[K],K}

and

Gψ = {mKψ:[0,)|mKψ(𝒚|𝒙)=k=1KGatek(𝒙)gψ(𝒚;𝝁k,σk),
gψψ,𝐆𝐚𝐭𝐞𝒢GK,𝝁k𝕐,σk(0,),k[K],K},

by showing that both Sψ and Gψ are dense in the class p, when 𝕏=[0,1]d and 𝕐 is a compact subset of q. Our denseness results are enabled by the indicator function approximation result of Jiang & Tanner, 1999b , and the finite mixture model denseness theorems of Nguyen et al., 2020a and Nguyen et al., 2020c .

Our theorems contribute to an enduring continuity of sustained interest in the approximation capabilities of MoE models. Related to our results are contributions regarding the approximation capabilities of the conditional expectation function of the classes Sψ and Gψ (Wang & Mendel,, 1992, Zeevi et al.,, 1998, Jiang & Tanner, 1999b, , Krzyzak & Schafer,, 2005, Mendes & Jiang,, 2012, Nguyen et al.,, 2016, 2019) and the approximation capabilities of subclasses of Sψ and Gψ, with respect to the Kullback–Leibler divergence (Jiang & Tanner, 1999a, , Norets,, 2010, Norets & Pelenis,, 2014). Our results can be seen as complements to the Kullback–Leibler approximation theorems of Norets, (2010) and Norets & Pelenis, (2014), by the relationship between the Kullback–Leibler divergence and the 2 norm (Zeevi & Meir,, 1997). That is, when f>1/κ, for all (𝒙,𝒚) and some constant κ>0, we have that the integrated conditional Kullback–Leibler divergence considered by Norets & Pelenis, (2014):

𝕏D(f(|𝒙)mKψ(|𝒙))dλ(𝒙)=𝕏𝕐f(𝒚|𝒙)logf(𝒚|𝒙)mKψ(𝒚|𝒙)dλ(𝒚)dλ(𝒙)

satisfies

𝕏D(f(|𝒙)mKψ(|𝒙))dλ(𝒙)κ2fmKψ2,2,

and thus a good approximation in the integrated Kullback–Leibler divergence is guaranteed if one can find a good approximation in the 2 norm, which is guaranteed by our main result.

The remainder of the manuscript proceeds as follows. The main result is presented in Section 2. Technical lemmas are provided in Section 3. The proofs of our results are then presented in Section 4. Proofs of required lemmas that do not appear elsewhere are provided in Section 5. A summary of our work and some conclusions are drawn in Section 6.

2 Main results

Denote the class of bounded functions on by

()={f()|a[0,), such that |f(𝒛)|a𝒛},

and write its norm as f()=sup𝒛|f(𝒛)|. Further, let 𝒞 denote the class of continuous functions. Note that if is compact and f𝒞, then f.

Theorem 1.

Assume that 𝕏=[0,1]d for d. There exists a sequence {mKψ}KSψ, such that if 𝕐q is compact, f𝒞, and ψ𝒞(q) is a PDF on support q, then limKfmKψp=0, for p[1,) .

Since convergence in Lebesgue spaces does not imply point-wise modes of convergence, the following result is also useful and interesting in some restricted scenarios. Here, we note that the mode of convergence is almost uniform, which implies almost everywhere convergence and convergence in measure (cf. Bartle, 1995, Lem 7.10 and Thm. 7.11). The almost uniform convergence of {mKψ}K to f in the following result is to be understood in the sense of Bartle, (1995, Def. 7.9). That is, for every δ>0, there exists a set 𝔼δ with λ()<δ, such that {mKψ}K converges to f, uniformly on \𝔼δ.

Theorem 2.

Assume that 𝕏=[0,1]. There exists a sequence {mKψ}KSψ, such that if 𝕐q is compact, f𝒞, and ψ𝒞(q) is a PDF on support q, then limKmKψ=f, almost uniformly.

The following result establishes the connection between the gating classes 𝒢SK and 𝒢GK.

Lemma 1.

For each K, 𝒢SK𝒢GK. Further, if we define the class of Gaussian gating vectors with equal covariance matrices:

𝒢EK={𝐆𝐚𝐭𝐞=(Gatek(;𝜸))k[K]|k[K],Gatek(;𝜸)=πkϕ(;𝝂k,𝚺)l=1Kπlϕ(;𝝂l,𝚺),𝜸𝔾EK},

where

𝔾EK={𝜸=(𝝅,𝝂1,,𝝂K,𝚺)ΠK1×(d)K×𝕊d},

then 𝒢EK𝒢SK.

We can directly apply Lemma 1 to establish the following corollary to Theorems 1 and 2, regarding the approximation capability of the class Gψ.

Corollary 1.

Theorems 1 and 2 hold when Sψ is replaced by Gψ in their statements.

3 Technical lemmas

Let 𝕂n={(k1,,kd)[n]d} and κ:𝕂n[nd] be a bijection for each n. For each (k1,,kd)𝕂n and k[nd], we define 𝕏kn=𝕏κ(k1,,kd)n=i=1d𝕀kin, where 𝕀kin=[(ki1)/n,ki/n) for ki[n1], and 𝕀nn=[(n1)/n,1].

We call {𝕏kn}k[nd] a fine partition of 𝕏, in the sense that 𝕏=[0,1]d=k=1nd𝕏kn, for each n, and that λ(𝕏kn)=nd gets smaller, as n increases. The following result from Jiang & Tanner, 1999b establishes the approximation capability of soft-max gates.

Lemma 2 (Jiang and Tanner, 1999, p. 1189).

For each n, p[1,) and ϵ>0, there exists a gating functions

𝐆𝐚𝐭𝐞=(Gatek(;𝜸))k[nd]𝒢Snd

for some 𝛄𝔾Snd, such that

supk[nd]𝟏{𝒙𝕏kn}Gatek(;𝜸)p,𝕏ϵ.

When, d=1, we have also the following almost uniform convergence alternative to Lemma 2.

Lemma 3.

Let 𝕏=[0,1]. Then, for each n, there exists a sequence of gating functions:

{𝐆𝐚𝐭𝐞l=(Gatek(;𝜸l))k[nd]}l𝒢Sn,

defined by {𝛄l}l𝔾Sn, such that

Gatek(;𝜸l)𝟏{𝒙𝕏kn},

almost uniformly, simultaneously for all k[nd].

For PDF ψ on support q, define the class of finite mixture models by

ψ = {hKψ:q[0,)|hKψ(𝒚)=k=1Kckgψ(𝒚;𝝁k,σk),
gψψ,(ck)k[K]ΠK1,𝝁k𝕐,σk(0,),k[K],K}.

We require the following result, from Nguyen et al., 2020a , regarding the approximation capabilities of ψ.

Lemma 4 (Nguyen et al., 2020a, Thm. 2(b)).

If f𝒞(𝕐) is a PDF on 𝕐, ψ𝒞(q) is a PDF on q, and 𝕐q is compact, then there exists a sequence {hKψ}Kψ, such that limKfhKψ(𝕐)=0.

4 Proofs of main results

4.1 Proof of Theorem 1

To prove the result, it suffices to show that for each ϵ>0, there exists a mKψSψ, such that

fmKψp<ϵ.

The main steps of the proof are as follows. We firstly approximate f(𝒚|𝒙) by

υn(𝒚|𝒙)=k=1nd𝟏{𝒙𝕏kn}f(𝒚|𝒙kn), (1)

where 𝒙kn𝕏kn, for each k[nd], such that

fυnp<ϵ3, (2)

for all nN1(ϵ), for some sufficiently large N1(ϵ). Then we approximate υn(𝒚|𝒙) by

ηn(𝒚|𝒙)=k=1ndGatek(𝒙;𝜸n)f(𝒚|𝒙kn), (3)

where 𝜸n𝔾Snd and 𝐆𝐚𝐭𝐞=(Gatek(;𝜸n))k[nd]𝒢Snd, so that

υnηnp supk[nd]Gatek(;𝜸)𝟏{𝒙𝕏kn}p,𝕏k=1ndf(|𝒙kn)p,𝕐<ϵ3, (4)

using Lemma 2.

Finally, we approximate ηn(𝒚|𝒙) by mKnψ(𝒚|𝒙), where

mKnψ(𝒚|𝒙)=k=1ndGatek(𝒙;𝜸)hnkk(𝒚|𝒙kn) (5)

and

hnkk(𝒚|𝒙kn)=i=1nkcikgψ(𝒚;𝝁ik,σik)ψ (6)

for nk (k[nd]), such that Kn=k=1ndnk. Here, we establish that there exists N2(ϵ,n,𝜸n), so that when nkN2(ϵ,n,𝜸n),

ηnmKnψpsupk[nd]Gatek(;𝜸)p,𝕏k=1ndf(|𝒙kn)hnkk(|𝒙kn)p,𝕐<ϵ3. (7)

Results (2)–(7) then imply that for each ϵ>0, there exists N1(ϵ), 𝜸n, and N2(ϵ,n,𝜸n), such that for all Kn=k=1ndnk, where nkN2(ϵ,n,𝜸n) (for each k[nd]) and nN1(ϵ). The following inequality results from an application of the triangle inequality:

fmKnψp fυnp+υnηnp+ηnmKnψp<3×ϵ3=ϵ.

We now focus our attention to proving each of the results: (2)–(7). To prove (2), we note that since f is uniformly continuous (because =𝕏×𝕐 is compact, and f𝒞), there exists a function (1) such that for all ε>0,

sup(𝒙,𝒚)|f(𝒚|𝒙)υ(𝒚|𝒙)|<ε. (8)

We can construct such an approximation by considering the fact that as n increases, the diameter δn=supknddiam(𝕏kn) of the fine partition goes to zero. By the uniform continuity of f, for every ε>0, there exists a δ(ϵ)>0, such that if (𝒙1,𝒚1)(𝒙2,𝒚2)<δ(ϵ), then |f(𝒚1|𝒙1)f(𝒚2|𝒙2)|<ε, for all pairs (𝒙1,𝒚1),(𝒙2,𝒚2). Here, denotes the Euclidean norm. Furthermore, for any (𝒙,𝒚), we have

|f(𝒚|𝒙)υn(𝒚|𝒙)| =|k=1nd𝟏{𝒙𝕏kn}[f(𝒚|𝒙)f(𝒚|𝒙kn)]|
k=1nd𝟏{𝒙𝕏kn}|f(𝒚|𝒙)f(𝒚|𝒙kn)|, (9)

by the triangle inequality.

Since 𝒙kn𝕏kn, for each k and n, we have the fact that (𝒙,𝒚)(𝒙kn,𝒚)<δn for (𝒙,𝒚)𝕏kn×𝕐. By uniform continuity, for each ε, we can find a sufficiently small δ(ϵ), such that |f(𝒚|𝒙)f(𝒚|𝒙kn)|<ε, if (𝒙,𝒚)(𝒙kn,𝒚)<δ(ϵ), for all k. The desired result (8) can be obtained by noting that the right hand side of (9) consists of only one non-zero summand for any (𝒙,𝒚), and by choosing n sufficiently large, so that δn<δ(ϵ).

By (8), we have the fact that υnf, point-wise. We can bound υn as follows:

υn(𝒚|𝒙)i=1np𝟏{𝒙𝕏kn}sup𝜻𝕐,𝝃𝕏f(𝜻|𝝃)=sup𝜻𝕐,𝝃𝕏f(𝜻|𝝃), (10)

where the right-hand side is a constant and is therefore in p, since is compact. An application of the Lebesgue dominated convergence theorem in p then yields (2).

Next we write

υnηnp =k=1nd𝟏{𝒙𝕏kn}f(𝒚|𝒙kn)k=1ndGatek(𝒙;𝜸n)f(𝒚|𝒙kn)p
k=1nd[𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸n)]f(𝒚|𝒙kn)p.

Since the norm arguments are separable in 𝒙 and 𝒚, we apply Fubini’s theorem to get

υnηnp =k=1nd[𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸n)]p,𝕏f(𝒚|𝒙kn)p,𝕐
supk[nd][𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸n)]p,𝕏k=1ndf(𝒚|𝒙kn)p,𝕐

Because f and nd is finite, for any fixed n, we have C1(n)=k=1ndf(𝒚|𝒙kn)p,𝕐<. For each ϵ>0, we need to choose a 𝜸n𝔾Snd, such that

supk[nd][𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸n)]p,𝕏<ϵ3C1(n),

which can be achieved via a direct application of Lemma 2. We have thus shown (4).

Lastly, we are required to approximate f(𝒚|𝒙kn) for each k[nd], by a function of form (6). Since 𝕐 is compact and f and ψ are continuous, we can apply of Lemma 4, directly. Note that over a set of finite measure, convergence in implies convergence in p norm, for all p[1,] (cf. Oden & Demkowicz, 2010, Prop. 3.9.3).

We can then write (5) as

mKnψ(𝒚|𝒙) =k=1ndexp(an,k+𝒃n,k𝒙)l=1ndexp(an,l+𝒃n,l𝒙)hnkk(𝒚|𝒙kn)
=k=1ndi=1nkexp(an,k+𝒃n,k𝒙)l=1ndexp(an,l+𝒃n,l𝒙)cikl=1nkclkgψ(𝒚;𝝁ik,σik)
=k=1ndi=1nkexp(logcik+an,k+𝒃n,k𝒙)l=1ndj=1nkexp(logcjk+an,l+𝒃n,l𝒙)gψ(𝒚;𝝁ik,σik), (11)

where 𝜸n=(an,1,,an,nd,𝒃n,1,,𝒃n,nd). From (11), we observe that mKnψSψ, with Kn=k=1ndnk.

To obtain (7), we write

ηnmKnψp =k=1ndGatek(𝒙;𝜸n)f(𝒚|𝒙kn)k=1ndGatek(𝒙;𝜸)hnkk(𝒚|𝒙kn)p
k=1ndGatek(𝒙;𝜸n)[f(𝒚|𝒙kn)hnkk(𝒚|𝒙kn)]p.

By separability and Fubini’s theorem, we then have

ηnmKnψ k=1ndGatek(𝒙;𝜸n)p,𝕏f(𝒚|𝒙kn)hnkk(𝒚|𝒙kn)p,𝕐
supk[nd]Gatek(𝒙;𝜸n)p,𝕏k=1ndf(𝒚|𝒙kn)hnkk(𝒚|𝒙kn)p,𝕐.

Let C2(n,𝜸n)=supk[nd]Gatek(𝒙;𝜸n)p,𝕏. Then, we apply Lemma 4 nd times to establish the existence of a constant N2(ϵ,n,𝜸n), such that for all k[nd] and nkN2(ϵ,n,𝜸n),

f(𝒚|𝒙kn)hnkk(𝒚|𝒙kn)p,𝕐ϵ3C2(n,𝜸n)nd.

Thus, we have

ηnmKnψC2(n,𝜸n)×nd×ϵ3C2(n,𝜸n)nd=ϵ3,

which completes our proof.

4.2 Proof of Theorem 2

The proof is procedurally similar to that of Theorem 1 and thus we only seek to highlight the important differences. Firstly, for any ϵ>0, we approximate f(𝒚|𝒙) by υn(𝒙|𝒚) of form (1), with d=1. Result (2) implies uniform convergence, in the sense that there exists an N1(ϵ), such that for all nN1(ϵ),

fυn<ϵ3. (12)

We now seek to approximate υn by ηn of form (3), with 𝜸n=𝜸l for some l. Upon application of Lemma 3, it follows that for each k[nd] and ε>0, there exists a measurable set 𝔹k(ε)𝕏, such that

λ(𝔹k(ε))<εndλ(𝕐)

and

Gatek(;𝜸l)𝟏{𝒙𝕏kn}(𝔹k(ε))<ϵ3

for all lMk(ϵ,n), for some Mk(ϵ,n). Here, () is the set complement operator.

Since f, we have the bound C(n)=k=1ndf(𝒚|𝒙kn)(𝕐)<. Write 𝔹(ε)=k=1nd𝔹k(ε). Then, 𝔹(ε)=k=1nd𝔹k(ε),

λ(𝔹(ε))k=1ndλ(𝔹k(ε))<ελ(𝕐),

and

Gatek(;𝜸l)𝟏{𝒙𝕏kn}(𝔹(ε)) mink[nd]Gatek(;𝜸l)𝟏{𝒙𝕏kn}(𝔹k(ε))<ϵ3C(n),

for all lM(ϵ,n)=maxk[nd]Mk(ϵ,n). Here we use the fact that the supremum over some intersect of sets is less than or equal to the minimum of the supremum over each individual set.

Upon defining (ε)=𝔹(ε)×𝕐, we observe that

λ((ε))=λ(𝔹(ε))λ(𝕐)ελ(𝕐)×λ(𝕐)=ε,

and (ε)𝔹(ε)×𝕐. Note also that

(𝔹(ε)×𝕐)=\(𝔹(ε)×𝕐)=𝔹(ε)×𝕐

and

(ε)=(𝔹(ε)×𝕐)(𝔹(ε)×𝕐)(𝔹(ε)×𝕐).

It follows that

υnηn((ε))max{υnηn(𝔹(ε)×𝕐),υnηn(𝔹(ε)×𝕐),υnηn(𝔹(ε)×𝕐)}.

Since 𝔹(ε)×𝕐 and 𝔹(ε)×𝕐 are empty, via separability, we have

υnηn((ε)) =υnηn(𝔹(ε)×𝕐)
=sup𝒛𝔹(ε)×𝕐|k=1nd[𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)]f(𝒚|𝒙kn)|
sup𝒛𝔹(ε)×𝕐k=1nd|𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)|f(𝒚|𝒙kn)
k=1ndsup𝒛𝔹(ε)×𝕐|𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)|f(𝒚|𝒙kn)
=k=1nd𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)(𝔹(ε))f(𝒚|𝒙kn)(𝕐)
supk[n]𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)(𝔹(ε))k=1ndf(𝒚|𝒙kn)(𝕐).

Recall that the k=1ndf(𝒚|𝒙kn)(𝕐)=C(n)< and that we can choose lM(ϵ,n) so that

supk[n]𝟏{𝒙𝕏kn}Gatek(𝒙;𝜸l)(𝔹(ε))<ϵ3C(n),

and thus

υnηn((ε))<ϵ3C(n)×C(n)=ϵ3, (13)

as required.

Finally, by noting that for each k[nd], both (6) and f(|𝒙kn) are continuous over 𝕐, we apply Lemma 4 to obtain an N2(ϵ,n,l), such that for any ϵ>0 and nkN2(ϵ,n,l), we have

f(|𝒙kn)hnkk(|𝒙kn)(𝕐)<ϵ3M1n.

Here M1=supk[nd]Gatek(;𝜸l)(𝕏)<, since Gatek(𝒙;𝜸l) is continuous in 𝒙, and 𝕏 is compact. Therefore, for all Kn=k=1ndnk, nkN2(ϵ,n,l),

ηnmKnψ supk[nd]Gatek(𝒙;𝜸l)(𝕏)k=1ndf(|𝒙kn)hnkk(|𝒙kn)(𝕐)
=M1×nd×ϵ3M1nd=ϵ3. (14)

In summary, via (12), (13), and (14), for each ϵ>0, for any ε>0, there exists a (ε) and constants N1(ϵ),M(ϵ,n),N2(ϵ,n,l), such that for all Kn=k=1ndnk, with nkN2(ϵ,n,l), lM(ϵ,n), and nN1(ϵ), it follows that λ((ε))<ε, and

fmKnψ((ε)) fυn((ε))+υnηn((ε))+ηnmKnψ((ε))
fυn+υnηn((ε))+ηnmKnψ
<3×ϵ3=ϵ.

This completes the proof.

5 Proofs of lemmas

5.1 Proof of Lemma 1

We firstly prove that any gating vector from 𝒢SK can be equivalently represented as an element of 𝒢GK. For any 𝒙d, d, k[K], ak, 𝒃kd, and K, choose 𝝂k=𝒃k, τk=ak+𝒃k𝒃k/2 and

πk=exp(τk)/l=1Kexp(τl).

This implies that l=1Kπl=1, πl>0, for all l[K], and

exp(ak+𝒃k𝒙)l=1Kexp(ak+𝒃k𝒙) =exp(τk𝒗k𝒗k/2+𝒗k𝒙)l=1Kexp(τl𝒗l𝒗l/2+𝒗l𝒙)
=exp(τk)exp((𝒙𝝂k)(𝒙𝝂k)/2)l=1Kexp(τl)exp((𝒙𝝂l)(𝒙𝝂l)/2)
=πk(2π)d/2exp((𝒙𝝂k)(𝒙𝝂k)/2)l=1Kπl(2π)d/2exp((𝒙𝝂l)(𝒙𝝂l)/2)
=πkϕ(𝒙;𝝂k,𝐈)l=1Kπlϕ(𝒙;𝝂l,𝐈),

where 𝐈 is the identity matrix of appropriate size. This proves that 𝒢SK𝒢GK.

Next, to show that 𝒢EK𝒢SK, we write

πkϕ(𝒙;𝝂k,𝚺)l=1Kπlϕ(𝒙;𝝂l,𝚺)
= πk|2π𝚺|1/2exp((𝒙𝝂k)𝚺1(𝒙𝝂k)/2)l=1Kπl|2π𝚺|1/2exp((𝒙𝝂l)𝚺1(𝒙𝝂l)/2)
= 1l=1Kexp(log(πl2/πk2)/2(𝒙𝝂l)𝚺1(𝒙𝝂l)/2(𝒙𝝂k)𝚺1(𝒙𝝂k)/2),

and note that

(𝒙𝝂l)𝚺1(𝒙𝝂l)(𝒙𝝂k)𝚺1(𝒙𝝂k)
= 2(𝝂l𝝂k)𝚺1𝒙+(𝝂l+𝝂k)𝚺1(𝝂l𝝂k).

Thus, we have

πkϕ(𝒙;𝝂k,𝚺)l=1Kπlϕ(𝒙;𝝂l,𝚺)
= 1l=1Kexp(log(πl2/πk2)/2(𝝂l+𝝂k)𝚺1(𝝂l𝝂k)/2(𝝂l𝝂k)𝚺1𝒙).

Next, notice that we can write

exp(ak+𝒃k𝒙)l=1Kexp(al+𝒃l𝒙)=1l=1Kexp(αl+𝜷l𝒙),

where αl=alak and 𝜷l=𝜷l𝜷k. We now choose ak and 𝒃k, such that for every l[K],

αl =alak=12log(πl2πk2)12(𝝂l𝚺1𝝂l𝝂k𝚺1𝝂k),

and

𝜷l=𝜷l𝜷k=𝝂l𝚺1𝝂k𝚺1.

To complete the proof, we choose

ak=log(πk)12𝝂k𝚺1𝝂k

and bk=𝝂k𝚺1, for each k[K].

5.2 Proof of Lemma 3

For l[0,), write

Gatek(x,l)=exp([xck]lk)i=1nexp([xci]li),

where x𝕏=[0,1], and ck=(k1)/(2k). We identify that 𝐆𝐚𝐭𝐞=(Gatek(x,l))k[n] belongs to the class 𝒢Sn. The proof of the Section 4 Proposition from Jiang & Tanner, 1999b reveals that for all k[n],

Gatek(x,l)𝟏{x𝕀kn}

almost everywhere in λ, as l. The result then follows via an application of Egorov’s theorem (cf. Folland, 1999, Thm. 2.33).

6 Summary and conclusions

Using recent results mixture model approximation results Nguyen et al., 2020a and Nguyen et al., 2020c , and the indicator approximation theorem of Jiang & Tanner, 1999b (cf. Section 3), we have proved two approximation theorems (Theorems 1 and 2) regarding the class of soft-max gated MoE models with experts arising from arbitrary location-scale families of conditional density functions. Via an equivalence result (Lemma 1), the results of Theorems 1 and 2 also extend to the setting of Gaussian gated MoE models (Corollary 1), which can be seen as a generalization of the soft-max gated MoE models.

Although we explicitly make the assumption that 𝕏=[0,1]d, for the sake of mathematical argument (so that we can make direct use of Lemma 2), a simple shift-and-scale argument can be used to generalize our result to cases where 𝕏 is any generic compact domain. The compactness assumption regarding the input domain is common in the MoE and mixture of regression models literature, as per the works of Jiang & Tanner, 1999a , Jiang & Tanner, 1999b , Norets, (2010), Montuelle & Le Pennec, (2014), Pelenis, (2014), Devijver, 2015a , and Devijver, 2015b .

The assumption permits the application of the result to the settings where the inputs 𝑿 is assumed to be non-random design vectors that take value on some compact set 𝕏. This is often the case when there is only a finite number of possible design vector elements for which 𝑿 can take. Otherwise, the assumption also permits the scenario where 𝑿 is some random element with compactly supported distribution, such as uniformly distributed, or beta distributed inputs. Unfortunately, the case of random 𝑿 over an unbounded domain (e.g., if 𝑿 has multivariate Gaussian distribution) is not covered under our framework. An extension to such cases would require a more general version of Lemma 2, which we believe is a nontrivial direction for future work.

Like the input, we also assume that the output domain is restricted to a compact set 𝕐. However, the output domain of the approximating class of MoE models is unrestricted to 𝕐 and thus the functions (i.e., we allow ψ to be a PDF over q). The restrictions placed on 𝕐 is also common in the mixture approximation literature, as per the works of Zeevi & Meir, (1997), Li & Barron, (1999), and Rakhlin et al., (2005), and is also often made in the context of nonparametric regression (see, e.g., Gyorfi et al.,, 2002 and Cucker & Zhou,, 2007). Here, our use of the compactness of 𝕐 is to bound the integral of vn, in (10). A more nuanced approach, such as via the use of a generalized Lebesgue spaces (see e.g., Castillo & Rafeiro,, 2010 and Cruze-Uribe & Fiorenza,, 2013), may lead to result for unbounded 𝕐. This is another exciting future direction of our research program.

A trivial modification to the proof of Lemma 4 allows us to replace the assumption that f is a PDF with a sub-PDF assumption (i.e., 𝕐fdλ1), instead. This in turn permits us to replace the assumption that f(|𝒙) is a conditional PDF in Theorems 1 and 2 with sub-PDF assumptions as well (i.e., for each 𝒙𝕏, 𝕐f(𝒚|𝒙)dλ(𝒚)1). Thus, in this modified form, we have a useful interpretation for situations when the input 𝒀 is unbounded. That is, when 𝒀 is unbounded, we can say that the conditional PDF f can be arbitrarily well approximated in p norm by a sequence {mKψ}K of either soft-max or Gaussian gated MoEs over any compact subdomain 𝕐 of the unbounded domain of 𝒀. Thus, although we cannot provide guarantees of the entire domain of 𝒀, we are able to guarantee arbitrary approximate fidelity over any arbitrarily large compact subdomain. This is a useful result in practice since one is often not interested in the entire domain of 𝒀, but only on some subdomain where the probability of 𝒀 is concentrated. This version of the result resembles traditional denseness results in approximation theory, such as those of Cheney & Light, (2000, Ch. 20).

Finally, our results can be directly applied to provide approximation guarantees for a large number of currently used models in applied statistics and machine learning research. Particularly, our approximation guarantees are applicable to the recent MoE models of Ingrassia et al., (2012), Chamroukhi et al., (2013), Ingrassia et al., (2014), Chamroukhi, (2016), Nguyen & McLachlan, (2016), Deleforge et al., 2015b , Deleforge et al., 2015a , Kalliovirta et al., (2016), and Perthame et al., (2018), among many others. Here, we may guarantee that the underlying data generating processes, if satisfying our assumptions, can be adequately well approximated by sufficiently complex forms of the models considered in each of the aforementioned work.

The rate and manner of which good approximation can be achieved as a function of the number of experts K and the sample size is a currently active research area, with pioneering work conducted in Cohen & Le Pennec, (2012) and Montuelle & Le Pennec, (2014). More recent results in this direction appear in Nguyen et al., 2020b , Nguyen et al., 2021b , and Nguyen et al., 2021a .

List of abbreviations

MoE

Mixture of experts

PDF

Probability density function

Acknowledgements

Hien Duy Nguyen and Geoffrey John McLachlan are funded by Australian Research Council grants: DP180101192 and IC170100035. TrungTin Nguyen is supported by a “Contrat doctoral” from the French Ministry of Higher Educationand Research. Faicel Chamroukhi is granted by the French National Research Agency (ANR) grant SMILES ANR-18-CE40-0014. The authors also thank the Editor and Reviewer, whose careful and considerate comments lead to improvements in of text.

Declarations

Funding: HDN and GJM are funded by Australian Research Council grants: DP180101192 and IC170100035. FC is funded by ANR grant: SMILES ANR-18-CE40-0014.
Conflicts of interest: None.
Availability of data and material: Not applicable.
Authors’ contributions: All authors contributed equally to the exposition and to the mathematical derivations.
Code availability: Not applicable.

References

  • Bartle, (1995) Bartle, R. (1995). The Elements of Integration and Lebesgue Measure. New York: Wiley. a b
  • Castillo & Rafeiro, (2010) Castillo, R. E. & Rafeiro, H. (2010). An Introductory Course in Lebesgue Spaces. Switzerland: Springer. a
  • Chamroukhi, (2016) Chamroukhi, F. (2016). Robust mixture of experts modeling using the t distribution. Neural Networks, 79, 20–36. a
  • Chamroukhi et al., (2013) Chamroukhi, F., Mohammed, S., Trabelsi, D., Oukhellou, L., & Amirat, Y. (2013). Joint segmentation of multivariate time series with hidden process regression for human activity recognition. Neurocomputing, 120, 633–644. a
  • Cheney & Light, (2000) Cheney, W. & Light, W. (2000). A Course in Approximation Theory. Pacific Grove: Brooks/Cole. a
  • Cohen & Le Pennec, (2012) Cohen, S. & Le Pennec, E. (2012). Conditional density estimation by penalized likelihood model selection and application. ArXiv, (arXiv:1103.2021). a
  • Cruze-Uribe & Fiorenza, (2013) Cruze-Uribe, D. V. & Fiorenza, A. (2013). Variable Lebesgue Spaces: Foundations and Harmonic Analysis. Basel: Birkhauser. a
  • Cucker & Zhou, (2007) Cucker, F. & Zhou, D.-X. (2007). Learning Theory: An Approximation Theory Viewpoint. Cambridge: Cambridge University Press. a
  • (9) Deleforge, A., Forbes, F., & Horaud, R. (2015a). Acoustic space learning for sound-source separation and localization on binaural manifolds. International Journal of Neural Systems, 25, 1440003. a
  • (10) Deleforge, A., Forbes, F., & Horaud, R. (2015b). High-dimensional regression with Gaussian mixtures and partially-latent response variables. Statistics and Computing, 25, 893–911. a
  • (11) Devijver, E. (2015a). An 1-oracle inequality for the Lasso in multivariate finite mixture of multivariate Gaussian regression models. ESAIM: Probability and Statistics, 19, 649–670. a
  • (12) Devijver, E. (2015b). Finite mixture regression: a sparse variable selection by model selection for clustering. Electronic Journal of Statistics, 9, 2642–2674. a
  • Folland, (1999) Folland, G. B. (1999). Real Analysis: Modern Techniques and Their Applications. New York: Wiley. a
  • Gyorfi et al., (2002) Gyorfi, L., Kohler, M., Krzyzak, A., & Walk, H. (2002). A Distribution-free Theory Of Nonparametric Regression. New York: Springer. a
  • Ingrassia et al., (2014) Ingrassia, S., Minotti, S. C., & Punzo, A. (2014). Model-based clustering via linear cluster-weighted models. Computational Statistics and Data Analysis, 71, 159–182. a
  • Ingrassia et al., (2012) Ingrassia, S., Minotti, S. C., & Vittadini, G. (2012). Local statistical modeling via a cluster-weighted approach with elliptical distributions. Journal of Classification, 29, 363–401. a
  • Jacobs et al., (1991) Jacobs, R. A., Jordan, M. I., Nowlan, S. J., & Hinton, G. E. (1991). Adaptive mixtures of local experts. Neural Computation, 3, 79–87. a
  • (18) Jiang, W. & Tanner, M. A. (1999a). Hierachical mixtures-of-experts for exponential family regression models: approximation and maximum likelihood estimation. Annals of Statistics, 27, 987–1011. a b
  • (19) Jiang, W. & Tanner, M. A. (1999b). On the approximation rate of hierachical mixtures-of-experts for generalized linear models. Neural Computation, 11, 1183–1198. a b c d e f
  • Jordan & Xu, (1995) Jordan, M. I. & Xu, L. (1995). Convergence results for the EM approach to mixtures of experts architectures. Neural Networks, 8, 1409–1431. a
  • Kalliovirta et al., (2016) Kalliovirta, L., Meitz, M., & Saikkonen, P. (2016). Gaussian mixture vector autoregression. Journal of Econometrics, 192, 485–498. a
  • Krzyzak & Schafer, (2005) Krzyzak, A. & Schafer, D. (2005). Nonparametric regression estimation by normalized radial basis function networks. IEEE Transactions on Information Theory, 51, 1003–1010. a
  • Li & Barron, (1999) Li, J. Q. & Barron, A. R. (1999). Mixture density estimation. In S. A. Solla, T. K. Leen, & K. R. Mueller (Eds.), Advances in Neural Information Processing Systems, volume 12 Cambridge: MIT Press. a
  • Masoudnia & Ebrahimpour, (2014) Masoudnia, S. & Ebrahimpour, R. (2014). Mixture of experts: a literature survey. Artificial Intelligence Review, 42, 275–293. a
  • Mendes & Jiang, (2012) Mendes, E. F. & Jiang, W. (2012). On convergence rates of mixture of polynomial experts. Neural Computation, 24. 3025-3051. a
  • Montuelle & Le Pennec, (2014) Montuelle, L. & Le Pennec, E. (2014). Mixture of Gaussian regressions model with logistic weights, a penalized maximum likelihood approach. Electronic Journal of Statistics, 8, 1661–1695. a b
  • Nguyen & Chamroukhi, (2018) Nguyen, H. D. & Chamroukhi, F. (2018). Practical and theoretical aspects of mixture-of-experts modeling: an overview. WIREs Data Mining and Knowledge Discovery, (pp. e1246). a
  • Nguyen et al., (2019) Nguyen, H. D., Chamroukhi, F., & Forbes, F. (2019). Approximation results regarding the multiple-output Gaussian gated mixture of linear experts model. Neurocomputing, 366, 208–214. a
  • Nguyen et al., (2016) Nguyen, H. D., Lloyd-Jones, L. R., & McLachlan, G. J. (2016). A universal approximation theorem for mixture-of-experts models. Neural Computation, 28, 2585–2593. a
  • Nguyen & McLachlan, (2016) Nguyen, H. D. & McLachlan, G. J. (2016). Laplace mixture of linear experts. Computational Statistics and Data Analysis, 93, 177–191. a
  • (31) Nguyen, T. T., Chamroukhi, F., Nguyen, H. D., & Forbes, F. (2021a). Non-asymptotic model selection in block-diagonal mixture of polynomial experts models. arXiv preprint arXiv:2104.08959. a
  • (32) Nguyen, T. T., Chamroukhi, F., Nguyen, H. D., & McLachlan, G. J. (2020a). Approximation of probability density functions via location-scale finite mixtures in Lebesgue spaces. arXiv:2008.09787. a b c
  • (33) Nguyen, T. T., Nguyen, H. D., Chamroukhi, F., & Forbes, F. (2021b). A non-asymptotic penalization criterion for model selection in mixture of experts models. ArXiv, (arXiv:2104.02640). a
  • (34) Nguyen, T. T., Nguyen, H. D., Chamroukhi, F., & McLachlan, G. J. (2020b). An l1-oracle inequality for the Lasso in mixture-of-experts regression models. ArXiv, (arXiv:2009.10622). a
  • (35) Nguyen, T. T., Nguyen, H. D., Chamroukhi, F., & McLachlan, G. J. (2020c). Approximation by finite mixtures of continuous density functions that vanish at infinity. Cogent Mathematics and Statistics, 7, 1750861. a b
  • Norets, (2010) Norets, A. (2010). Approximation of conditional densities by smooth mixtures of regressions. Annals of Statistics, 38, 1733–1766. a b c
  • Norets & Pelenis, (2014) Norets, A. & Pelenis, J. (2014). Posterior consistency in conditional density estimation by covariate dependent mixtures. Econometric Theory, 30, 606–646. a b c
  • Oden & Demkowicz, (2010) Oden, J. T. & Demkowicz, L. F. (2010). Applied Functional Analysis. Boca Raton: CRC Press. a
  • Pelenis, (2014) Pelenis, J. (2014). Bayesian regression with heteroscedastic error density and parametric mean function. Journal of Econometrics, 178, 624–638. a
  • Perthame et al., (2018) Perthame, E., Forbes, F., & Deleforge, A. (2018). Inverse regression approach to robust nonlinear high-to-low dimensional mapping. Journal of Multivariate Analysis, 163, 1–14. a
  • Rakhlin et al., (2005) Rakhlin, A., Panchenko, D., & Mukherjee, S. (2005). Risk bounds for mixture density estimation. ESAIM: Probability and Statistics, 9, 220–229. a
  • Wang & Mendel, (1992) Wang, L.-X. & Mendel, J. M. (1992). Fuzzy basis functions, universal approximation, and orthogonal least-squares learning. IEEE Transactions on Neural Networks, 3, 807–814. a
  • Yuksel et al., (2012) Yuksel, S. E., Wilson, J. N., & Gader, P. D. (2012). Twenty years of mixture of experts. IEEE Transactions on Neural Networks and Learning Systems, 23, 1177–1193. a
  • Zeevi & Meir, (1997) Zeevi, A. J. & Meir, R. (1997). Density estimation through convex combinations of densities: approximation and estimation bounds. Neural Computation, 10, 99–109. a b
  • Zeevi et al., (1998) Zeevi, A. J., Meir, R., & Maiorov, V. (1998). Error bounds for functional approximation and estimation using mixtures of experts. IEEE Transactions on Information Theory, 44, 1010–1025. a

Cite this paper

Please cite the published version. Venue: Journal of Statistical Distributions and Applications, Open-access journal article (Vol. 8, Art. 13, 2021). DOI: 10.1186/s40488-021-00125-0. Official record: SpringerOpen.

BibTeX
@article{nguyen2021approximations,
  title     = {Approximations of conditional probability density functions in Lebesgue spaces via mixture of experts models},
  author    = {Nguyen, Hien D. and Nguyen, TrungTin and Chamroukhi, Faicel and McLachlan, Geoffrey J.},
  journal   = {Journal of Statistical Distributions and Applications},
  volume    = {8}, number = {1}, pages = {13},
  year      = {2021}, publisher = {Springer},
  doi       = {10.1186/s40488-021-00125-0},
}