Deep generative models of 3D grids for structure-based drug discovery

Last update: Jan 03, 2023

Overview

What is liGAN?

liGAN is a research codebase for training and evaluating deep generative models for de novo drug design based on 3D atomic density grids. It is based on libmolgrid and the gnina fork of caffe.

VAE paper - 2 minute talk

CVAE paper - 15 minute talk

Dependencies

numpy
pandas
scikit-image
openbabel
rdkit
molgrid
torch
protobuf
gnina version of caffe

Usage

You can use the scripts download_data.sh and download_weights.sh to download the test data and weights that were evaluated in the above papers.

The script generate.py is used to generate atomic density grids and molecular structures from a trained generative model.

Its basic usage can be seen in the scripts generate_vae.sh:

LIG_FILE=$1 # e.g. data/molport/0/102906000_8.sdf

python3 generate.py \
  --data_model_file models/data_48_0.5_molport.model \
  --gen_model_file models/vae.model \
  --gen_weights_file weights/gen_e_0.1_1_disc_x_10_0.molportFULL_rand_.0.0_gen_iter_100000.caffemodel \
  --rec_file data/molport/10gs_rec.pdb \
  --lig_file $LIG_FILE \
  --out_prefix VAE \
  --n_samples 10 \
  --fit_atoms \
  --dkoes_make_mol \
  --output_sdf \
  --output_dx \
  --gpu

And generate_cvae.sh:

REC_FILE=$1 # e.g. data/crossdock2020/PARP1_HUMAN_775_1012_0/2rd6_A_rec.pdb
LIG_FILE=$2 # e.g. data/crossdock2020/PARP1_HUMAN_775_1012_0/2rd6_A_rec_2rd6_78p_lig_tt_min.sdf

python3 generate.py \
  --data_model_file models/data_48_0.5_crossdock.model \
  --gen_model_file models/cvae.model \
  --gen_weights_file weights/lessskip_crossdocked_increased_1.lowrmsd.0_gen_iter_1500000.caffemodel \
  --rec_file $REC_FILE \
  --lig_file $LIG_FILE \
  --out_prefix CVAE \
  --n_samples 10 \
  --fit_atoms \
  --dkoes_make_mol \
  --output_sdf \
  --output_dx \
  --gpu

Both scripts can be run from the root directory of the repository.

Deep generative models of 3D grids for structure-based drug discovery

Related tags

Overview

What is liGAN?

Dependencies

Usage

Owner

Matt Ragoza

This repository contains tutorials for the py4DSTEM Python package

PyTorch implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC

TagLab: an image segmentation tool oriented to marine data analysis

Keras udrl - Keras implementation of Upside Down Reinforcement Learning

Semantic segmentation models, datasets and losses implemented in PyTorch.

LightLog is an open source deep learning based lightweight log analysis tool for log anomaly detection.

Advanced Deep Learning with TensorFlow 2 and Keras (Updated for 2nd Edition)

OpenPose: Real-time multi-person keypoint detection library for body, face, hands, and foot estimation

Code for the paper "Unsupervised Contrastive Learning of Sound Event Representations", ICASSP 2021.

A pytorch implementation of faster RCNN detection framework (Use detectron2, it's a masterpiece)

PyTorch framework for Deep Learning research and development.

Dynamic vae - Dynamic VAE algorithm is used for anomaly detection of battery data

TransZero++: Cross Attribute-guided Transformer for Zero-Shot Learning

An example of Scatterbrain implementation (combining local attention and Performer)

Editing a classifier by rewriting its prediction rules

Official implementation of Deep Convolutional Dictionary Learning for Image Denoising.

Turning SymPy expressions into PyTorch modules.

ERISHA is a mulitilingual multispeaker expressive speech synthesis framework. It can transfer the expressivity to the speaker's voice for which no expressive speech corpus is available.

Re-implementation of the Noise Contrastive Estimation algorithm for pyTorch, following "Noise-contrastive estimation: A new estimation principle for unnormalized statistical models." (Gutmann and Hyvarinen, AISTATS 2010)

Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.