Control, Transport and Sampling: Towards Better Loss Design

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jiang, Qijia, Nabergoj, David
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910643129417728
author Jiang, Qijia
Nabergoj, David
author_facet Jiang, Qijia
Nabergoj, David
contents Leveraging connections between diffusion-based sampling, optimal transport, and stochastic optimal control through their shared links to the Schrödinger bridge problem, we propose novel objective functions that can be used to transport $ν$ to $μ$, consequently sample from the target $μ$, via optimally controlled dynamics. We highlight the importance of the pathwise perspective and the role various optimality conditions on the path measure can play for the design of valid training losses, the careful choice of which offer numerical advantages in implementation. Basing the formalism on Schrödinger bridge comes with the additional practical capability of baking in inductive bias when it comes to Neural Network training.
format Preprint
id arxiv_https___arxiv_org_abs_2405_13731
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Control, Transport and Sampling: Towards Better Loss Design
Jiang, Qijia
Nabergoj, David
Machine Learning
Computation
Leveraging connections between diffusion-based sampling, optimal transport, and stochastic optimal control through their shared links to the Schrödinger bridge problem, we propose novel objective functions that can be used to transport $ν$ to $μ$, consequently sample from the target $μ$, via optimally controlled dynamics. We highlight the importance of the pathwise perspective and the role various optimality conditions on the path measure can play for the design of valid training losses, the careful choice of which offer numerical advantages in implementation. Basing the formalism on Schrödinger bridge comes with the additional practical capability of baking in inductive bias when it comes to Neural Network training.
title Control, Transport and Sampling: Towards Better Loss Design
topic Machine Learning
Computation
url https://arxiv.org/abs/2405.13731