PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials

Yunqi Shao,Matti Hellström,Pavlin D. Mitev,Lisanne Knijff,Chao Zhang
DOI: https://doi.org/10.1021/acs.jcim.9b00994
2020-01-12
Abstract:Atomic neural networks (ANNs) constitute a class of machine learning methods for predicting potential energy surfaces and physico-chemical properties of molecules and materials. Despite many successes, developing interpretable ANN architectures and implementing existing ones efficiently are still challenging. This calls for reliable, general-purpose and open-source codes. Here, we present a python library named PiNN as a solution toward this goal. In PiNN, we designed a new interpretable and high-performing graph convolutional neural network variant, PiNet, as well as implemented the established Behler-Parrinello high-dimensional neural network. These implementations were tested using datasets of isolated small molecules, crystalline materials, liquid water and an aqueous alkaline electrolyte. PiNN comes with a visualizer called PiNNBoard to extract chemical insight "learned" by ANNs, provides analytical stress tensor calculations and interfaces to both the Atomic Simulation Environment and a development version of the Amsterdam Modeling Suite. Moreover, PiNN is highly modularized which makes it useful not only as a standalone package but also as a chain of tools to develop and to implement novel ANNs. The code is distributed under a permissive BSD license and is freely accessible at <a class="link-external link-https" href="https://github.com/Teoroo-CMC/PiNN/" rel="external noopener nofollow">this https URL</a> with full documentation and tutorials.
Computational Physics,Disordered Systems and Neural Networks,Chemical Physics
What problem does this paper attempt to address?