Block Neural Autoregressive Flow

By Nicola De Cao, Ivan Titov, Wilker Aziz. 2019

142 citations recorded by the source corpus.

Normalising flows (NFS) map two density functions via a differentiable bijection whose Jacobian determinant can be computed efficiently. Recently, as an alternative to hand-crafted bijections, Huang et al. (2018) proposed neural autoregressive flow (NAF) which is a universal approximator for density functions. Their flow is a neural network (NN) whose parameters are predicted by another NN. The latter grows quadratically with the size of the former and thus an efficient technique for parametrization is needed. We propose block neural autoregressive flow (B-NAF), a much more compact universal approximator of density functions, where we model a bijection directly using a single feed-forward network. Invertibility is ensured by carefully designing each affine transformation with block matrices that make the flow autoregressive and (strictly) monotone. We compare B-NAF to NAF and other established flows on density estimation and approximate inference for latent variable models. Our proposed flow is competitive across datasets while using orders of magnitude fewer parameters.

Subfields: Artificial Intelligence, Generative Models, Model Architectures

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Balaji Lakshminarayanan
  2. Andrew Gordon Wilson
  3. Gilles Louppe
  4. Pavel Izmailov
  5. Danilo Jimenez Rezende
  6. Nicola De Cao
  7. Shakir Mohamed
  8. George Papamakarios

In Generative Models

  1. Gilles Louppe
  2. Balaji Lakshminarayanan
  3. Frank No'e
  4. Shakir Mohamed
  5. Danilo Jimenez Rezende
  6. George Papamakarios
  7. Jonas Köhler
  8. Nicola De Cao

In Model Architectures

  1. Gilles Louppe
  2. Antoine Wehenkel
  3. Nicola De Cao
  4. Jonas Köhler
  5. Frank No'e
  6. Leon Klein
  7. Danilo Jimenez Rezende
  8. Yang Long

Citations in the corpus