用深度学习预测等离子体湍流演化并反推物理参数,精度高且计算快。
Convolution Operator Network for Forward and Inverse Problems (FI-Conv): Application to Plasma Turbulence Simulations
- 基于ConvNeXt V2改进U-Net,兼顾高频变化处理与低算力需求。
- 短时预测(t~3)误差小,长时统计特性(t~100)捕捉准确。
- 无需重训练即可反推方程参数,适合复杂动力系统研究者使用。
我们提出卷积算子网络FI-Conv,用于预测复杂时空动力系统的演化并估计其物理参数,如湍流。该框架基于U-Net结构,将多数卷积层替换为ConvNeXt V2块,在保持对高频输入的性能同时降低计算复杂度。输入包括初始状态、偏微分方程(PDE)参数和演化时间,输出为系统未来状态。以描述二维静电漂移波湍流的Hasegawa-Wakatani(HW)方程为例,采用自回归预报流程,FI-Conv在短时间尺度(t ~ 3)实现精准前向预测,并在长时间尺度(t ~ 100)有效捕捉关键物理量的统计特性。此外,我们设计了一种基于梯度下降的逆向估计方法,可从演化数据中准确推断PDE参数,无需修改已训练模型权重。结果表明,FI-Conv是复杂时空动力系统中物理信息机器学习的有效替代方案。
原文摘要 · Abstract (English)
We propose the Convolutional Operator Network for Forward and Inverse Problems (FI-Conv), a framework capable of predicting system evolution and estimating parameters in complex spatio-temporal dynamics, such as turbulence. FI-Conv is built on a U-Net architecture, in which most convolutional layers are replaced by ConvNeXt V2 blocks. This design preserves U-Net performance on inputs with high-frequency variations while maintaining low computational complexity. FI-Conv uses an initial state, PDE parameters, and evolution time as input to predict the system future state. As a representative example of a system exhibiting complex dynamics, we evaluate the performance of FI-Conv on the task of predicting turbulent plasma fields governed by the Hasegawa-Wakatani (HW) equations. The HW system models two-dimensional electrostatic drift-wave turbulence and exhibits strongly nonlinear behavior, making accurate approximation and long-term prediction particularly challenging. Using an autoregressive forecasting procedure, FI-Conv achieves accurate forward prediction of the plasma state evolution over short times (t ~ 3) and captures the statistic properties of derived physical quantities of interest over longer times (t ~ 100). Moreover, we develop a gradient-descent-based inverse estimation method that accurately infers PDE parameters from plasma state evolution data, without modifying the trained model weights. Collectively, our results demonstrate that FI-Conv can be an effective alternative to existing physics-informed machine learning methods for systems with complex spatio-temporal dynamics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。