用轻量结构高效学习时变偏微分方程的解算子。
Merging Memory and Space: A State Space Neural Operator
- 将状态空间模型扩展至时空联合建模,引入自适应阻尼与可学习频段调制。
- 在多个经典方程上达顶尖性能,参数量显著少于现有方法。
- 适合需要高效高精度求解偏微分方程的研究者和工程应用。
我们提出状态空间神经算子(SS-NO),一种用于学习时变偏微分方程(PDE)解算子的紧凑架构。该方法将结构化状态空间模型(SSMs)扩展至联合时空建模,引入两个关键机制:自适应阻尼通过局部化感受野稳定训练,可学习频率调制实现数据驱动的频谱选择。这些组件在保持参数效率的同时捕捉长程依赖。理论上,我们建立了SSMs与神经算子之间的联系,证明了全视野卷积架构的通用性。实验上,SS-NO在多种PDE基准测试中表现优异,包括1D Burgers方程、Kuramoto-Sivashinsky方程,以及2D Navier-Stokes和可压缩欧拉流,且参数量远低于对比方法。其因子分解变体在挑战性2D问题上仍具可扩展性能。结果表明,阻尼与频段学习对算子建模至关重要,而轻量因子化为大规模高效PDE学习提供新路径。
原文摘要 · Abstract (English)
We propose the *State Space Neural Operator* (SS-NO), a compact architecture for learning solution operators of time-dependent partial differential equations (PDEs). Our formulation extends structured state space models (SSMs) to joint spatiotemporal modeling, introducing two key mechanisms: *adaptive damping*, which stabilizes learning by localizing receptive fields, and *learnable frequency modulation*, which enables data-driven spectral selection. These components provide a unified framework for capturing long-range dependencies with parameter efficiency. Theoretically, we establish connections between SSMs and neural operators, proving a universality theorem for convolutional architectures with full field-of-view. Empirically, SS-NO achieves state-of-the-art performance across diverse PDE benchmarks-including 1D Burgers' and Kuramoto-Sivashinsky equations, and 2D Navier-Stokes and compressible Euler flows-while using significantly fewer parameters than competing approaches. A factorized variant of SS-NO further demonstrates scalable performance on challenging 2D problems. Our results highlight the effectiveness of damping and frequency learning in operator modeling, while showing that lightweight factorization provides a complementary path toward efficient large-scale PDE learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。