arXiv:2607.29135cs.LGcs.AI2026-07

通过历史对比提升长时序神经算子的预测稳定性

HERO: History-Enriched Rollout Training for Long-Horizon Autoregressive Neural Operators

论文配图:HERO: History-Enriched Rollout Training for Long-Horizon Autoregressive Neural Operators
图 1 · 摘自论文原文
  • 引入模型优化历史构建相对监督,动态对比候选轨迹
  • 在9个PDE基准上实现更长稳定滚动、更高精度与更强泛化能力
  • 适合需要高可靠性长时序模拟的研究者与工程应用

神经算子通过递归应用学习到的演化算子来快速替代时变偏微分方程(PDE),但自回归滚动会将每一步预测误差回传作为输入,导致局部误差累积。现有滚动训练策略虽缓解了训练输入与自生成状态之间的不匹配,但其监督仍仅衡量与真实轨迹的绝对偏差,无法反映模型是否克服了早期优化中出现的长时序失败行为。本文提出历史增强滚动训练(HERO),在传统绝对轨迹监督基础上,加入基于模型优化历史的相对监督。HERO通过周期性刷新的滞后算子、当前模型和扰动输入,对分离的候选滚动轨迹进行排序,依据滚动误差、谱差异、能量漂移和误差增长等指标,选取最强失败轨迹作为固定参考基线。该参考进入基于边距的目标函数,诱导有界且样本依赖的真值滚动梯度重加权,而非独立梯度方向,我们进一步进行了理论分析。在九个具有谱和注意力骨干网络的PDE基准上的实验表明,HERO在不增加推理成本的前提下,持续提升了长时序精度、稳定滚动长度和分布外鲁棒性。结果表明,基于历史的相对监督能有效稳定长时序自回归预测。

原文摘要 · Abstract (English)

Neural operators provide fast surrogates for time-dependent partial differential equations (PDEs) by applying a learned evolution operator recursively to its own predictions, but this autoregressive rollout feeds every prediction error back as input, so local errors accumulate. Existing rollout-training strategies reduce the mismatch between training inputs and self-generated states, yet their supervision still measures only the absolute discrepancy from the ground-truth trajectory. Such supervision is therefore uninformative about whether the operator has overcome the long-horizon failure behaviors it exhibited earlier during optimization. We propose history-enriched rollout training (HERO), which augments conventional absolute trajectory supervision with relative supervision derived from the model's optimization history. HERO ranks detached candidate rollouts from a periodically refreshed lagged operator, the current model, and a perturbed input by rollout error, spectral discrepancy, energy drift, and error growth, and selects the strongest failure trajectory as reference. This reference enters a margin-based objective as a fixed comparison baseline, inducing a bounded, sample-dependent reweighting of the ground-truth rollout gradient rather than an independent gradient direction, which we further analyze theoretically. Experiments on nine PDE benchmarks with spectral and attention-based backbones show that HERO consistently improves long-horizon accuracy, stable rollout length, and out-of-distribution robustness at no inference-time cost. These results indicate that history-enriched relative supervision is effective for stabilizing long-horizon autoregressive prediction.

神经算子长时序预测自回归PDE求解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。