Generative adversarial networks for transition state geometry prediction

This work introduces a novel application of generative adversarial networks (GANs) for the prediction of starting geometries in transition state (TS) searches based on the geometries of reactants and products. The multi-dimensional potential energy space of a chemical reaction often complicates the location of a starting TS geometry, leading to the correct TS combining reactants and products in question. The proposed TS-GAN efficiently maps the space between reactants and products and generates reliable TS guess geometries, and it can be easily combined with any quantum chemical software package performing geometry optimizations. The TS-GAN was trained and applied to generate TS guess structures for typical chemical reactions, such as hydrogen migration, isomerization, and transition metal-catalyzed reactions. The performance of the TS-GAN was directly compared to that of classical approaches, proving its high accuracy and efficiency. The current TS-GAN can be extended to any dataset that contains sufficient chemical reactions for training. The software is freely available for training, experimentation, and prediction at https://github.com/ekraka/TS-GAN.