MolecularRNN: Generating realistic molecular graphs with optimized properties
Designing new molecules with a set of predefined properties is a core problem in modern drug discovery and development. There is a growing need for de-novo design methods that would address this problem. We present MolecularRNN, the graph recurrent generative model for molecular structures. Our model generates diverse realistic molecular graphs after likelihood pretraining on a big database of molecules. We perform an analysis of our pretrained models on large-scale generated datasets of 1 million samples. Further, the model is tuned with policy gradient algorithm, provided a critic that estimates the reward for the property of interest. We show a significant distribution shift to the desired range for lipophilicity, drug-likeness, and melting point outperforming state-of-the-art works. With the use of rejection sampling based on valency constraints, our model yields 100% validity. Moreover, we show that invalid molecules provide a rich signal to the model through the use of structure penalty in our reinforcement learning pipeline.
Code (2)
Tasks
Drug DiscoveryMolecular Graph GenerationReinforcement LearningSimilar Papers 제목 키워드 기반
MARS: Markov Molecular Sampling for Multi-objective Drug Discovery
Searching for novel molecules with desired chemical properties is crucial in drug discovery. Existing work focuses on developing neural models to generate either molecular sequences or chemical graphs. However, it remain…
Drug DiscoveryGraph Neural NetworkMolecular Graph GenerationJunction Tree Variational Autoencoder for Molecular Graph Generation
We seek to automate the design of molecules based on specific chemical properties. In computational terms, this task involves continuous embedding and generation of molecular graphs. Our primary contribution is the direc…
Drug DiscoveryGraph GenerationMolecular Graph GenerationMoFlow: An Invertible Flow Model for Generating Molecular Graphs
Generating molecular graphs with desired chemical properties driven by deep graph generative models provides a very promising way to accelerate drug discovery process. Such graph generative models usually consist of two …
Drug DiscoveryGraph GenerationMolecular Graph GenerationvalidScaling Novel Graph Generation via Lightweight Structure-Guided Autoregressive Models
Generating realistic and diverse graphs is a key problem in machine learning, with applications in molecular discovery, circuit design, cybersecurity, and beyond. However, current graph generative models remain limited b…
Graph GenerationDeep Learning for Molecular Graphs with Tiered Graph Autoencoders and Graph Prediction
Tiered graph autoencoders provide the architecture and mechanisms for learning tiered latent representations and latent spaces for molecular graphs that explicitly represent and utilize groups (e.g., functional groups). …
General ClassificationGraph ClassificationPrediction