Hostname: page-component-7b7567c8c-jswmv Total loading time: 0 Render date: 2026-11-15T09:37:36.945Z Has data issue: false hasContentIssue false

Filtered partial differential equations: a robust surrogate constraint in physics-informed deep learning framework

Published online by Cambridge University Press:  15 November 2024

Dashan Zhang
Affiliation:
Department of Mechanics and Aerospace Engineering, Southern University of Science and Technology, Shenzhen 518055, PR China
Yuntian Chen*
Affiliation:
Ningbo Institute of Digital Twin, Eastern Institute of Technology, Ningbo 315200, PR ChinaZhejiang Key Laboratory of Industrial Intelligence and Digital Twin, Eastern Institute of Technology, Ningbo 315200, PR China
Shiyi Chen*
Affiliation:
Department of Mechanics and Aerospace Engineering, Southern University of Science and Technology, Shenzhen 518055, PR ChinaNingbo Institute of Digital Twin, Eastern Institute of Technology, Ningbo 315200, PR China
*
†Email addresses for correspondence: cyt_cn@126.com, chensy@sustech.edu.cn
†Email addresses for correspondence: cyt_cn@126.com, chensy@sustech.edu.cn

Abstract

Embedding physical knowledge into neural network (NN) training has been a hot topic. However, when facing the complex real world, most of the existing methods still strongly rely on the quantity and quality of observation data. Furthermore, the NNs often struggle to converge when the solution to the real equation is very complex. Inspired by large eddy simulation in computational fluid dynamics, we propose an improved method based on filtering. We analysed the causes of the difficulties in physics-informed machine learning, and proposed a surrogate constraint (filtered partial differential equation, FPDE) of the original physical equations to reduce the influence of noisy and sparse observation data. In the noise and sparsity experiment, the proposed FPDE models (which are optimized by FPDE constraints) have better robustness than the conventional PDE models. Experiments demonstrate that the FPDE model can obtain the same quality solution with 100 % higher noise and 12 % quantity of observation data of the baseline. Besides, two groups of real measurement data are used to show the FPDE improvements in real cases. The final results show that the FPDE still gives more physically reasonable solutions when facing the incomplete equation problem and the extremely sparse and high-noise conditions. The proposed FPDE constraint is helpful for merging real-world experimental data into physics-informed training, and it works effectively in two real-world experiments: simulating cell movement in scratches and blood velocity in vessels.

Information

Type
JFM Papers
Copyright
© The Author(s), 2024. Published by Cambridge University Press
Figure 0

Figure 1.The fluid flow through a cylinder wall (streak lines). To keep the consistency of data, the reference simulation solutions are from the open resource data of hidden fluid mechanics (Raissi, Yazdani & Karniadakis 2020).

Figure 1

Figure 2.Summary of the physics-informed training tasks. (a) The extrapolation, where known and unknown parts conform to the different data distributions. The NN is trained to learn the hidden pattern in the known part and extrapolate it in the unknown part. (b) The restoration, of which the given and unknown parts conform to the same data distribution. Meanwhile, the data has poor quality and quantity. Panel (b i) shows the restoration from sparse data and (b ii) is the restoration from noisy data. (c) Sufficient and accurate data with the same distribution, which makes the task an ordinary interpolation problem.

Figure 2

Figure 3.The NN output guided by different equations. (a) The Burgers equation is the classic example in PINN (Raissi et al.2019), which is used to show the effect of regularity. (b) The exponential function, which shows the difficulty of the NN to handle the interval with a small loss value. The NN always focuses on the large loss interval. (c) The N–S equation. Facing the complex solution, the NN tends to be trapped in the local minimum and obtain the trivial solution.

Figure 3

Figure 4.The classic filters in physical and spectral space. The most common Gaussian weight filter kernel will be used in subsequent experiments.

Figure 4

Figure 5.The physics-informed training framework and the proposed FPDE method. The example of solving the N–S equation governed problem. (a) The fully connected NN. A NN with multiple residual connections is used to model the observation. (b) The conventional PDE loss calculation process, which uses NN outputs in differentiation directly. (c) The improved framework in the PDE loss calculation iteration. The orange part is used to calculate the cross-terms in the equations and the green part is used to filter all variables. The PDEs are calculated with filtered variables instead. (d) The auto differentiation part. Derivatives are automatically generated from the computational graph.

