Chemical diversity in molecular orbital energy predictions with kernel ridge regression

Annika Stuke, Milica Todorović, Matthias Rupp, Christian Kunkel, Kunal Ghosh, Lauri Himanen, Patrick Rinke

Research output: Contribution to journalArticleScientificpeer-review

54 Citations (Scopus)
262 Downloads (Pure)


Instant machine learning predictions of molecular properties are desirable for materials design, but the predictive power of the methodology is mainly tested on well-known benchmark datasets. Here, we investigate the performance of machine learning with kernel ridge regression (KRR) for the prediction of molecular orbital energies on three large datasets: the standard QM9 small organic molecules set, amino acid and dipeptide conformers, and organic crystal-forming molecules extracted from the Cambridge Structural Database. We focus on the prediction of highest occupied molecular orbital (HOMO) energies, computed at the density-functional level of theory. Two different representations that encode the molecular structure are compared: the Coulomb matrix (CM) and the many-body tensor representation (MBTR). We find that KRR performance depends significantly on the chemistry of the underlying dataset and that the MBTR is superior to the CM, predicting HOMO energies with a mean absolute error as low as 0.09 eV. To demonstrate the power of our machine learning method, we apply our model to structures of 10k previously unseen molecules. We gain instant energy predictions that allow us to identify interesting molecules for future applications.

Original languageEnglish
Article number204121
Pages (from-to)1-13
JournalJournal of Chemical Physics
Issue number20
Publication statusPublished - 28 May 2019
MoE publication typeA1 Journal article-refereed


Dive into the research topics of 'Chemical diversity in molecular orbital energy predictions with kernel ridge regression'. Together they form a unique fingerprint.

Cite this