Prediction Errors of Molecular Machine Learning Models Lower than Hybrid DFT Error

被引:442
作者
Faber, Felix A. [1 ,2 ]
Hutchison, Luke [3 ]
Huang, Bing [1 ,2 ]
Gilmer, Justin [3 ]
Schoenholz, Samuel S. [3 ]
Dahl, George E. [3 ]
Vinyals, Oriol [4 ]
Kearnes, Steven [3 ]
Riley, Patrick F. [3 ]
von Lilienfeld, O. Anatole [1 ,2 ]
机构
[1] Univ Basel, Dept Chem, Inst Phys Chem, Klingelbergstr 80, CH-4056 Basel, Switzerland
[2] Univ Basel, Dept Chem, Natl Ctr Computat Design & Discovery Novel Mat, Klingelbergstr 80, CH-4056 Basel, Switzerland
[3] Google, 1600 Amphitheatre Pkwy, Mountain View, CA 94043 USA
[4] Google, 5 New St Sq, London EC4A 3TW, England
基金
瑞士国家科学基金会;
关键词
FORCE-FIELD; PARAMETERS; KERNEL;
D O I
10.1021/acs.jctc.7b00577
中图分类号
O64 [物理化学(理论化学)、化学物理学];
学科分类号
070304 ; 081704 ;
摘要
We investigate the impact of choosing regressors and molecular representations for the construction of fast machine learning (ML) models of 13 electronic ground-state properties of organic molecules. The performance of each regressor/representation/property combination is assessed using learning curves which report out-of-sample errors as a function of training set size with up to similar to 118k distinct molecules. Molecular structures and properties at the hybrid density functional theory (DFT) level of theory come from the QM9 database [Ramakrishnan et al. Sri. Data 2014 1, 140022] and include enthalpies and free energies of atomization, HOMO/LUMO energies and gap, dipole moment, polarizability, zero point-vibrational energy, heat capacity, and the highest fundamental vibrational frequency. Various molecular representations have been studied (Coulomb matrix, bag of bonds, BAML and ECFP4, molecular graphs (MG)), as well as newly developed distribution based variants including histograms of distances (HD), angles (HDA/MARAD), and dihedrals (HDAD). Regressors include linear models (Bayesian ridge regression (BR) and linear regression with elastic net regularization (EN)), random forest (RF), kernel ridge regression (KRR), and two types of neural networks, graph convolutions (GC) and gated graph networks (GG). Out-of sample errors are strongly dependent on the choice of representation and regressor and molecular property. Electronic properties are typically best accounted for by MG and GC, while energetic properties are better described by HDAD and KRK The specific combinations with the lowest out-of-sample errors in the similar to 118k training set size limit are (free) energies and enthalpies of atomization (HDAD/KRR), HOMO/LUMO eigenvalue and gap (MG/GC), dipole moment (MG/GC), static polarizability (MG/GG), zero point vibrational energy (HDAD/KRR), heat capacity at room temperature (HDAD/KRR), and highest fundamental vibrational frequency (BAML/RF). We present numerical evidence that ML model predictions deviate from DFT (B3LYP) less than DFT (B3LYP) deviates from experiment for all properties. Furthermore, out-of-sample prediction errors with respect to hybrid DFT reference are on par with, or close to, chemical accuracy. The results suggest that ML models could be more accurate than hybrid DFT if explicitly electron correlated quantum (or experimental) data were available.
引用
收藏
页码:5255 / 5264
页数:10
相关论文
共 63 条
  • [1] [Anonymous], 2017, Nature
  • [2] [Anonymous], ARXIV170600179
  • [3] [Anonymous], 2016, ARXIV161105126
  • [4] On representing chemical environments
    Bartok, Albert P.
    Kondor, Risi
    Csanyi, Gabor
    [J]. PHYSICAL REVIEW B, 2013, 87 (18)
  • [5] Automated design of ligands to polypharmacological profiles
    Besnard, Jeremy
    Ruda, Gian Filippo
    Setola, Vincent
    Abecassis, Keren
    Rodriguiz, Ramona M.
    Huang, Xi-Ping
    Norval, Suzanne
    Sassano, Maria F.
    Shin, Antony I.
    Webster, Lauren A.
    Simeons, Frederick R. C.
    Stojanovski, Laste
    Prat, Annik
    Seidah, Nabil G.
    Constam, Daniel B.
    Bickerton, G. Richard
    Read, Kevin D.
    Wetsel, William C.
    Gilbert, Ian H.
    Roth, Bryan L.
    Hopkins, Andrew L.
    [J]. NATURE, 2012, 492 (7428) : 215 - +
  • [6] Random forests
    Breiman, L
    [J]. MACHINE LEARNING, 2001, 45 (01) : 5 - 32
  • [7] Perspective on density functional theory
    Burke, Kieron
    [J]. JOURNAL OF CHEMICAL PHYSICS, 2012, 136 (15)
  • [8] Challenges for Density Functional Theory
    Cohen, Aron J.
    Mori-Sanchez, Paula
    Yang, Weitao
    [J]. CHEMICAL REVIEWS, 2012, 112 (01) : 289 - 320
  • [9] Collins C. R., 2016, ARXIV170106649
  • [10] Assessment of Gaussian-2 and density functional theories for the computation of enthalpies of formation
    Curtiss, LA
    Raghavachari, K
    Redfern, PC
    Pople, JA
    [J]. JOURNAL OF CHEMICAL PHYSICS, 1997, 106 (03) : 1063 - 1079