David Duvenaud, Dougal Maclaurin, Jorge Aguilera-Iparraguirre, Rafael Gómez-Bombarelli, Timothy Hirzel, Alán Aspuru-Guzik, Ryan P. Adams
We introduce a convolutional neural network that operates directly on graphs. These networks allow end-to-end learning of prediction pipelines whose inputs are graphs of arbitrary size and shape. The architecture we present generalizes standard molecular feature extraction methods based on circular fingerprints. We show that these data-driven features are more interpretable, and have better predictive performance on a variety of tasks.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Drug Discovery | Tox21 | AUC | 0.846 | GraphConv |
| Drug Discovery | HIV dataset | AUC | 0.822 | GraphConv |
| Drug Discovery | ToxCast | AUC | 0.754 | GraphConv |
| Drug Discovery | MUV | AUC | 0.836 | GraphConv |
| Drug Discovery | PCBA | AUC | 0.855 | GraphConv |
| Graph Regression | Lipophilicity | RMSE | 0.655 | GC |