Unsupervised Learning Method for the Wave Equation Based on Finite Difference Residual Constraints Loss

Xin Feng,Yi Jiang,Jia-Xian Qin,Lai-Ping Zhang,Xiao-Gang Deng
2024-01-23
Abstract:The wave equation is an important physical partial differential equation, and in recent years, deep learning has shown promise in accelerating or replacing traditional numerical methods for solving it. However, existing deep learning methods suffer from high data acquisition costs, low training efficiency, and insufficient generalization capability for boundary conditions. To address these issues, this paper proposes an unsupervised learning method for the wave equation based on finite difference residual constraints. We construct a novel finite difference residual constraint based on structured grids and finite difference methods, as well as an unsupervised training strategy, enabling convolutional neural networks to train without data and predict the forward propagation process of waves. Experimental results show that finite difference residual constraints have advantages over physics-informed neural networks (PINNs) type physical information constraints, such as easier fitting, lower computational costs, and stronger source term generalization capability, making our method more efficient in training and potent in application.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The problems that this paper attempts to solve are several key issues existing in the existing deep - learning methods when solving the wave equation: 1. **High cost of data set acquisition**: Traditional data - driven deep - learning methods require a large amount of training data, and this data is usually difficult to obtain and costly. 2. **Low training efficiency**: The existing deep - learning methods have problems of high computational complexity and long training time during the training process. 3. **Insufficient generalization ability of boundary conditions**: When the boundary conditions change, the model needs to be retrained, resulting in a limited application range. To solve these problems, the author proposes an unsupervised learning method for the wave equation based on Finite Difference Residual Constraints (FDRC). Specifically: - **Finite Difference Residual Constraints**: By combining the traditional finite - difference method with physics - informed learning, a novel finite - difference residual constraint is constructed, enabling the convolutional neural network to be trained without data and predict the forward process of waves. - **Unsupervised training strategy**: An unsupervised training strategy based on the training pool is proposed, which improves the training efficiency of the model and solves the problem of difficult data acquisition at the same time. - **Improved generalization ability of boundary conditions**: Compared with Physics - Informed Neural Networks (PINNs), the FDRC method has better adaptability to changes in boundary conditions and does not need to retrain the model. The experimental results show that this method is superior to the existing PINNs - like methods in terms of fitting speed, computational cost, and source - term generalization ability, and has higher training efficiency and application potential.