LoGAN: Local generative adversarial network for novel structure prediction

The efficient generation and filtering of candidate structures for new materials is becoming increasingly important as starting points for computational studies. In this work, we introduce an approach to Wasserstein generative adversarial networks for predicting unique crystal and molecular structures. Leveraging translation- and rotation-invariant atom-centered local descriptors address some of the major challenges faced by similar methods. Our models require only small sets of known structures as training data. Furthermore, the approach is able to generate both non-periodic and periodic structures based on local coordination. We showcase the data efficiency and versatility of the LoGAN approach by recovering all stable C5H12O isomers using only 39 C4H10O and C6H14O training examples, as well as all known low-energy SiO2 crystal structures utilizing only 167 training examples of other SiO2 crystal structures. We also introduce a filtration technique to reduce the computational cost of subsequent characterization steps by selecting samples from unique basins on the potential energy surface, which allows to minimize the number of geometry relaxations needed after structure generation. LoGAN thus represents a new, versatile approach to generative modeling of crystal and molecular structures in the low-data regime, and is available open-source.