cs.LGOct 2, 2022

GFlowNets and variational inference

Authors: Esmeralda S. WhitammerSalem LahlouTristan DeleuXu JiEdward HuKatie EverettDinghuai ZhangYoshua Bengio

Organizations: Mila, Universit´e de Montr´eal · Google Research · Mila, Universit´e de Montr´eal, CIFAR

Abstract

This paper builds bridges between two families of probabilistic algorithms: (hierarchical) variational inference (VI), which is typically used to model distributions over continuous spaces, and generative flow networks (GFlowNets), which have been used for distributions over discrete structures such as graphs. We demonstrate that, in certain cases, VI algorithms are equivalent to special cases of GFlowNets in the sense of equality of expected gradients of their learning objectives. We then point out the differences between the two families and show how these differences emerge experimentally. Notably, GFlowNets, which borrow ideas from reinforcement learning, are more amenable than VI to off-policy training without the cost of high gradient variance induced by importance sampling. We argue that this property of GFlowNets can provide advantages for capturing diversity in multimodal target distributions.

Explore similar work

CardsList
  1. Proximal Policy Optimization for Amortized Discrete Sampling

    Jun 14, 2026Anna Zykova-Myzina, Timofei Gritsaev, Daniil Tiapkin +1Generative Flow NetworksFlow-Based Policy

  2. Stable GFlowNets with Probabilistic Guarantees

    May 3, 2026Zengxiang Lei, Ananth Shreekumar, Jonathan Rosenthal +6Generative Flow NetworksDistributional Learning