The Novel Graph Transformer-Based Surrogate Model for Learning Physical Systems

Bo Feng,Xiao-Ping Zhou
DOI: https://doi.org/10.1016/j.cma.2024.117410
IF: 6.588
2024-01-01
Computer Methods in Applied Mechanics and Engineering
Abstract:Predicting physical systems over long-term horizons has a significant challenge. Although prevalent machine learning techniques, such as Physics-Informed Neural Networks (PINNs), can achieve high accuracy, they primarily focus on single-step solutions and entail high computational costs. To address this gap, we propose a novel surrogate model leveraging graph representation for mesh-based physical systems. The proposed model incorporates Graph Transformer with attention mechanisms to aggregate information from neighboring nodes efficiently. Furthermore, we employ a multi-step prediction strategy in the loss function formulation to ensure robust long-term prediction capabilities. Additionally, in order to predict stress in solid mechanics, innovative symlog and symexp functions are proposed to enhance the robustness and reliability of the proposed model. Numerical results indicate that the proposed model can achieve high-accuracy and efficient long-term evolution prediction of physical systems.
What problem does this paper attempt to address?