Generative Flow Networks (GFlowNets) are used for molecular graph generation.
Previous methods restricted exploration by using predefined molecular fragments.
Atomic GFlowNets (A-GFNs) introduce a new generative model using individual atoms as building blocks for drug-like chemical space exploration.
Unsupervised pre-training with drug-like molecule datasets is proposed for A-GFNs, focusing on molecular descriptors like drug-likeliness and synthetic accessibility scores as rewards.
These rewards guide A-GFNs towards regions of chemical space with desired pharmacological properties.
Goal-conditioned finetuning helps adapt A-GFNs for specific target properties.
Pretraining A-GFN on a subset of ZINC dataset shows effectiveness in drug design tasks compared to baseline methods.
The code for A-GFN is available at https://github.com/diamondspark/AGFN.