Let the flows tell: Solving graph combinatorial problems with gflownets

D Zhang, H Dai, N Malkin… - Advances in neural …, 2023 - proceedings.neurips.cc
Combinatorial optimization (CO) problems are often NP-hard and thus out of reach for exact
algorithms, making them a tempting domain to apply machine learning methods. The highly …

Amortizing intractable inference in large language models

EJ Hu, M Jain, E Elmoznino, Y Kaddar, G Lajoie… - arxiv preprint arxiv …, 2023 - arxiv.org
Autoregressive large language models (LLMs) compress knowledge from their training data
through next-token conditional distributions. This limits tractable querying of this knowledge …

GFlowNet-EM for learning compositional latent variable models

EJ Hu, N Malkin, M Jain, KE Everett… - International …, 2023 - proceedings.mlr.press
Latent variable models (LVMs) with discrete compositional latents are an important but
challenging setting due to a combinatorially large number of possible configurations of the …

Gflownets for ai-driven scientific discovery

M Jain, T Deleu, J Hartford, CH Liu… - Digital …, 2023 - pubs.rsc.org
Tackling the most pressing problems for humanity, such as the climate crisis and the threat
of global pandemics, requires accelerating the pace of scientific discovery. While science …

Amortizing intractable inference in diffusion models for vision, language, and control

S Venkatraman, M Jain, L Scimeca, M Kim… - arxiv preprint arxiv …, 2024 - arxiv.org
Diffusion models have emerged as effective distribution estimators in vision, language, and
reinforcement learning, but their use as priors in downstream tasks poses an intractable …

Generative flow networks as entropy-regularized rl

D Tiapkin, N Morozov, A Naumov… - International …, 2024 - proceedings.mlr.press
The recently proposed generative flow networks (GFlowNets) are a method of training a
policy to sample compositional discrete objects with probabilities proportional to a given …

Diffusion generative flow samplers: Improving learning signals through partial trajectory optimization

D Zhang, RTQ Chen, CH Liu, A Courville… - arxiv preprint arxiv …, 2023 - arxiv.org
We tackle the problem of sampling from intractable high-dimensional density functions, a
fundamental task that often appears in machine learning and statistics. We extend recent …

Stochastic generative flow networks

L Pan, D Zhang, M Jain, L Huang… - Uncertainty in Artificial …, 2023 - proceedings.mlr.press
Abstract Generative Flow Networks (or GFlowNets for short) are a family of probabilistic
agents that learn to sample complex combinatorial structures through the lens of “inference …

Local search gflownets

M Kim, T Yun, E Bengio, D Zhang, Y Bengio… - arxiv preprint arxiv …, 2023 - arxiv.org
Generative Flow Networks (GFlowNets) are amortized sampling methods that learn a
distribution over discrete objects proportional to their rewards. GFlowNets exhibit a …

Improved off-policy training of diffusion samplers

M Sendera, M Kim, S Mittal, P Lemos… - Advances in …, 2025 - proceedings.neurips.cc
We study the problem of training diffusion models to sample from a distribution with a given
unnormalized density or energy function. We benchmark several diffusion-structured …