Crystal structure identification with 3D convolutional neural networks with application to high-pressure phase transitions in SiO2

Efficient, reliable and easy-to-use structure recognition of atomic environments is essential for the analysis of atomic scale computer simulations. In this work, we train two neuronal network (NN) architectures, namely PointNet and dynamic graph convolutional NN (DG-CNN) using different hyperparameters and training regimes to assess their performance in structure identification tasks of atomistic structure data. We show benchmarks on simple crystal structures, where we can compare against established methods. The approach is subsequently extended to structurally more complex SiO2 phases. By making use of this structure recognition tool, we are able to achieve a deeper understanding of the crystallization process in amorphous SiO2 under shock compression. Lastly, we show how the NN based structure identification workflows can be integrated into OVITO using its python interface.