Abstract:The development of accurate and transferable machine learning (ML) potentials for predicting molecular energetics is a challenging task. The process of data generation to train such ML potentials is a task neither well understood nor researched in detail. In this work, we present a fully automated approach for the generation of datasets with the intent of training universal ML potentials. It is based on the concept of active learning (AL) via Query by Committee (QBC), which uses the disagreement between an ensemble of ML potentials to infer the reliability of the ensemble's prediction. QBC allows the presented AL algorithm to automatically sample regions of chemical space where the ML potential fails to accurately predict the potential energy. AL improves the overall fitness of ANAKIN-ME (ANI) deep learning potentials in rigorous test cases by mitigating human biases in deciding what new training data to use. AL also reduces the training set size to a fraction of the data required when using naive random sampling techniques. To provide validation of our AL approach we develop the COMP6 benchmark (publicly available on GitHub), which contains a diverse set of organic molecules. Through the AL process, it is shown that the AL-based potentials perform as well as the ANI-1 potential on COMP6 with only 10% of the data, and vastly outperforms ANI-1 with 25% the amount of data. Finally, we show that our proposed AL technique develops a universal ANI potential (ANI-1x) that provides accurate energy and force predictions on the entire COMP6 benchmark. This universal ML potential achieves a level of accuracy on par with the best ML potentials for single molecule or materials, while remaining applicable to the general class of organic molecules comprised of the elements CHNO.

Exploring chemical and conformational spaces by batch mode deep active learning

Less is more: sampling chemical space with active learning

Active learning strategies for atomic cluster expansion models

Machine learning of molecular properties: locality and active learning

A transferable active-learning strategy for reactive molecular force fields

Active learning of linearly parametrized interatomic potentials

In operando active learning of interatomic interaction during large-scale simulations

Learning Neural Generative Dynamics for Molecular Conformation Generation

Active Learning of Molecular Data for Task-Specific Objectives

Finding the most potent compounds using active learning on molecular pairs

Traversing Chemical Space with Active Deep Learning: A Computational Framework for Low-data Drug Discovery

Training Transferable Interatomic Neural Network Potentials for Reactive Chemistry: Improved Chemical Space Sampling

Deep Learning for Multi-Scale Molecular Modeling

Active Deep Kernel Learning of Molecular Functionalities: Realizing Dynamic Structural Embeddings

CURATOR: Building Robust Machine Learning Potentials for Atomistic Simulations Autonomously with Batch Active Learning

Charting electronic-state manifolds across molecules with multi-state learning and gap-driven dynamics via efficient and robust active learning

On-the-fly active learning of interpretable Bayesian force fields for atomistic rare events

Optimal Molecular Design: Generative Active Learning Combining REINVENT with Precise Binding Free Energy Ranking Simulations

Ab-Initio Molecular Dynamics Acceleration Scheme with an Adaptive Machine Learning Framework

Machine Learning Inter-Atomic Potentials Generation Driven by Active Learning: A Case Study for Amorphous and Liquid Hafnium dioxide

ChemSpaceAL: An Efficient Active Learning Methodology Applied to Protein-Specific Molecular Generation