arXiv:2605.10451cs.LGcs.NA2026-05被引 1

让神经算子自己学基函数,更好捕捉复杂物理变化。

Don't Fix the Basis -- Learn It: Spectral Representation with Adaptive Basis Learning for PDEs

论文配图:Don't Fix the Basis -- Learn It: Spectral Representation with Adaptive Basis Learning for PDEs
图 1 · 摘自论文原文
  • 用学习的密度函数构建自适应谱基,替代固定基
  • 在保持快速傅里叶计算复杂度下提升精度
  • 适合处理梯度陡峭、多尺度的PDE问题

谱神经算子在求解偏微分方程(PDE)中表现优异,但依赖于固定的全局基函数,难以刻画空间异质性和多尺度动态。本文提出自适应基学习(ABLE),通过学习辅助密度函数构建空间自适应的Parseval框架,使算子在提升后的谱空间中操作,同时保证可逆性与$O(N/log N)$复杂度(基于FFT实现)。该方法将表达能力从谱系数转移到表示本身,更高效地捕捉局部结构和非平移不变相互作用。ABLE可无缝嵌入现有神经算子架构,作为谱层的即插即用替换。在多个基准测试中,相比强基线显著提升精度,尤其在具有陡峭梯度和多尺度行为的场景下增益最大。为现有模型(如U-FNO、HPM)引入ABLE进一步提升性能,验证其作为通用互补谱优化手段的有效性。结果表明,数据驱动的表示选择比算子复杂度本身是神经算子设计的关键瓶颈。通过学习基函数,ABLE为改进PDE学习中的谱方法提供了原理清晰且高效的框架。

原文摘要 · Abstract (English)

Spectral neural operators achieve strong performance for PDE learning, but rely on fixed global bases that limit their ability to represent spatially heterogeneous and multiscale dynamics. We propose Adaptive Basis Learning (ABLE), a framework that learns data-dependent spectral representations instead of relying on predefined bases. ABLE constructs a spatially adaptive Parseval frame via a learned ancillary density, enabling the operator to act in a lifted spectral space while preserving invertibility and maintaining $O(N\log N)$ complexity through FFT-based implementation. This shifts the source of expressivity from spectral coefficients to the representation itself, allowing the model to capture localized structures and non-translation-invariant interactions more efficiently. ABLE integrates seamlessly into existing neural operator architectures as a drop-in replacement for spectral layers. Across a range of benchmarks ABLE improves accuracy over strong baselines, with the largest gains in regimes characterized by sharp gradients and multiscale behavior. Moreover, augmenting existing models (e.g., U-FNO, HPM) with ABLE further enhances their performance, demonstrating its role as a general and complementary spectral refinement. Our results highlight that the data-driven choice of representation, rather than operator complexity alone, is a key bottleneck in neural operator design. By learning the basis itself, ABLE provides a principled and efficient framework for improving spectral methods in PDE learning.

PDE求解神经算子自适应基谱方法

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