English
Related papers

Related papers: Towards Understanding and Improving GFlowNet Train…

200 papers

While Markov chain Monte Carlo methods (MCMC) provide a general framework to sample from a probability distribution defined up to normalization, they often suffer from slow convergence to the target distribution when the latter is highly…

Machine Learning · Computer Science 2023-07-06 Tristan Deleu , Yoshua Bengio

Generative flow networks (GFNs) are a class of models for sequential sampling of composite objects, which approximate a target distribution that is defined in terms of an energy function or a reward. GFNs are typically trained using a flow…

Machine Learning · Statistics 2022-10-17 Heiko Zimmermann , Fredrik Lindsten , Jan-Willem van de Meent , Christian A. Naesseth

Generative Flow Networks (GFlowNets) enable structured generation with inherent diversity, but existing sampling strategies often rely on weak guided exploration, slowing early discovery of high-reward candidates. In tasks such as molecular…

Machine Learning · Computer Science 2026-02-03 Rui Zhu , Yudong Zhang , Xuan Yu , Chen Zhang , Xu Wang , Yang Wang

Generative Flow Networks (GFlowNets) aim to generate diverse trajectories from a distribution in which the final states of the trajectories are proportional to the reward, serving as a powerful alternative to reinforcement learning for…

Artificial Intelligence · Computer Science 2024-08-14 Shuang Luo , Yinchuan Li , Shunyu Liu , Xu Zhang , Yunfeng Shao , Chao Wu

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 of GFlowNets in non-acyclic environments in relation to…

Machine Learning · Computer Science 2026-03-03 Nikita Morozov , Ian Maksimov , Daniil Tiapkin , Sergey Samsonov

Finding the best way to schedule operations in a computation graph is a classical NP-hard problem which is central to compiler optimization. However, evaluating the goodness of a schedule on the target hardware can be very time-consuming.…

Artificial Intelligence · Computer Science 2023-02-15 David W. Zhang , Corrado Rainone , Markus Peschl , Roberto Bondesan

Generative Flow Networks, or GFlowNets, offer a promising framework for molecular design, but their internal decision policies remain opaque. This limits adoption in drug discovery, where chemists require clear and interpretable rationales…

Machine Learning · Computer Science 2025-11-25 Amirtha Varshini A S , Duminda S. Ranasinghe , Hok Hei Tam

Advancements in robotics have opened possibilities to automate tasks in various fields such as manufacturing, emergency response and healthcare. However, a significant challenge that prevents robots from operating in real-world environments…

Robotics · Computer Science 2025-01-08 Zahin Sufiyan , Shadan Golestan , Shotaro Miwa , Yoshihiro Mitsuka , Osmar Zaiane

Unitary Synthesis, the decomposition of a unitary matrix into a sequence of quantum gates, is a fundamental challenge in quantum compilation. Prevailing reinforcement learning (RL) approaches are often hampered by sparse reward signals,…

Quantum Physics · Physics 2026-03-05 Inhoe Koo , Hyunho Cha , Jungwoo Lee

Generative Flow Networks (GFlowNets) treat sampling from distributions over compositional discrete spaces as a sequential decision-making problem, training a stochastic policy to construct objects step by step. Recent studies have revealed…

Machine Learning · Computer Science 2024-06-21 Nikita Morozov , Daniil Tiapkin , Sergey Samsonov , Alexey Naumov , Dmitry Vetrov

The performance of sensor arrays in sensing and wireless communications improves with more elements, but this comes at the cost of increased energy consumption and hardware expense. This work addresses the challenge of selecting $k$ sensor…

Machine Learning · Computer Science 2024-07-30 Spilios Evmorfos , Zhaoyi Xu , Athina Petropulu

The scalability of pool-based active learning is limited by the computational cost of evaluating large unlabeled datasets, a challenge that is particularly acute in virtual screening for drug discovery. While active learning strategies such…

Machine Learning · Computer Science 2025-09-03 Renfei Zhang , Mohit Pandey , Artem Cherkasov , Martin Ester

The use of guidance to steer sampling toward desired outcomes has been widely explored within diffusion models, especially in applications such as image and trajectory generation. However, incorporating guidance during training remains…

Machine Learning · Computer Science 2025-05-21 Marvin Alles , Nutan Chen , Patrick van der Smagt , Botond Cseke

Although Generative Flow Networks (GFlowNets) are designed to capture multiple modes of a reward function, they often suffer from mode collapse in practice, getting trapped in early-discovered modes and requiring prolonged training to find…

Machine Learning · Computer Science 2025-11-11 Idriss Malek , Aya Laajil , Abhijith Sharma , Eric Moulines , Salem Lahlou

GFlowNets have exhibited promising performance in generating diverse candidates with high rewards. These networks generate objects incrementally and aim to learn a policy that assigns probability of sampling objects in proportion to…

Machine Learning · Computer Science 2024-06-11 George Ma , Emmanuel Bengio , Yoshua Bengio , Dinghuai Zhang

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 sampling-based approaches that leverage controlled stochastic…

Machine Learning · Computer Science 2024-03-12 Dinghuai Zhang , Ricky T. Q. Chen , Cheng-Hao Liu , Aaron Courville , Yoshua Bengio

We introduce BatchGFN -- a novel approach for pool-based active learning that uses generative flow networks to sample sets of data points proportional to a batch reward. With an appropriate reward function to quantify the utility of…

Machine Learning · Computer Science 2023-06-28 Shreshth A. Malik , Salem Lahlou , Andrew Jesson , Moksh Jain , Nikolay Malkin , Tristan Deleu , Yoshua Bengio , Yarin Gal

Generative Flow Networks (GFNs) have emerged as a powerful tool for sampling discrete objects from unnormalized distributions, offering a scalable alternative to Markov Chain Monte Carlo (MCMC) methods. While GFNs draw inspiration from…

Machine Learning · Computer Science 2025-02-24 Sobhan Mohammadpour , Emmanuel Bengio , Emma Frejinger , Pierre-Luc Bacon

Personalized recommender systems fulfill the daily demands of customers and boost online businesses. The goal is to learn a policy that can generate a list of items that matches the user's demand or interest. While most existing methods…

Information Retrieval · Computer Science 2023-06-12 Shuchang Liu , Qingpeng Cai , Zhankui He , Bowen Sun , Julian McAuley , Dong Zheng , Peng Jiang , Kun Gai

We explore a data-driven approach for learning to optimize neural networks. We construct a dataset of neural network checkpoints and train a generative model on the parameters. In particular, our model is a conditional diffusion transformer…

Machine Learning · Computer Science 2022-09-27 William Peebles , Ilija Radosavovic , Tim Brooks , Alexei A. Efros , Jitendra Malik