提出新型神经算子,显著提升长时程偏微分方程预测稳定性。
SGNO: Spectral Generator Neural Operators for Stable Long Horizon PDE Rollouts
- 用谱生成器结构设计每步演化,控制频谱增益并修正相位误差。
- 在10个任务上平均降低74.8%的长期预测误差,部分任务降幅超90%。
- 特别适合色散、输运主导及非线性耦合问题,适合做物理模拟研究者。
自回归神经偏微分方程代理模型通过重复应用学习到的单步算子预测未来状态,但小的单步误差在长时间滚动中会累积,导致频谱幅值失真、相位错位和非线性模态耦合误差,尤其在具有明显傅里叶结构的时间依赖型偏微分方程中表现突出。本文提出谱生成神经算子(SGNO),一种面向长时程偏微分方程预报的结构化自回归神经算子。该方法将每个学习到的单步映射组织为结构化的谱演化更新:由实值非正对角生成器提供可控增益的谱骨架,结合复值谱混合作用的可学习校正路径完成残差演化。这种设计既保留了演化特性,又具备描述耗散、色散、输运主导和非线性偏微分方程的灵活性。SGNO适用于周期性线性和半线性演化型偏微分方程,涵盖傅里叶乘子线性动力学。在十个机制匹配的APEBench任务中,其长期滚动预测精度持续优于强基线,相较于最强非SGNO基线,平均降低GMean100达74.8%,各任务降幅介于13.6%至92.9%之间。性能优势在色散与输运主导任务,以及涉及非线性闭包与模态耦合的任务中尤为显著。谱诊断显示能量误差更低,滚动阶段相位保真度更高。消融实验表明,约束生成器、结构化更新和可学习校正路径均对性能有贡献。代码已公开于https://github.com/cruiseresearchgroup/SGNO。
原文摘要 · Abstract (English)
Autoregressive neural PDE surrogates predict future states by repeatedly applying a learned one-step operator. This is a simple and widely used method, but small one-step errors can accumulate during long rollouts. The resulting drift often appears as spectral amplitude distortion, phase misalignment, and nonlinear mode-interaction error. These effects are especially important for time-dependent PDEs with clear Fourier structure. We introduce the Spectral Generator Neural Operator (SGNO), a structured autoregressive neural operator for long-horizon PDE forecasting. SGNO organizes each learned one-step map as a structured spectral evolution update. A real-valued nonpositive diagonal generator provides a gain-controlled spectral backbone, while a learned correction pathway with complex-valued spectral mixing completes the residual evolution. This design gives the autoregressive step an evolution-like structure while retaining the flexibility needed for dissipative, dispersive, transport-dominated, and nonlinear PDEs. SGNO is designed for periodic linear and semilinear evolution PDEs with Fourier multiplier linear dynamics. Across ten mechanism-matched APEBench tasks spanning this regime, SGNO consistently outperforms strong single-step autoregressive baselines in long-horizon rollout accuracy, reducing GMean100 by a median of 74.8% relative to the strongest available non-SGNO baseline, with per-task reductions ranging from 13.6% to 92.9%. The gains are strongest on dispersive and transport-dominated tasks, as well as tasks involving nonlinear closure and mode coupling. Spectral diagnostics show lower spectral energy error and improved rollout-level phase fidelity. Ablations show that the constrained generator, the structured update, and the learned correction pathway each contribute to performance. The code is available at https://github.com/cruiseresearchgroup/SGNO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。