Graph networks for molecular design

Roco Mercado,Tobias Rastemo,Edvard Lindelf,Gnter Klambauer,Ola Engkvist,Hongming Chen,Esben Jannik Bjerrum,Rocío Mercado,Edvard Lindelöf,Günter Klambauer
DOI: https://doi.org/10.1088/2632-2153/abcf91
2021-03-02
Machine Learning: Science and Technology
Abstract:Abstract Deep learning methods applied to chemistry can be used to accelerate the discovery of new molecules. This work introduces GraphINVENT, a platform developed for graph-based molecular design using graph neural networks (GNNs). GraphINVENT uses a tiered deep neural network architecture to probabilistically generate new molecules a single bond at a time. All models implemented in GraphINVENT can quickly learn to build molecules resembling the training set molecules without any explicit programming of chemical rules. The models have been benchmarked using the MOSES distribution-based metrics, showing how GraphINVENT models compare well with state-of-the-art generative models. This work compares six different GNN-based generative models in GraphINVENT, and shows that ultimately the gated-graph neural network performs best against the metrics considered here.
What problem does this paper attempt to address?