aalto1 untyped-item.component.html

Conditioning diffusion models by explicit forward-backward bridging

Loading...
Thumbnail Image

Access rights

openAccess
CC BY

Creative Commons license

Except where otherwised noted, this item's license is described as openAccess
publishedVersion

URL

Journal Title

Journal ISSN

Volume Title

A4 Artikkeli konferenssijulkaisussa

Date

Major/Subject

Mcode

Degree programme

Language

en

Pages

9

Series

Proceedings of Machine Learning Research, Volume 258, pp. 3709-3717

Abstract

Given an unconditional diffusion model targeting a joint model π(x, y), using it to perform conditional simulation π(x | y) is still largely an open question and is typically achieved by learning conditional drifts to the denoising SDE after the fact. In this work, we express exact conditional simulation within the approximate diffusion model as an inference problem on an augmented space corresponding to a partial SDE bridge. This perspective allows us to implement efficient and principled particle Gibbs and pseudo-marginal samplers marginally targeting the conditional distribution π(x | y). Contrary to existing methodology, our methods do not introduce any additional approximation to the unconditional diffusion model aside from the Monte Carlo error. We showcase the benefits and drawbacks of our approach on a series of synthetic and real data examples.

Description

Publisher Copyright: Copyright 2025 by the author(s).

Keywords

Other note

Citation

Corenflos, A, Zhao, Z, Särkkä, S, Sjölund, J & Schön, T B 2025, 'Conditioning diffusion models by explicit forward-backward bridging', Proceedings of Machine Learning Research, vol. 258, pp. 3709-3717. < https://raw.githubusercontent.com/mlresearch/v258/main/assets/corenflos25a/corenflos25a.pdf >

Endorsement

Review

Supplemented By

Referenced By