Skip to yearly menu bar Skip to main content


Poster
in
Workshop: New Frontiers of AI for Drug Discovery and Development

DGFN: Double Generative Flow Networks

Elaine Lau · Nikhil Murali Vemgal · Doina Precup · Emmanuel Bengio

Keywords: [ GFlowNets ] [ Q-Learning ] [ Generative Modelling ] [ Drug Discovery ] [ Reinforcement Learning ]


Abstract:

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) are a recently introduced method recognized for the ability to generate diverse candidates, in particular in small molecule generation tasks. In this work, we introduce double GFlowNets (DGFNs). Drawing inspiration from reinforcement learning and Double Deep Q-Learning, we introduce a target network used to sample trajectories, while updating the main network with these sampled trajectories. Empirical results confirm that DGFNs effectively enhance exploration in sparse reward domains and high-dimensional state spaces, both challenging aspects of de-novo design in drug discovery.

Chat is not available.