Local Component Analysis

TL;DR

This paper introduces a method to learn full Euclidean metrics for kernel density estimation using EM, improving density estimation and unsupervised learning tasks like clustering and manifold learning.

Contribution

It proposes a novel EM-based approach to learn full metrics in kernel density estimation, including semi-parametric models, with closed-form updates and scalability enhancements.

Findings

Higher test-likelihoods than competing methods

Metrics improve spectral clustering and manifold learning

Effective in large-scale settings

Abstract

Kernel density estimation, a.k.a. Parzen windows, is a popular density estimation method, which can be used for outlier detection or clustering. With multivariate data, its performance is heavily reliant on the metric used within the kernel. Most earlier work has focused on learning only the bandwidth of the kernel (i.e., a scalar multiplicative factor). In this paper, we propose to learn a full Euclidean metric through an expectation-minimization (EM) procedure, which can be seen as an unsupervised counterpart to neighbourhood component analysis (NCA). In order to avoid overfitting with a fully nonparametric density estimator in high dimensions, we also consider a semi-parametric Gaussian-Parzen density model, where some of the variables are modelled through a jointly Gaussian density, while others are modelled through Parzen windows. For these two models, EM leads to simple closed-form updates based on matrix inversions and eigenvalue decompositions. We show empirically that our method leads to density estimators with higher test-likelihoods than natural competing methods, and that the metrics may be used within most unsupervised learning techniques that rely on such metrics, such as spectral clustering or manifold learning methods. Finally, we present a stochastic approximation scheme which allows for the use of this method in a large-scale setting.