用物理约束生成对抗网络,从演化后状态反推初始条件。
Backward Reconstruction of the Chafee--Infante Equation via Physics-Informed WGAN-GP
- 结合物理规律与生成对抗网络,设计带梯度惩罚的U-Net架构。
- 测试集平均绝对误差仅0.24,可精准恢复界面结构。
- 适合需要高鲁棒性逆问题求解的研究者参考。
本文提出一种物理信息引导的Wasserstein GAN带梯度惩罚(WGAN-GP),用于求解二维区域上具有狄利克雷边界条件的逆Chafee--Infante问题。目标是从经过100次显式欧拉迭代后的近平衡态中重构未知初始条件。由于该映射强烈抑制高频成分,逆问题严重不适定且对噪声敏感。方法融合了U-Net生成器、带谱归一化的PatchGAN判别器、Wasserstein损失与梯度惩罚,以及若干物理信息辅助项,包括李雅普诺夫能量匹配、分布统计和关键的前向模拟惩罚项。该惩罚确保预测初始条件与其在相同前向欧拉离散格式下的演化一致。先前使用Eyre型半隐式求解器的实验因批量GPU训练中牛顿迭代的成本与不稳定性,无法兼容此残差机制。在5万训练对与1万测试对的$128\times128$网格数据集上(自然幅度范围为$[-1,1]$),最佳模型在全测试集上的平均绝对误差(MAE)约为0.23988159,样本间标准差约0.00266345。结果表明该方法具备稳定逆推能力,能准确恢复界面结构,并对初始数据中的高频噪声具有强鲁棒性。
原文摘要 · Abstract (English)
We present a physics-informed Wasserstein GAN with gradient penalty (WGAN-GP) for solving the inverse Chafee--Infante problem on two-dimensional domains with Dirichlet boundary conditions. The objective is to reconstruct an unknown initial condition from a near-equilibrium state obtained after 100 explicit forward Euler iterations of the reaction-diffusion equation \[ u_t - γΔu + κ\left(u^3 - u\right)=0. \] Because this mapping strongly damps high-frequency content, the inverse problem is severely ill-posed and sensitive to noise. Our approach integrates a U-Net generator, a PatchGAN critic with spectral normalization, Wasserstein loss with gradient penalty, and several physics-informed auxiliary terms, including Lyapunov energy matching, distributional statistics, and a crucial forward-simulation penalty. This penalty enforces consistency between the predicted initial condition and its forward evolution under the \emph{same} forward Euler discretization used for dataset generation. Earlier experiments employing an Eyre-type semi-implicit solver were not compatible with this residual mechanism due to the cost and instability of Newton iterations within batched GPU training. On a dataset of 50k training and 10k testing pairs on $128\times128$ grids (with natural $[-1,1]$ amplitude scaling), the best trained model attains a mean absolute error (MAE) of approximately \textbf{0.23988159} on the full test set, with a sample-wise standard deviation of about \textbf{0.00266345}. The results demonstrate stable inversion, accurate recovery of interfacial structure, and robustness to high-frequency noise in the initial data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。