Gradient Estimation for Binary Latent Variables via Gradient Variance Clipping

Russell Z. Kunes, Mingzhang Yin, Max Land +3

Gradient estimation is often necessary for fitting generative models with discrete latent variables, in contexts such as reinforcement learning and variational autoencoder (VAE) tr…