arxiv.org
Implicit Diffusion: Efficient Optimization through Stochastic Sampling
February 2024 • Pierre Marion, Anna Korba, Peter L. Bartlett, Mathieu Blondel, Valentin De Bortoli, Randal Douc, Felipe Llinares-López, Courtney Paquette, Quentin Be…
We present a new algorithm to optimize distributions defined implicitly by parameterized stochastic diffusions. Doing so allows us to modify the outcome distribution of sampling processes by optimizing over their parameters. We introduce a general framework for first-order optimization of these processes, that performs 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 …