Abstract

A normalizing flow models a complex probability density as an invertibletransformation of a simple base density. Flows based on either coupling orautoregressive transforms both offer exact density evaluation and sampling, butrely on the parameterization of an easily invertible elementwisetransformation, whose choice determines the flexibility of these models.Building upon recent work, we propose a fully-differentiable module based onmonotonic rational-quadratic splines, which enhances the flexibility of bothcoupling and autoregressive transforms while retaining analytic invertibility.We demonstrate that neural spline flows improve density estimation, variationalinference, and generative modeling of images.

 

Quick Read (beta)