Kernel Mean Embedding of Distributions: A Review and Beyond

TL;DR

This paper reviews the kernel mean embedding framework, highlighting its theoretical foundations, applications in machine learning, and open problems, emphasizing its versatility in handling probability measures in RKHS.

Contribution

It provides a comprehensive survey of kernel mean embeddings, covering recent advances, theoretical guarantees, and diverse applications across machine learning and statistical inference.

Findings

Kernel mean embedding enables non-parametric inference on distributions.

It facilitates applications like two-sample testing and probabilistic modeling.

The framework connects to various areas such as causal discovery and deep learning.

Abstract

A Hilbert space embedding of a distribution---in short, a kernel mean embedding---has recently emerged as a powerful tool for machine learning and inference. The basic idea behind this framework is to map distributions into a reproducing kernel Hilbert space (RKHS) in which the whole arsenal of kernel methods can be extended to probability measures. It can be viewed as a generalization of the original "feature map" common to support vector machines (SVMs) and other kernel methods. While initially closely associated with the latter, it has meanwhile found application in fields ranging from kernel machines and probabilistic modeling to statistical inference, causal discovery, and deep learning. The goal of this survey is to give a comprehensive review of existing work and recent advances in this research area, and to discuss the most challenging issues and open problems that could lead to new research directions. The survey begins with a brief introduction to the RKHS and positive definite kernels which forms the backbone of this survey, followed by a thorough discussion of the Hilbert space embedding of marginal distributions, theoretical guarantees, and a review of its applications. The embedding of distributions enables us to apply RKHS methods to probability measures which prompts a wide range of applications such as kernel two-sample testing, independent testing, and learning on distributional data. Next, we discuss the Hilbert space embedding for conditional distributions, give theoretical insights, and review some applications. The conditional mean embedding enables us to perform sum, product, and Bayes' rules---which are ubiquitous in graphical model, probabilistic inference, and reinforcement learning---in a non-parametric way. We then discuss relationships between this framework and other related areas. Lastly, we give some suggestions on future research directions.