← Search

Will Grathwohl

7 accepted papers

2021

Oops I Took A Gradient: Scalable Sampling for Discrete Distributions

ICML 2021oral

We propose a general and scalable approximate sampling strategy for probabilistic models with discrete variables. Our approach uses gradients of the likelihood function with respect to its discrete inputs to propose updates in a Metropolis-Hastings sampler. We show empirically that this approach out…

2020

Learning the Stein Discrepancy for Training and Evaluating Energy-Based Models without Sampling

ICML 2020poster

We present a new method for evaluating and training unnormalized density models. Our approach only requires access to the gradient of the unnormalized model’s log-density. We estimate the Stein discrepancy between the data density p(x) and the model density q(x) based on a vector function of the dat…

2020

Understanding the Limitations of Conditional Generative Models

ICLR 2020poster

Class-conditional generative models hold promise to overcome the shortcomings of their discriminative counterparts. They are a natural choice to solve discriminative tasks in a robust manner as they jointly optimize for predictive performance and accurate modeling of the input distribution. In this…

Cited by 59SourceScholar
2020

Your classifier is secretly an energy based model and you should treat it like one

ICLR 2020talk

We propose to reinterpret a standard discriminative classifier of p(y|x) as an energy based model for the joint distribution p(x, y). In this setting, the standard class probabilities can be easily computed as well as unnormalized values of p(x) and p(x|y). Within this framework, standard discrimina…

Cited by 674SourcecodeScholar
2019

FFJORD: Free-Form Continuous Dynamics for Scalable Reversible Generative Models

ICLR 2019oral

A promising class of generative models maps points from a simple distribution to a complex distribution through an invertible neural network. Likelihood-based training of these models requires restricting their architectures to allow cheap computation of Jacobian determinants. Alternati…

Cited by 1014SourcePDFScholar
2019

Invertible Residual Networks

ICML 2019oral

We show that standard ResNet architectures can be made invertible, allowing the same model to be used for classification, density estimation, and generation. Typically, enforcing invertibility requires partitioning dimensions or restricting network architectures. In contrast, our approach only requi…

Cited by 759SourcePDFScholar
2018

Backpropagation through the Void: Optimizing control variates for black-box gradient estimation

ICLR 2018poster

Gradient-based optimization is the foundation of deep learning and reinforcement learning. Even when the mechanism being optimized is unknown or not differentiable, optimization using high-variance or biased gradient estimates is still often the best strategy. We introduce a general framework for le…