MolGPT: Molecular Generation Using a Transformer-Decoder Model

By Viraj Bagal, Rishal Aggarwal, P. K. Vinod, U. Priyakumar. Journal of Chemical Information and Modeling, 2021

527 citations recorded by the source corpus.

Application of deep learning techniques for de novo generation of molecules, termed as inverse molecular design, has been gaining enormous traction in drug design. The representation of molecules in SMILES notation as a string of characters enables the usage of state of the art models in natural language processing, such as Transformers, for molecular design in general. Inspired by generative pre-training (GPT) models that have been shown to be successful in generating meaningful text, we train a transformer-decoder on the next token prediction task using masked self-attention for the generation of druglike molecules in this study. We show that our model, MolGPT, performs on par with other previously proposed modern machine learning frameworks for molecular generation in terms of generating valid, unique, and novel molecules. Furthermore, we demonstrate that the model can be trained conditionally to control multiple properties of the generated molecules. We also show that the model can be used to generate molecules with desired scaffolds as well as desired molecular properties by conditioning the generation on scaffold SMILES strings of desired scaffolds and property values. Using saliency maps, we highlight the interpretability of the generative process of the model.

Subfields: Computational Theory and Mathematics, Materials Chemistry, Molecular Biology

View Paper: https://www.semanticscholar.org/paper/77805d75199e7b9e580b4827f56a069ba0ddd13f

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Theory and Mathematics

  1. Heng Ji
  2. Tuan Manh Lai
  3. Jiawei Han
  4. Garrett Honke
  5. Carl N. Edwards
  6. Kevin Ros
  7. Cynthia Rudin
  8. Tong Wang

In Materials Chemistry

  1. Heng Ji
  2. Meng Jiang

In Molecular Biology

  1. Heng Ji
  2. Meng Jiang
  3. Jiawei Han
  4. Bowen Jin
  5. Chi Han
  6. Gang Liu

Citations in the corpus