NOTE: The code now only supports tensorflow-gpu >= 2.0.
Code for the purposes of Direct Steering of de novo Molecular Generation using Descriptor Conditional Recurrent Neural Networks (cRNNs).
Cheers if you were brought here by this blog post. If not, give it a read :)
Deep learning has acquired considerable momentum over the past couple of years in the domain of de-novo drug design. Particularly, transfer and reinforcement learning have demonstrated the capability of steering the generative process towards chemical regions of interest. In this work, we propose a simple approach to the focused generative task by constructing a conditional recurrent neural network (cRNN). For this purpose, we aggregate selected molecular descriptors along with a QSAR-based bioactivity label and transform them into initial LSTM states before starting the generation of SMILES strings that are focused towards the aspired properties. We thus tackle the inverse QSAR problem directly by training on molecular descriptors, instead of iteratively optimizing around a set of candidate molecules. The trained cRNNs are able to generate molecules near multiple specified conditions, while maintaining an output that is more focused than traditional RNNs yet less focused than autoencoders. The method shows promise for applications in both scaffold hoping and ligand series generation, depending on whether the cRNN is trained on calculated scalar molecular properties or structural fingerprints. This also demonstrates that fingerprint-to-molecule decoding is feasible, leading to molecules that are similar – if not identical – to the ones the fingerprints originated from. Additionally, the cRNN is able to generate a larger fraction of predicted active compounds against the DRD2 receptor when compared to an RNN trained with the transfer learning model.
Only a GPU version of the model is supported. You need access to a GPU to use it.
Please refer to the demo notebooks for usage details.
- Install
git-lfsas instructed here. This is necessary in order to download the datasets. - Clone the repo and navigate to it.
- Create a predefined Python3.6 conda environment by
conda env create -f env/ddc_env.yml. This ensures that you have the correct version ofrdKitandcudatoolkit. - Run
pip install .to install remaining dependencies and add the package to the Python path. - Add the environment in the drop-down list of jupyter by
python -m ipykernel install --user --name ddc_env --display-name "ddc_env (python_3.6.7)".
conda activate ddc_envfrom ddc_pub import ddc_v3 as ddcfit(): Fit a DDC model to the dataset.vectorize(): Convert a binary RDKit molecule to its One-Hot-Encoded representation.transform(): Encode a vectorized molecule to its latent representation.predict(): Decode a latent representation into a SMILES string and return its Negative Log Likelihood (NLL).predict_batch(): Decode a list of latent representations into SMILES strings and return their NLLs.get_smiles_nll(): Back-calculate the NLL of a known SMILES string, if it was to be sampled by the biased decoder.get_smiles_nll_batch(): Back-calculate the NLLs of a batch of known SMILES strings, if they were to be sampled by the biased decoder.summary(): Display essential architectural parameters.get_graphs(): Export model graphs to .png files usingpydotandgraphviz(might fail).save(): Save the model in a .zip directory.
Please report all installation / usage issues by opening an issue at this repo.
Kotsias, P.-C. et al. Direct steering of de novo molecular generation with descriptor conditional recurrent neural networks. Nat. Mach. Intell. 2, (2020)
or in bibtex:
@article{Kotsias2020,
isbn = {4225602001},
issn = {2522-5839},
journal = {Nature Machine Intelligence},
number = {May},
publisher = {Springer US},
title = {Direct steering of de novo molecular generation with descriptor conditional recurrent neural networks},
url = {https://doi.org/10.1038/s42256-020-0174-5},
volume = {2},
year = {2020}
}