Variational Control for Guidance in Diffusion Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Pandey, Kushagra, Sofian, Farrin Marouf, Draxler, Felix, Karaletsos, Theofanis, Mandt, Stephan
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915302855409664
author Pandey, Kushagra
Sofian, Farrin Marouf
Draxler, Felix
Karaletsos, Theofanis
Mandt, Stephan
author_facet Pandey, Kushagra
Sofian, Farrin Marouf
Draxler, Felix
Karaletsos, Theofanis
Mandt, Stephan
contents Diffusion models exhibit excellent sample quality, but existing guidance methods often require additional model training or are limited to specific tasks. We revisit guidance in diffusion models from the perspective of variational inference and control, introducing Diffusion Trajectory Matching (DTM) that enables guiding pretrained diffusion trajectories to satisfy a terminal cost. DTM unifies a broad class of guidance methods and enables novel instantiations. We introduce a new method within this framework that achieves state-of-the-art results on several linear, non-linear, and blind inverse problems without requiring additional model training or specificity to pixel or latent space diffusion models. Our code will be available at https://github.com/czi-ai/oc-guidance
format Preprint
id arxiv_https___arxiv_org_abs_2502_03686
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Variational Control for Guidance in Diffusion Models
Pandey, Kushagra
Sofian, Farrin Marouf
Draxler, Felix
Karaletsos, Theofanis
Mandt, Stephan
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Diffusion models exhibit excellent sample quality, but existing guidance methods often require additional model training or are limited to specific tasks. We revisit guidance in diffusion models from the perspective of variational inference and control, introducing Diffusion Trajectory Matching (DTM) that enables guiding pretrained diffusion trajectories to satisfy a terminal cost. DTM unifies a broad class of guidance methods and enables novel instantiations. We introduce a new method within this framework that achieves state-of-the-art results on several linear, non-linear, and blind inverse problems without requiring additional model training or specificity to pixel or latent space diffusion models. Our code will be available at https://github.com/czi-ai/oc-guidance
title Variational Control for Guidance in Diffusion Models
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.03686