activity
20152025
most citedConditional Computation in Neural Networks for faster models

143 citations · 365 across the 31 of their papers we have counts for

collaborators

31 papers

cs.LG2025

Torsional-GFN: a conditional conformation generator for small molecules

Alexandra Volokhova, Léna Néhale Ezzine, Piotr Gaiński +5

Generating stable molecular conformations is crucial in several drug discovery applications, such as estimating the binding affinity of a molecule to a target. Recently, generative…

cs.LG2025

Solving Bayesian inverse problems with diffusion priors and off-policy RL

Luca Scimeca, Siddarth Venkatraman, Moksh Jain +14

This paper presents a practical application of Relative Trajectory Balance (RTB), a recently introduced off-policy reinforcement learning (RL) objective that can asymptotically sol…

cs.LG2025★ 1 cited

Pretraining Generative Flow Networks with Inexpensive Rewards for Molecular Graph Generation

Mohit Pandey, Gopeshh Subbaraj, Artem Cherkasov +2

Generative Flow Networks (GFlowNets) have recently emerged as a suitable framework for generating diverse and high-quality molecular structures by learning from rewards treated as…

cs.LG2024

Action abstractions for amortized sampling

Oussama Boussif, Léna Néhale Ezzine, Joseph D Viviano +6

As trajectories sampled by policies used by reinforcement learning (RL) and generative flow networks (GFlowNets) grow longer, credit assignment and exploration become more challeng…

cs.LG2024

Improved Off-policy Reinforcement Learning in Biological Sequence Design

Hyeonah Kim, Minsu Kim, Taeyoung Yun +4

Designing biological sequences with desired properties is challenging due to vast search spaces and limited evaluation budgets. Although reinforcement learning methods use proxy mo…

cs.LG2024

Adaptive teachers for amortized samplers

Minsu Kim, Sanghyeok Choi, Taeyoung Yun +7

Amortized inference is the task of training a parametric model, such as a neural network, to approximate a distribution with a given unnormalized density where exact sampling is in…

cs.LG2024

GFlowNet Pretraining with Inexpensive Rewards

Mohit Pandey, Gopeshh Subbaraj, Emmanuel Bengio

Generative Flow Networks (GFlowNets), a class of generative models have recently emerged as a suitable framework for generating diverse and high-quality molecular structures by lea…

cs.LG2024

Baking Symmetry into GFlowNets

George Ma, Emmanuel Bengio, Yoshua Bengio +1

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

cs.LG2024

Random Policy Evaluation Uncovers Policies of Generative Flow Networks

Haoran He, Emmanuel Bengio, Qingpeng Cai +1

The Generative Flow Network (GFlowNet) is a probabilistic framework in which an agent learns a stochastic policy and flow functions to sample objects proportionally to an unnormali…

q-bio.BM2024★ 5 cited

Generative Active Learning for the Search of Small-molecule Protein Binders

Maksym Korablyov, Cheng-Hao Liu, Moksh Jain +31

Despite substantial progress in machine learning for scientific discovery in recent years, truly de novo design of small molecules which exhibit a property of interest remains a si…

cs.LG2024★ 1 cited

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

Siddarth Venkatraman, Moksh Jain, Luca Scimeca +12

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…

cs.LG2024

Investigating Generalization Behaviours of Generative Flow Networks

Lazar Atanackovic, Emmanuel Bengio

Generative Flow Networks (GFlowNets, GFNs) are a generative framework for learning unnormalized probability mass functions over discrete spaces. Since their inception, GFlowNets ha…

cs.LG2024

QGFN: Controllable Greediness with Action Values

Elaine Lau, Stephen Zhewen Lu, Ling Pan +2

Generative Flow Networks (GFlowNets; GFNs) are a family of energy-based generative methods for combinatorial objects, capable of generating diverse and high-utility samples. Howeve…

cs.LG2023

Maximum entropy GFlowNets with soft Q-learning

Sobhan Mohammadpour, Emmanuel Bengio, Emma Frejinger +1

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…

cs.LG2023

DGFN: Double Generative Flow Networks

Elaine Lau, Nikhil Vemgal, Doina Precup +1

Deep learning is emerging as an effective tool in drug discovery, with potential applications in both predictive and generative models. Generative Flow Networks (GFlowNets/GFNs) ar…

cs.LG2023

Local Search GFlowNets

Minsu Kim, Taeyoung Yun, Emmanuel Bengio +4

Generative Flow Networks (GFlowNets) are amortized sampling methods that learn a distribution over discrete objects proportional to their rewards. GFlowNets exhibit a remarkable ab…

cs.LG2023★ 1 cited

Goal-conditioned GFlowNets for Controllable Multi-Objective Molecular Design

Julien Roy, Pierre-Luc Bacon, Christopher Pal +1

In recent years, in-silico molecular design has received much attention from the machine learning community. When designing a new compound for pharmaceutical applications, there ar…

cs.LG2023★ 1 cited

Towards Understanding and Improving GFlowNet Training

Max W. Shen, Emmanuel Bengio, Ehsan Hajiramezanali +3

Generative flow networks (GFlowNets) are a family of algorithms that learn a generative policy to sample discrete objects with non-negative reward . Learning objectives g…

cs.LG2022★ 4 cited

Multi-Objective GFlowNets

Moksh Jain, Sharath Chandra Raparthy, Alex Hernandez-Garcia +4

We study the problem of generating diverse candidates in the context of Multi-Objective Optimization. In many applications of machine learning such as drug discovery and material d…

cs.LG2022★ 6 cited

Learning GFlowNets from partial episodes for improved convergence and stability

Kanika Madan, Jarrid Rector-Brooks, Maksym Korablyov +6

Generative flow networks (GFlowNets) are a family of algorithms for training a sequential sampler of discrete objects under an unnormalized target density and have been successfull…

q-bio.BM2022★ 21 cited

Biological Sequence Design with GFlowNets

Moksh Jain, Emmanuel Bengio, Alex-Hernandez Garcia +10

Design of de novo biological sequences with desired properties, like protein and DNA sequences, often involves an active loop with several rounds of molecule ideation and expensive…

cs.LG2022★ 7 cited

Trajectory balance: Improved credit assignment in GFlowNets

Esmeralda S. Whitammer, Moksh Jain, Emmanuel Bengio +2

Generative flow networks (GFlowNets) are a method for learning a stochastic policy for generating compositional objects, such as graphs or strings, from a given unnormalized densit…

cs.LG2021

GFlowNet Foundations

Yoshua Bengio, Salem Lahlou, Tristan Deleu +3

Generative Flow Networks (GFlowNets) have been introduced as a method to sample a diverse set of candidates in an active learning context, with a training objective that makes them…

cs.LG2021★ 1 cited

Correcting Momentum in Temporal Difference Learning

Emmanuel Bengio, Joelle Pineau, Doina Precup

A common optimization tool used in deep reinforcement learning is momentum, which consists in accumulating and discounting past gradients, reapplying them at each iteration. We arg…