Applying machine learning techniques to predict the properties of energetic materials

Abstract

We present a proof of concept that machine learning techniques can be used to predict the properties of CNOHF energetic molecules from their molecular structures. We focus on a small but diverse dataset consisting of 109 molecular structures spread across ten compound classes. Up until now, candidate molecules for energetic materials have been screened using predictions from expensive quantum simulations and thermochemical codes. We present a comprehensive comparison of machine learning models and several molecular featurization methods - sum over bonds, custom descriptors, Coulomb matrices, bag of bonds, and fingerprints. The best featurization was sum over bonds (bond counting), and the best model was kernel ridge regression. Despite having a small data set, we obtain acceptable errors and Pearson correlations for the prediction of detonation pressure, detonation velocity, explosive energy, heat of formation, density, and other properties out of sample. By including another dataset with 309 additional molecules in our training we show how the error can be pushed lower, although the convergence with number of molecules is slow. Our work paves the way for future applications of machine learning in this domain, including automated lead generation and interpreting machine learning models to obtain novel chemical insights.

published in Scientific Reports

References in corpus (5)

• Machine Learning Unifies the Modelling of Materials and Molecules
• Machine learning prediction errors better than DFT accuracy
• Understanding molecular representations in machine learning: The role of uniqueness and target similarity
• Learning molecular energies using localized graph kernels
• Machine Learning of Energetic Material Properties

Cited by in corpus (9)

• Deep learning for molecular design - a review of the state of the art
• A Deep Dive into Machine Learning Density Functional Theory for Materials Science and Chemistry
• Advances of Machine Learning in Materials Science: Ideas and Techniques
• Electronic structure calculations of twisted multi-layer graphene superlattices
• Artificial intelligence approaches for materials-by-design of energetic materials: state-of-the-art, challenges, and future directions
• High Pressure and Temperature Neural Network Reactive Force Field for Energetic Materials
• Advancing Molecular Machine Learning Representations with Stereoelectronics-Infused Molecular Graphs
• Unsupervised learning of representative local atomic arrangements in molecular dynamics data
• Applying Deutsch's concept of good explanations to artificial intelligence and neuroscience -- an initial exploration