Skip to yearly menu bar Skip to main content


Poster

Implicit Diffusion: Efficient optimization through stochastic sampling

Pierre Marion · Anna Korba · Peter Bartlett · Mathieu Blondel · Valentin De Bortoli · Arnaud Doucet · Felipe Llinares-López · Courtney Paquette · Quentin Berthet

Hall A-E 7
[ ]
 
Oral presentation: Oral Session 3: Optimization
Sat 3 May 8:30 p.m. PDT — 9:30 p.m. PDT

Abstract:

Sampling and automatic differentiation are both ubiquitous in modern machine learning. At its intersection, differentiating through a sampling operation, with respect to the parameters of the sampling process, is a problem that is both challenging and broadly applicable. We introduce a general framework and a new algorithm for first-order optimization of parameterized stochastic diffusions, performing jointly, in a single loop, optimization and sampling steps. This approach is inspired by recent advances in bilevel optimization and automatic implicit differentiation, leveraging the point of view of sampling as optimization over the space of probability distributions. We provide theoretical and experimental results showcasing the performance of our method.

Live content is unavailable. Log in and register to view live content