README.md 4.31 KB
Newer Older
1
2
3
4
5
6
7
# Work Implemented in DGL-LifeSci

We provide various examples across 3 applications -- property prediction, generative models and protein-ligand binding affinity prediction.

## Datasets/Benchmarks

- MoleculeNet: A Benchmark for Molecular Machine Learning [[paper]](https://arxiv.org/abs/1703.00564), [[website]](http://moleculenet.ai/)
Mufei Li's avatar
Mufei Li committed
8
9
    - [Tox21 with DGL](../python/dgllife/data/tox21.py)
    - [PDBBind with DGL](../python/dgllife/data/pdbbind.py)
10
- Alchemy: A Quantum Chemistry Dataset for Benchmarking AI Models [[paper]](https://arxiv.org/abs/1906.09427), [[github]](https://github.com/tencent-alchemy/Alchemy)
Mufei Li's avatar
Mufei Li committed
11
    - [Alchemy with DGL](../python/dgllife/data/alchemy.py)
12
13
14

## Property Prediction

15
16
17
- Molecular graph convolutions: moving beyond fingerprints (Weave) [[paper]](https://arxiv.org/abs/1603.00856), [[github]](https://github.com/deepchem/deepchem)
    - [Weave Predictor with DGL](../python/dgllife/model/model_zoo/weave_predictor.py)
    - [Example for Molecule Classification](property_prediction/classification.py)
18
- Semi-Supervised Classification with Graph Convolutional Networks (GCN) [[paper]](https://arxiv.org/abs/1609.02907), [[github]](https://github.com/tkipf/gcn)
Mufei Li's avatar
Mufei Li committed
19
    - [GCN-Based Predictor with DGL](../python/dgllife/model/model_zoo/gcn_predictor.py)
20
21
    - [Example for Molecule Classification](property_prediction/classification.py)
- Graph Attention Networks (GAT) [[paper]](https://arxiv.org/abs/1710.10903), [[github]](https://github.com/PetarV-/GAT)
Mufei Li's avatar
Mufei Li committed
22
    - [GAT-Based Predictor with DGL](../python/dgllife/model/model_zoo/gat_predictor.py)
23
24
    - [Example for Molecule Classification](property_prediction/classification.py)
- SchNet: A continuous-filter convolutional neural network for modeling quantum interactions [[paper]](https://arxiv.org/abs/1706.08566), [[github]](https://github.com/atomistic-machine-learning/SchNet)
Mufei Li's avatar
Mufei Li committed
25
    - [SchNet with DGL](../python/dgllife/model/model_zoo/schnet_predictor.py)
26
27
    - [Example for Molecule Regression](property_prediction/regression.py)
- Molecular Property Prediction: A Multilevel Quantum Interactions Modeling Perspective (MGCN) [[paper]](https://arxiv.org/abs/1906.11081)
Mufei Li's avatar
Mufei Li committed
28
    - [MGCN with DGL](../python/dgllife/model/model_zoo/mgcn_predictor.py)
29
30
    - [Example for Molecule Regression](property_prediction/regression.py)
- Neural Message Passing for Quantum Chemistry (MPNN) [[paper]](https://arxiv.org/abs/1704.01212), [[github]](https://github.com/brain-research/mpnn)
Mufei Li's avatar
Mufei Li committed
31
    - [MPNN with DGL](../python/dgllife/model/model_zoo/mpnn_predictor.py)
32
33
    - [Example for Molecule Regression](property_prediction/regression.py)
- Pushing the Boundaries of Molecular Representation for Drug Discovery with the Graph Attention Mechanism (AttentiveFP) [[paper]](https://pubs.acs.org/doi/abs/10.1021/acs.jmedchem.9b00959)
Mufei Li's avatar
Mufei Li committed
34
    - [AttentiveFP with DGL](../python/dgllife/model/model_zoo/attentivefp_predictor.py)
35
36
37
38
39
    - [Example for Molecule Regression](property_prediction/regression.py)

## Generative Models

- Learning Deep Generative Models of Graphs (DGMG) [[paper]](https://arxiv.org/abs/1803.03324)
Mufei Li's avatar
Mufei Li committed
40
    - [DGMG with DGL](../python/dgllife/model/model_zoo/dgmg.py)
41
42
    - [Example Training Script](generative_models/dgmg)
- Junction Tree Variational Autoencoder for Molecular Graph Generation (JTNN) [[paper]](https://arxiv.org/abs/1802.04364)
Mufei Li's avatar
Mufei Li committed
43
    - [JTNN with DGL](../python/dgllife/model/model_zoo/jtnn)
44
45
46
47
48
    - [Example Training Script](generative_models/jtnn)

## Binding Affinity Prediction

- Atomic Convolutional Networks for Predicting Protein-Ligand Binding Affinity (ACNN) [[paper]](https://arxiv.org/abs/1703.10603), [[github]](https://github.com/deepchem/deepchem/tree/master/contrib/atomicconv)
Mufei Li's avatar
Mufei Li committed
49
    - [ACNN with DGL](../python/dgllife/model/model_zoo/acnn.py)
50
51
52
53
54
    - [Example Training Script](binding_affinity_prediction)

## Reaction Prediction
- A graph-convolutional neural network model for the prediction of chemical reactivity [[paper]](https://pubs.rsc.org/en/content/articlelanding/2019/sc/c8sc04228d#!divAbstract), [[github]](https://github.com/connorcoley/rexgen_direct)
    - An earlier version was published in NeurIPS 2017 as "Predicting Organic Reaction Outcomes with Weisfeiler-Lehman Network" [[paper]](https://arxiv.org/abs/1709.04555)
Mufei Li's avatar
Mufei Li committed
55
    - [WLN with DGL for Reaction Center Prediction](../python/dgllife/model/model_zoo/wln_reaction_center.py)
56
    - [Example Script](reaction_prediction/rexgen_direct)