Figure 5

Table 1.The FPDE algorithm and comparison with the conventional algorithm.

Figure 6

Figure 6.Given possible mechanism of the FPDE method for processing noisy data. (a) Though the noisy data does not directly participate in the PDE loss calculation, it still affects the PDE loss by changing the NN output from the previous iteration. (b) The filtered NN outputs reduce the noise, making PDE calculations more robust and accurate.

Figure 7

Figure 7.The pre-possessing in full simulation data. In the experiments the training data are randomly sampled in the full dataset. In the sparse group the sampling ratio is the adjustable variable. In the noisy group the noisy level is the adjustable variable. Len is the total size of the dataset, and the sampling ratio indicates the proportion of training data to the dataset.

Figure 8

Figure 8.The numerical simulation data of the cylinder flow problem. The training data are sampled from the original data and contaminated with noise. Similar to the BC, the black circle is the cylinder wall.

Figure 9

Table 2.Summary of two groups of data. These groups of data are used to train the FPDE and conventional PDE models and validate their flow restoration abilities under sparse and noisy conditions.

Figure 10

Figure 9.The experiment of cell migration and data collection. (a) The square petri and counting method of blue lines. (b) Cells fill in the scratch by migrating. (c) Measurement data distribution.

Figure 11

Figure 10.The experiment of one-dimensional blood flow in the $Y$-shaped artery. (a) The theoretical model (including the definitions of $A$, $A_i$ and $interface$). Equation (3.7) used in the training are derived from this model. (b) The location of five measurement points and the schematic diagram of the artery. (c) Overview of measurement data (area and velocity). (d) Multi-head NN to predict the velocity in different vessels.

Figure 12

Figure 11.The NN outputs and MSE of the FPDE and baseline models. (a) The MSE and the residual between the ground truth and output. This result shows the FPDE method works better with sparse observation. (b) A typical example with a $2^{-16}$ sampling ratio. The comparisons of NN outputs ($u,v,p$) are below.

Figure 13

Figure 12.The NN outputs and MSE of the FPDE and baseline models under the noisy data condition. (a) The MSE and residual plots. (b) The FPDE and PDE outputs under the additive white Gaussian noise with 150 % standard deviation of observation data; the comparison of NN outputs is below.

Figure 14

Figure 13.Comparison in the cell migration test data ($\textrm {initial cell number} = [16\,000, 18\,000]$). Inside the red boxes, the difference between two outputs shows that the FPDE can overcome the influence of outliers better. This characteristic shows that the FPDE method has better modelling ability when faced with real, high-noise measurement data.

Figure 15

Figure 14.Comparison in the arterial flow test data. The comparison of velocity and area at the test point is used to show the modelling ability of the FPDE and baseline models. Compared with the velocity field in the baseline model, the FPDE model learns better in large-scale fluctuations.

Figure 16

Figure 15.Decoupling achieved by the FPDE method (reflected in training loss). (a) The PDE and data losses of two models until convergence. (b) The S and ‘exclusive OR’ colour bars. In order to show the distribution of the derivative sign, the colour bar is used in the after-pooling data of the original S. In the S plot, closer to red means more 0–1 oscillating and no effective gradient descent. In the $S_1 \oplus S_2$ plot, the blue colour indicates no conflict training process. Therefore, the effective descent in the baseline model only happened in the initial part, while the FPDE model realized decoupling throughout the whole training. (c) The definition and statistic result for each group of the conflict rate and conflict cost, which indicates higher conflict tolerance of the FPDE than the baseline.

Figure 17

Figure 16.The MSE of the FPDE models with different filters (including ‘no filter’). In this experiment the chosen classic filters are the box filter, the Gaussian filter and the low-pass filter. The conventional no-filter model is trained as the control group.

Figure 18

Figure 17.The MSE of the FPDE models with different size Gaussian filters (including ‘no filter’). In this figure, $\Delta x$ is used to stand for the original filter of size 0.1.

Figure 19

Figure 18.The training data in cell migration modelling. The dataset's initial cell numbers are 14 000 and 20 000 for the NN training. Depending on the initial cell number, different groups fill the scratch at different speeds.