Machine Learning for the Prediction of Ionization Potential and Electron Affinity Energies Obtained by Density Functional Theory

Abstract

Quantum chemical (QC) calculations based on density functional theory (DFT) provide increasingly accurate estimates of various properties, but with a relatively high computational cost. Machine learning (ML) techniques can be envisaged to extract new knowledge from these large volumes of data, creating empirical models to fast predict QC calculations in new situations. Here, ML algorithms were explored for the fast estimation of ionization potential (IP) and electron affinity (EA) energies calculated by DFT using the B3LYP and PBE0 with 6–31G** basic set on molecular descriptors generated from DFT-optimized geometries. A database of 9,410 and 9,627 small organic structures for IP and EA energies modelling were used, respectively. Several ML algorithms such as random forest, support vector machines, deep learning multilayer perceptron networks, and light gradient-boosting machine were screened. The best performance was achieved with a consensus regression model predicted an external test set of 972 and 963 small organic molecules achieving a mean absolute error up to 0.23 eV and 0.32 eV for modelling IP and EA energies, respectively.