7 papers
Stop the Sampler! Classifier-Based Adaptive Stopping for Sampling Kernels
Kirill Korolev, Nikita Morozov, Stepan Pavlenko +2
Sampling from complex, unnormalized probability densities is a fundamental challenge in Bayesian inference and probabilistic modeling. While Markov chain Monte Carlo (MCMC) methods…
Your GFlowNet Secretly Learns an Optimal Transport Plan
Ian Maksimov, Nikita Morozov, Denis Belomestny +1
Generative Flow Networks (GFlowNets) are a framework for sampling structured objects via stochastic trajectories in a directed graph. In this work, we establish a theoretical conne…
Learning Shortest Paths with Generative Flow Networks
Nikita Morozov, Ian Maksimov, Daniil Tiapkin +1
In this paper, we present a novel learning framework for finding shortest paths in graphs utilizing Generative Flow Networks (GFlowNets). First, we examine theoretical properties o…
gfnx: Fast and Scalable Library for Generative Flow Networks in JAX
Daniil Tiapkin, Artem Agarkov, Nikita Morozov +4
In this paper, we present gfnx, a fast and scalable package for training and evaluating Generative Flow Networks (GFlowNets) written in JAX. gfnx provides an extensive set of envir…
Revisiting Non-Acyclic GFlowNets in Discrete Environments
Nikita Morozov, Ian Maksimov, Daniil Tiapkin +1
Generative Flow Networks (GFlowNets) are a family of generative models that learn to sample objects from a given probability distribution, potentially known up to a normalizing con…
Adaptive Destruction Processes for Diffusion Samplers
Timofei Gritsaev, Nikita Morozov, Kirill Tamogashev +5
This paper explores the challenges and benefits of a trainable destruction process in diffusion samplers -- diffusion-based generative models trained to sample an unnormalised dens…