Abstract:In this work, we have proposed a generative model, called VAE-KRnet, for density estimation or approximation, which combines the canonical variational autoencoder (VAE) with our recently developed flow-based generative model, called KRnet. VAE is used as a dimension reduction technique to capture the latent space, and KRnet is used to model the distribution of the latent variable. Using a linear model between the data and the latent variable, we show that VAE-KRnet can be more effective and robust than the canonical VAE. VAE-KRnet can be used as a density model to approximate either data distribution or an arbitrary probability density function (PDF) known up to a constant. VAE-KRnet is flexible in terms of dimensionality. When the number of dimensions is relatively small, KRnet can effectively approximate the distribution in terms of the original random variable. For high-dimensional cases, we may use VAE-KRnet to incorporate dimension reduction. One important application of VAE-KRnet is the variational Bayes for the approximation of the posterior distribution. The variational Bayes approaches are usually based on the minimization of the Kullback-Leibler (KL) divergence between the model and the posterior. For high-dimensional distributions, it is very challenging to construct an accurate density model due to the curse of dimensionality, where extra assumptions are often introduced for efficiency. For instance, the classical mean-field approach assumes mutual independence between dimensions, which often yields an underestimated variance due to oversimplification. To alleviate this issue, we include into the loss the maximization of the mutual information between the latent random variable and the original random variable, which helps keep more information from the region of low density such that the estimation of variance is improved.

Minimax optimal density estimation using a shallow generative model with a one-dimensional latent variable

VAE-KRnet and its applications to variational Bayes

Deep data density estimation through Donsker-Varadhan representation

Generalized Gumbel-Softmax Gradient Estimator for Generic Discrete Random Variables

Hidden Talents of the Variational Autoencoder

Near-Minimax Optimal Estimation With Shallow ReLU Neural Networks

Rates of convergence for nonparametric estimation of singular distributions using generative adversarial networks

Generalized multi-view model: Adaptive density estimation under low-rank constraints

Adaptive Density Estimation for Generative Models

Density-Softmax: Efficient Test-time Model for Uncertainty Estimation and Robustness under Distribution Shifts

Concurrent Density Estimation with Wasserstein Autoencoders: Some Statistical Insights

Minimax density estimation for growing dimension

Learning Multi-layer Latent Variable Model via Variational Optimization of Short Run MCMC for Approximate Inference

On the capacity of deep generative networks for approximating distributions

A likelihood approach to nonparametric estimation of a singular distribution using deep generative models

Model-free Estimation of Latent Structure via Multiscale Nonparametric Maximum Likelihood

Evidential Softmax for Sparse Multimodal Distributions in Deep Generative Models

Learning Implicit Generative Models by Teaching Density Estimators

Modeling Extremes with d-max-decreasing Neural Networks

Improving Variational Autoencoders with Density Gap-based Regularization

Asymptotically minimax Bayesian predictive densities for multinomial models