Improved learning algorithms for mixture of experts in multiclass classification
Publication in refereed journal

Times Cited
Web of Science76WOS source URL (as at 27/11/2021) Click here for the latest count
Altmetrics Information

Other information
AbstractMixture of experts (ME) is a modular neural network architecture for supervised learning. A double-loop Expectation-Maximization (EM) algorithm has been introduced to the ME architecture for adjusting the parameters and the iteratively reweighted least squares (IRLS) algorithm is used to perform maximization in the inner loop [Jordan, M.L, Jacobs, R.A. (1994). Hierarchical mixture of experts and the EM algorithm, Neural Computation, 6(2), 181-214]. However, it is reported in literature that the IRLS algorithm is of instability and the ME architecture trained by the EM algorithm, where IRLS algorithm is used in the inner loop, often produces the poor performance in multiclass classification. In this paper, the reason of this instability is explored. We find out that due to an implicitly imposed incorrect assumption on parameter independence in multiclass classification, an incomplete Hessian matrix is used in that IRLS algorithm. Based on this finding, we apply the Newton-Raphson method to the inner loop of the EM algorithm in the case of multiclass classification, where the exact Hessian matrix is adopted. To tackle the expensive computation of the Hessian matrix and its inverse, we propose an approximation to the Newton-Raphson algorithm based on a so-called generalized Bernoulli density. The Newton-Raphson algorithm and its approximation have been applied to synthetic data, benchmark, and real-world multiclass classification tasks, For comparison, the IRLS algorithm and a quasi-Newton algorithm called BFGS have also been applied to the same tasks. Simulation results have shown that the use of the proposed learning algorithms avoids the instability problem and makes the ME architecture produce good performance in multiclass classification. In particular, our approximation algorithm leads to fast learning. In addition, the limitation of our approximation algorithm is also empirically investigated in this paper. (C) 1999 Published by Elsevier Science Ltd. All rights reserved.
All Author(s) ListChen K, Xu L, Chi H
Journal nameNeural Networks
Volume Number12
Issue Number9
Pages1229 - 1252
LanguagesEnglish-United Kingdom
KeywordsBFGS algorithm; Expectation-Maximization (EM) algorithm; generalized Bernoulli density; iterative reweighted least squares (IRLS) algorithm; mixture of experts; multiclass classification; multinomial density; Newton-Raphson method
Web of Science Subject CategoriesComputer Science; Computer Science, Artificial Intelligence; COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE; NEUROSCIENCES

Last updated on 2021-28-11 at 00:35