Tree based machine learning framework for predicting ground state energies of molecules

Abstract

We present an application of the boosted regression tree algorithm for predicting ground state energies of molecules made up of C, H, N, O, P, and S (CHNOPS). The PubChem chemical compound database has been incorporated to construct a dataset of 16,242 molecules, whose electronic ground state energies have been computed using density functional theory. This dataset is used to train the boosted regression tree algorithm, which allows a computationally efficient and accurate prediction of molecular ground state energies. Predictions from boosted regression trees are compared with neural network regression, a widely used method in the literature, and shown to be more accurate with significantly reduced computational cost. The performance of the regression model trained using the CHNOPS set is also tested on a set of distinct molecules that contain additional Cl and Si atoms. It is shown that the learning algorithms lead to a rich and diverse possibility of applications in molecular discovery and materials informatics.

23 pages, 8 figures, 4 tables. Data and scripts can be found at https://github.com/bhimmetoglu/RoboBohr

References in corpus (10)

• Quantum ESPRESSO: a modular and open-source software project for quantum simulations of materials
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• AFLOW: An automatic framework for high-throughput materials discovery
• Predictive learning via rule ensembles
• Finding Density Functionals with Machine Learning
• Machine Learning of Molecular Electronic Properties in Chemical Compound Space
• AiiDA: Automated Interactive Infrastructure and Database for Computational Science
• How to represent crystal structures for machine learning: towards fast prediction of electronic properties
• Prediction model of band-gap for AX binary compounds by combination of density functional theory calculations and machine learning techniques
• A high-throughput ab initio review of platinum-group alloy systems

Cited by in corpus (5)

• Quantum Machine Learning for Chemistry and Physics
• Design and Analysis of Machine Learning Exchange-Correlation Functionals via Rotationally Invariant Convolutional Descriptors
• Disorder information from conductance: a quantum inverse problem
• Spatial mapping of disordered 2D materials: the conductance Sudoku
• Decoding the conductance of disordered nanostructures: a quantum inverse problem