让神经网络模拟波动时保持能量和质量守恒,更稳定准确。
Structure-Preserving Physics-Informed Neural Network for the Korteweg--de Vries (KdV) Equation
- 用正弦激活函数+物理守恒项改进损失函数,增强对波的捕捉能力。
- 长期模拟中能量不漂移,单波、双波、脉冲等典型行为均正确重现。
- 无需分阶段训练,收敛快且稳定性高,适合研究哈密顿型方程。
物理信息神经网络(PINNs)为求解非线性偏微分方程提供了灵活框架,但传统方法在长时间积分中常无法保持关键物理守恒量。本文提出一种结构保持型PINN框架,用于非线性Korteweg--de Vries(KdV)方程——描述非线性与色散波传播的典型模型。该方法将质量与哈密顿能量守恒直接嵌入损失函数,确保训练与预测全程物理一致且能量稳定。不同于传统的tanh激活函数,本方法采用正弦激活函数,提升频谱表达力,精准捕捉KdV孤子的振荡与色散特性。通过单孤子传播(形状不变平移)、双孤子相互作用(弹性碰撞带相位移动)、余弦脉冲初始化(非线性色散破裂)等案例验证,模型成功复现了KdV动力学的核心特征并维持守恒量。消融实验表明,守恒约束优化与正弦特征映射结合可加速收敛、提升长期稳定性、抑制漂移,且无需多阶段预训练。结果表明,计算高效、守恒感知的正则化搭配正弦表示,能生成针对哈密顿型偏微分方程(如KdV)的鲁棒、能量一致的PINN。
原文摘要 · Abstract (English)
Physics-Informed Neural Networks (PINNs) offer a flexible framework for solving nonlinear partial differential equations (PDEs), yet conventional implementations often fail to preserve key physical invariants during long-term integration. This paper introduces a \emph{structure-preserving PINN} framework for the nonlinear Korteweg--de Vries (KdV) equation, a prototypical model for nonlinear and dispersive wave propagation. The proposed method embeds the conservation of mass and Hamiltonian energy directly into the loss function, ensuring physically consistent and energy-stable evolution throughout training and prediction. Unlike standard \texttt{tanh}-based PINNs~\cite{raissi2019pinn,wang2022modifiedpinn}, our approach employs sinusoidal activation functions that enhance spectral expressiveness and accurately capture the oscillatory and dispersive nature of KdV solitons. Through representative case studies -- including single-soliton propagation (shape-preserving translation), two-soliton interaction (elastic collision with phase shift), and cosine-pulse initialization (nonlinear dispersive breakup) -- the model successfully reproduces hallmark behaviors of KdV dynamics while maintaining conserved invariants. Ablation studies demonstrate that combining invariant-constrained optimization with sinusoidal feature mappings accelerates convergence, improves long-term stability, and mitigates drift without multi-stage pretraining. These results highlight that computationally efficient, invariant-aware regularization coupled with sinusoidal representations yields robust, energy-consistent PINNs for Hamiltonian partial differential equations such as the KdV equation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。