Verlet Flows: Exact-Likelihood Integrators for Flow-Based Generative Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Erives, Ezra, Jing, Bowen, Jaakkola, Tommi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913341633462272
author Erives, Ezra
Jing, Bowen
Jaakkola, Tommi
author_facet Erives, Ezra
Jing, Bowen
Jaakkola, Tommi
contents Approximations in computing model likelihoods with continuous normalizing flows (CNFs) hinder the use of these models for importance sampling of Boltzmann distributions, where exact likelihoods are required. In this work, we present Verlet flows, a class of CNFs on an augmented state-space inspired by symplectic integrators from Hamiltonian dynamics. When used with carefully constructed Taylor-Verlet integrators, Verlet flows provide exact-likelihood generative models which generalize coupled flow architectures from a non-continuous setting while imposing minimal expressivity constraints. On experiments over toy densities, we demonstrate that the variance of the commonly used Hutchinson trace estimator is unsuitable for importance sampling, whereas Verlet flows perform comparably to full autograd trace computations while being significantly faster.
format Preprint
id arxiv_https___arxiv_org_abs_2405_02805
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Verlet Flows: Exact-Likelihood Integrators for Flow-Based Generative Models
Erives, Ezra
Jing, Bowen
Jaakkola, Tommi
Machine Learning
Approximations in computing model likelihoods with continuous normalizing flows (CNFs) hinder the use of these models for importance sampling of Boltzmann distributions, where exact likelihoods are required. In this work, we present Verlet flows, a class of CNFs on an augmented state-space inspired by symplectic integrators from Hamiltonian dynamics. When used with carefully constructed Taylor-Verlet integrators, Verlet flows provide exact-likelihood generative models which generalize coupled flow architectures from a non-continuous setting while imposing minimal expressivity constraints. On experiments over toy densities, we demonstrate that the variance of the commonly used Hutchinson trace estimator is unsuitable for importance sampling, whereas Verlet flows perform comparably to full autograd trace computations while being significantly faster.
title Verlet Flows: Exact-Likelihood Integrators for Flow-Based Generative Models
topic Machine Learning
url https://arxiv.org/abs/2405.02805