Relative Entropy Gradient Sampler for Unnormalized Distributions

Xingdong Feng, Yuan Gao, Jian Huang, Yuling Jiao, Xu Liu

Keywords: Density ratio estimation, gradient flow, neural networks, particles, velocity fields

Abstract: We propose a relative entropy gradient sampler (REGS) for sampling from unnormalized distributions. REGS is a particle method that seeks a sequence of simple nonlinear transforms iteratively pushing the initial samples from a reference distribution into the samples from an unnormalized target distribution. To determine the nonlinear transforms at each iteration, we consider the Wasserstein gradient flow of relative entropy. This gradient flow determines a path of probability distributions that interpolates the reference distribution and the target distribution. It is characterized by an ODE system with velocity fields depending on the density ratios of the density of evolving particles and the unnormalized target density. To sample with REGS, we need to estimate the density ratios and simulate the ODE system with particle evolution. We propose a novel nonparametric approach to estimating the logarithmic density ratio using neural networks. Extensive simulation studies on challenging multimodal 1D and 2D distributions and Bayesian logistic regression on real datasets demonstrate that the REGS outperforms the state-of-the-art sampling methods included in the comparison.

One-sentence Summary: We proposed a novel sampling method for unnormalized distributions based on relative entropy gradient flow and neural network approximation.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/arxiv:2110.02787/code)