Command Palette
Search for a command to run...
Justin Gilmer Samuel S. Schoenholz Patrick F. Riley Oriol Vinyals George E. Dahl

摘要
分子上的监督学习在化学、药物发现和材料科学中具有巨大的应用潜力。幸运的是,已有文献描述了几种有前景且与分子对称性无关的神经网络模型。这些模型通过学习消息传递算法和聚合过程来计算其整个输入图的函数。目前,下一步是找到这一通用方法的一个特别有效的变体,并将其应用于化学预测基准测试,直到我们解决问题或达到该方法的极限。在本文中,我们将现有模型重新表述为一个统一的框架,称为消息传递神经网络(Message Passing Neural Networks, MPNNs),并在该框架内探索额外的新变体。利用MPNNs,我们在一个重要分子性质预测基准测试中展示了当前最佳的结果;这些结果足够强大,我们认为未来的研究应集中于包含更大分子的数据集或更准确的真实标签数据集。
代码仓库
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| drug-discovery-on-qm9 | MPNNs | Error ratio: 0.68 |
| formation-energy-on-qm9 | MPNN | MAE: 0.49 |
| graph-regression-on-lipophilicity | MPNN | RMSE: 0.719 |
| graph-regression-on-zinc-100k | MPNN | MAE: 0.288 |
| graph-regression-on-zinc-500k | MPNN (sum) | MAE: 0.145 |
| graph-regression-on-zinc-500k | MPNN (max) | MAE: 0.252 |
| node-classification-on-citeseer-with-public | MPNN | Accuracy: 64.0 |