A community-powered search of machine learning strategy space to find NMR property prediction models

L. A. Bratholm, Will Gerrard, Brandon Anderson, Shaojie Bai, Sunghwan Choi, Lam Dang, Pavel Hanchar, Addison Howard, Sanghoon Kim, Zico Kolter, Risi Kondor, Mordechai Kornbluth, Youhan Lee, Youngsoo Lee, Jonathan P. Mailoa, Thanh Tu Nguyen, Milos Popovic, Goran Rakocevic, Walter Reade, Wonho SongLuka Stojanovic, Erik H. Thiede, Nebojsa Tijanic, Andres Torrubia, Devin Willmott, Craig P. Butts, David R. Glowacki*

*Corresponding author for this work

Research output: Contribution to journalArticle (Academic Journal)peer-review

1 Downloads (Pure)

Abstract

The rise of machine learning (ML) has created an explosion in the potential strategies for using data to make scientific predictions. For physical scientists wishing to apply ML strategies to a particular domain, it can be difficult to assess in advance what strategy to adopt within a vast space of possibilities. Here we outline the results of an online community-powered effort to swarm search the space of ML strategies and develop algorithms for predicting atomic-pairwise nuclear magnetic resonance (NMR) properties in molecules. Using an open-source dataset, we worked with Kaggle to design and host a 3-month competition which received 47,800 ML model predictions from 2,700 teams in 84 countries. Within 3 weeks, the Kaggle community produced models with comparable accuracy to our best previously published in-house efforts. A meta-ensemble model constructed as a linear combination of the top predictions has a prediction accuracy which exceeds that of any individual model, 7-19x better than our previous state-of-The-Art. The results highlight the potential of transformer architectures for predicting quantum mechanical (QM) molecular properties.

Original languageEnglish
Article numbere0253612
JournalPLoS ONE
Volume16
Issue number7
DOIs
Publication statusPublished - 20 Jul 2021

Bibliographical note

Publisher Copyright:
© 2021 Public Library of Science. All rights reserved.

Fingerprint

Dive into the research topics of 'A community-powered search of machine learning strategy space to find NMR property prediction models'. Together they form a unique fingerprint.

Cite this