StringNET: Neural Network based Variational Method for Transition Pathways

Jiayue Han,Shuting Gu,Xiang Zhou
2024-08-13
Abstract:Rare transition events in meta-stable systems under noisy fluctuations are crucial for many non-equilibrium physical and chemical processes. In these processes, the primary contributions to reactive flux are predominantly near the transition pathways that connect two meta-stable states. Efficient computation of these paths is essential in computational chemistry. In this work, we examine the temperature-dependent maximum flux path, the minimum energy path, and the minimum action path at zero temperature. We propose the StringNET method for training these paths using variational formulations and deep learning techniques. Unlike traditional chain-of-state methods, StringNET directly parametrizes the paths through neural network functions, utilizing the arc-length parameter as the main input. The tasks of gradient descent and re-parametrization in the string method are unified into a single framework using loss functions to train deep neural networks. More importantly, the loss function for the maximum flux path is interpreted as a softmax approximation to the numerically challenging minimax problem of the minimum energy path. To compute the minimum energy path efficiently and robustly, we developed a pre-training strategy that includes the maximum flux path loss in the early training stage, significantly accelerating the computation of minimum energy and action paths. We demonstrate the superior performance of this method through various analytical and chemical examples, as well as the two- and four-dimensional Ginzburg-Landau functional energy.
Chemical Physics,Machine Learning
What problem does this paper attempt to address?
The paper is primarily dedicated to addressing the problem of efficiently computing transition paths in metastable systems with noisy fluctuations. These issues are crucial in non-equilibrium physical and chemical processes, especially when studying rare transition events between one metastable state and another. Specifically, the paper focuses on the following points: 1. **Temperature-dependent Maximum Flux Path (MFP)**: This is the path that connects two metastable states and maximizes the probability flux. This method captures the most likely route the system takes during the transition. 2. **Minimum Energy Path (MEP)**: This is the path with the lowest energy, which is particularly important for understanding energy barriers in chemical reactions. 3. **Minimum Action Path (MAP) at zero temperature**: This is a method based on large deviation theory, suitable for general dynamical systems subjected to small noise perturbations. To compute these paths, the authors propose a method called StringNET. This is a new method based on variational formulas and deep learning techniques. Unlike traditional chain-of-states methods, StringNET directly parameterizes the path through neural network functions and uses the arc length parameter as the primary input. This method unifies the tasks of gradient descent and reparameterization within a single framework, using a loss function to train the deep neural network. The paper also proposes a pre-training strategy that includes the loss function of the maximum flux path in the early training stages, significantly accelerating the computation of the minimum energy path and the minimum action path. Additionally, the paper demonstrates the superior performance of the StringNET method in various analytical and chemical examples, including applications to 2D and 4D Ginzburg-Landau functional energies. In summary, the goal of this paper is to develop an efficient and robust computational method for finding critical transition paths in complex systems, which is crucial for understanding the dynamic behavior in physical, chemical, and biological processes.