Improved description of atomic environments using low-cost polynomial functions with compact support

Bircher, Martin P and Singraber, Andreas and Dellago, Christoph (2021) Improved description of atomic environments using low-cost polynomial functions with compact support. Machine Learning: Science and Technology, 2 (3). 035026. ISSN 2632-2153

[thumbnail of Bircher_2021_Mach._Learn.__Sci._Technol._2_035026.pdf] Text
Bircher_2021_Mach._Learn.__Sci._Technol._2_035026.pdf - Published Version

Download (1MB)

Abstract

The prediction of chemical properties using machine learning techniques calls for a set of appropriate descriptors that accurately describe atomic and, on a larger scale, molecular environments. A mapping of conformational information on a space spanned by atom-centred symmetry functions (SF) has become a standard technique for energy and force predictions using high-dimensional neural network potentials (HDNNP). An appropriate choice of SFs is particularly crucial for accurate force predictions. Established atom-centred SFs, however, are limited in their flexibility, since their functional form restricts the angular domain that can be sampled without introducing problematic derivative discontinuities. Here, we introduce a class of atom-centred SFs based on polynomials with compact support called polynomial symmetry functions (PSF), which enable a free choice of both, the angular and the radial domain covered. We demonstrate that the accuracy of PSFs is either on par or considerably better than that of conventional, atom-centred SFs. In particular, a generic set of PSFs with an intuitive choice of the angular domain inspired by organic chemistry considerably improves prediction accuracy for organic molecules in the gaseous and liquid phase, with reductions in force prediction errors over a test set approaching 50% for certain systems. Contrary to established atom-centred SFs, computation of PSF does not involve any exponentials, and their intrinsic compact support supersedes use of separate cutoff functions, facilitating the choice of their free parameters. Most importantly, the number of floating point operations required to compute polynomial SFs introduced here is considerably lower than that of other state-of-the-art SFs, enabling their efficient implementation without the need of highly optimised code structures or caching, with speedups with respect to other state-of-the-art SFs reaching a factor of 4.5 to 5. This low-effort performance benefit substantially simplifies their use in new programs and emerging platforms such as graphical processing units. Overall, polynomial SFs with compact support improve accuracy of both, energy and force predictions with HDNNPs while enabling significant speedups compared to their well-established counterparts.

Item Type: Article
Subjects: STM Digital Press > Multidisciplinary
Depositing User: Unnamed user with email support@stmdigipress.com
Date Deposited: 04 Sep 2024 04:15
Last Modified: 04 Sep 2024 04:15
URI: http://publications.articalerewriter.com/id/eprint/1275

Actions (login required)

View Item
View Item