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 and , for . Suppose that the input and output random variables, and , are related via the conditional probability density function (PDF) in the functional class:
where denotes the Lebesgue measure. The MoE approach seeks to approximate the unknown target conditional PDF by a function of the MoE form:
where (), , and . Here, we say that is a MoE model with gates arising from the class and experts arising from , where is a class of PDFs with support .
The most popular choices for are the parametric soft-max and Gaussian gating classes:
and
respectively, where
and
Here,
is the multivariate normal density function with mean vector and covariance matrix , is a vector of weights in the simplex:
and is the class of 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:
where is a PDF, with respect to in the sense that and .
We shall say that for any if
where is the indicator function that takes value when , and 0 otherwise. Further, we say that if
We shall refer to as the norm on , for , and where the context is obvious, we shall drop the reference to .
Suppose that the target conditional PDF is in the class . We address the problem of approximating , with respect to the norm, using MoE models in the soft-max and Gaussian gated classes,
and
by showing that both and are dense in the class , when and is a compact subset of . 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 and (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 and , 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 norm (Zeevi & Meir,, 1997). That is, when , for all and some constant , we have that the integrated conditional Kullback–Leibler divergence considered by Norets & Pelenis, (2014):
satisfies
and thus a good approximation in the integrated Kullback–Leibler divergence is guaranteed if one can find a good approximation in the 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
and write its norm as . Further, let denote the class of continuous functions. Note that if is compact and , then .
Theorem 1.
Assume that for . There exists a sequence , such that if is compact, , and is a PDF on support , then , for .
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 to in the following result is to be understood in the sense of Bartle, (1995, Def. 7.9). That is, for every , there exists a set with , such that converges to , uniformly on .
Theorem 2.
Assume that . There exists a sequence , such that if is compact, , and is a PDF on support , then , almost uniformly.
The following result establishes the connection between the gating classes and .
Lemma 1.
For each , . Further, if we define the class of Gaussian gating vectors with equal covariance matrices:
where
then .
3 Technical lemmas
Let and be a bijection for each . For each and , we define , where for , and .
We call a fine partition of , in the sense that , for each , and that gets smaller, as 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 , and , there exists a gating functions
for some , such that
When, , we have also the following almost uniform convergence alternative to Lemma 2.
Lemma 3.
Let . Then, for each , there exists a sequence of gating functions:
defined by , such that
almost uniformly, simultaneously for all .
For PDF on support , define the class of finite mixture models by
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 is a PDF on , is a PDF on , and is compact, then there exists a sequence , such that .
4 Proofs of main results
4.1 Proof of Theorem 1
To prove the result, it suffices to show that for each , there exists a , such that
The main steps of the proof are as follows. We firstly approximate by
| (1) |
where , for each , such that
| (2) |
for all , for some sufficiently large . Then we approximate by
Finally, we approximate by , where
| (5) |
and
| (6) |
for (), such that . Here, we establish that there exists , so that when ,
| (7) |
Results (2)–(7) then imply that for each , there exists , , and , such that for all , where (for each ) and . The following inequality results from an application of the triangle inequality:
We now focus our attention to proving each of the results: (2)–(7). To prove (2), we note that since is uniformly continuous (because is compact, and ), there exists a function (1) such that for all ,
| (8) |
We can construct such an approximation by considering the fact that as increases, the diameter of the fine partition goes to zero. By the uniform continuity of , for every , there exists a , such that if , then , for all pairs . Here, denotes the Euclidean norm. Furthermore, for any , we have
| (9) |
by the triangle inequality.
Since , for each and , we have the fact that for . By uniform continuity, for each , we can find a sufficiently small , such that , if , for all . 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 sufficiently large, so that .
By (8), we have the fact that , point-wise. We can bound as follows:
| (10) |
where the right-hand side is a constant and is therefore in , since is compact. An application of the Lebesgue dominated convergence theorem in then yields (2).
Next we write
Since the norm arguments are separable in and , we apply Fubini’s theorem to get
Because and is finite, for any fixed , we have . For each , we need to choose a , such that
Lastly, we are required to approximate for each , by a function of form (6). Since is compact and and are continuous, we can apply of Lemma 4, directly. Note that over a set of finite measure, convergence in implies convergence in norm, for all (cf. Oden & Demkowicz, 2010, Prop. 3.9.3).
Let . Then, we apply Lemma 4 times to establish the existence of a constant , such that for all and ,
Thus, we have
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 , we approximate by of form (1), with . Result (2) implies uniform convergence, in the sense that there exists an , such that for all ,
| (12) |
We now seek to approximate by of form (3), with for some . Upon application of Lemma 3, it follows that for each and , there exists a measurable set , such that
and
for all , for some . Here, is the set complement operator.
Since , we have the bound . Write . Then, ,
and
for all . 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
Since and are empty, via separability, we have
Recall that the and that we can choose so that
and thus
| (13) |
as required.
5 Proofs of lemmas
5.1 Proof of Lemma 1
We firstly prove that any gating vector from can be equivalently represented as an element of . For any , , , , , and , choose , and
This implies that , , for all , and
where is the identity matrix of appropriate size. This proves that .
Next, to show that , we write
and note that
Thus, we have
Next, notice that we can write
where and . We now choose and , such that for every ,
and
To complete the proof, we choose
and , for each .
5.2 Proof of Lemma 3
For , write
where , and . We identify that belongs to the class . The proof of the Section 4 Proposition from Jiang & Tanner, 1999b reveals that for all ,
almost everywhere in , as . 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 , 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 ). 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 , 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 is a PDF with a sub-PDF assumption (i.e., ), instead. This in turn permits us to replace the assumption that is a conditional PDF in Theorems 1 and 2 with sub-PDF assumptions as well (i.e., for each , ). 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 can be arbitrarily well approximated in norm by a sequence 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 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
-
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 -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 -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