深度潜变量核学习

Deep Latent-Variable Kernel Learning

IEEE Transactions on Cybernetics · 2021
被引 8
ABS 3

中文导读

提出一种深度潜变量核学习模型,通过随机编码实现正则化表示,结合神经随机微分方程提升近似质量,在小数据集上表现与高斯过程相当,大数据集上更优。

Abstract

Deep kernel learning (DKL) leverages the connection between the Gaussian process (GP) and neural networks (NNs) to build an end-to-end hybrid model. It combines the capability of NN to learn rich representations under massive data and the nonparametric property of GP to achieve automatic regularization that incorporates a tradeoff between model fit and model complexity. However, the deterministic NN encoder may weaken the model regularization of the following GP part, especially on small datasets, due to the free latent representation. We, therefore, present a complete deep latent-variable kernel learning (DLVKL) model wherein the latent variables perform stochastic encoding for regularized representation. We further enhance the DLVKL from two aspects: 1) the expressive variational posterior through neural stochastic differential equation (NSDE) to improve the approximation quality and 2) the hybrid prior taking knowledge from both the SDE prior and the posterior to arrive at a flexible tradeoff. Extensive experiments imply that DLVKL-NSDE performs similar to the well-calibrated GP on small datasets, and shows superiority on large datasets.

机器学习高斯过程深度学习核方法潜变量模型