Free-form Flows: Make Any Architecture a Normalizing Flow

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Draxler, Felix, Sorrenson, Peter, Zimmermann, Lea, Rousselot, Armand, Köthe, Ullrich
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910420293386240
author Draxler, Felix
Sorrenson, Peter
Zimmermann, Lea
Rousselot, Armand
Köthe, Ullrich
author_facet Draxler, Felix
Sorrenson, Peter
Zimmermann, Lea
Rousselot, Armand
Köthe, Ullrich
contents Normalizing Flows are generative models that directly maximize the likelihood. Previously, the design of normalizing flows was largely constrained by the need for analytical invertibility. We overcome this constraint by a training procedure that uses an efficient estimator for the gradient of the change of variables formula. This enables any dimension-preserving neural network to serve as a generative model through maximum likelihood training. Our approach allows placing the emphasis on tailoring inductive biases precisely to the task at hand. Specifically, we achieve excellent results in molecule generation benchmarks utilizing $E(n)$-equivariant networks. Moreover, our method is competitive in an inverse problem benchmark, while employing off-the-shelf ResNet architectures.
format Preprint
id arxiv_https___arxiv_org_abs_2310_16624
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Free-form Flows: Make Any Architecture a Normalizing Flow
Draxler, Felix
Sorrenson, Peter
Zimmermann, Lea
Rousselot, Armand
Köthe, Ullrich
Machine Learning
Normalizing Flows are generative models that directly maximize the likelihood. Previously, the design of normalizing flows was largely constrained by the need for analytical invertibility. We overcome this constraint by a training procedure that uses an efficient estimator for the gradient of the change of variables formula. This enables any dimension-preserving neural network to serve as a generative model through maximum likelihood training. Our approach allows placing the emphasis on tailoring inductive biases precisely to the task at hand. Specifically, we achieve excellent results in molecule generation benchmarks utilizing $E(n)$-equivariant networks. Moreover, our method is competitive in an inverse problem benchmark, while employing off-the-shelf ResNet architectures.
title Free-form Flows: Make Any Architecture a Normalizing Flow
topic Machine Learning
url https://arxiv.org/abs/2310.16624