arXiv:2504.20249cs.LG2025-04被引 12

提出TNO模型,高效建模随时间演化的物理方程,支持长时间预测和多输入场景。

Temporal Neural Operator for Modeling Time-Dependent Physical Phenomena

  • 在DeepONet基础上增加时间分支,融合多种训练策略提升时序建模能力。
  • 在多个测试问题上实现长时程外推,误差积累少且对分辨率不敏感。
  • 适合需要高精度长期模拟的科学计算与工程仿真场景。

神经算子(NOs)是用于求解偏微分方程(PDEs)的机器学习模型,能够学习函数空间间的映射。尽管如DeepONet和FNO等模型在空间函数映射中表现出色,但在处理随时间变化的PDE的动态行为时仍存在局限,尤其对训练中未显式出现的时间步难以准确预测,导致时间精度下降。此外,多数神经算子训练成本高昂,尤其在高维情况下。本文提出时空神经算子(TNO),专为时间依赖型PDE的时空算子学习设计。TNO通过在DeepONet框架中引入时间分支,结合多个最优架构选择及训练策略——包括马尔可夫假设、教师强制、时间分组以及对当前或历史状态条件输出的灵活性,显著提升了建模能力。在多样化的基准问题上进行广泛测试与消融实验,验证了TNO在长时程外推、抗误差累积、分辨率不变性以及处理多重输入函数方面的优越性能。

原文摘要 · Abstract (English)

Neural Operators (NOs) are machine learning models designed to solve partial differential equations (PDEs) by learning to map between function spaces. Neural Operators such as the Deep Operator Network (DeepONet) and the Fourier Neural Operator (FNO) have demonstrated excellent generalization properties when mapping between spatial function spaces. However, they struggle in mapping the temporal dynamics of time-dependent PDEs, especially for time steps not explicitly seen during training. This limits their temporal accuracy as they do not leverage these dynamics in the training process. In addition, most NOs tend to be prohibitively costly to train, especially for higher-dimensional PDEs. In this paper, we propose the Temporal Neural Operator (TNO), an efficient neural operator specifically designed for spatio-temporal operator learning for time-dependent PDEs. TNO achieves this by introducing a temporal-branch to the DeepONet framework, leveraging the best architectural design choices from several other NOs, and a combination of training strategies including Markov assumption, teacher forcing, temporal bundling, and the flexibility to condition the output on the current state or past states. Through extensive benchmarking and an ablation study on a diverse set of example problems we demonstrate the TNO long range temporal extrapolation capabilities, robustness to error accumulation, resolution invariance, and flexibility to handle multiple input functions.

神经算子时间序列PDE求解深度学习

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