Gaussian Process-Gated Hierarchical Mixtures of Experts

Yuhao Liu, Marzieh Ajirak, Petar Djuric
2/9/2023
cs.LGstat.ML

No official code found. Best implementation: yuhaoliu94/GP-HME (2 stars, Apache-2.0, last commit 2024-06-28).

Abstract

In this paper, we propose novel Gaussian process-gated hierarchical mixtures of experts (GPHMEs). Unlike other mixtures of experts with gating models linear in the input, our model employs gating functions built with Gaussian processes (GPs). These processes are based on random features that are non-linear functions of the inputs. Furthermore, the experts in our model are also constructed with GPs. The optimization of the GPHMEs is performed by variational inference. The proposed GPHMEs have several advantages. They outperform tree-based HME benchmarks that partition the data in the input space, and they achieve good performance with reduced complexity. Another advantage is the interpretability they provide for deep GPs, and more generally, for deep Bayesian neural networks. Our GPHMEs demonstrate excellent performance for large-scale data sets, even with quite modest sizes.

View on arXivView PDF

Code Implementations(1)

yuhaoliu94/GP-HMECommunity100%
Apache-2.0

Gaussian Process-Gated Hierarchical Mixture of Experts

20PythonFeb 24, 20222 years agoApache-2.0
decision-treesgaussian-processesmixture-of-experts

Cite this paper

@article{liu2023gaussian,
  title  = {Gaussian Process-Gated Hierarchical Mixtures of Experts},
  author = {Yuhao Liu and Marzieh Ajirak and Petar Djuric},
  year   = {2023},
  eprint = {2302.04947},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2302.04947v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2302.04947.svg)](https://codeofpaper.com/paper/2302.04947)

Discussion