将小波分解与拉普拉斯算子结合,提升求解偏微分方程的多尺度建模能力。
WLNO: Wavelet-Laplace Neural Operator for Solving Partial Differential Equations

- 融合哈尔小波多尺度分解与拉普拉斯域极点残差结构
- 在五个基准问题上均优于原版LNO,尤其在激波和涡旋结构问题中提升显著
- 适合需要捕捉复杂空间多尺度特征的物理模拟任务
本文提出一种新型神经算子Wavelet-Laplace Neural Operator(WLNO),通过将哈尔小波多尺度空间分解与拉普拉斯域极点残差形式的拉普拉斯神经算子(LNO)结合,解决传统LNO缺乏显式空间局部多尺度特征提取机制的问题。WLNO在LNO核心基础上增加并行单层哈尔离散小波变换(DWT)分支,将升维特征图分解为四个频带:近似(LL)、水平细节(LH)、垂直细节(HL)和对角细节(HH),并对每个子带分别应用可学习的$1\times1$卷积,再通过逆DWT重构。两分支通过可学习的sigmoid门控权重$α_\mathrm{wav}$融合,初始值较小以保证训练初期主干主导,实现动态平衡拉普拉斯域动力学与空间多尺度特征。在五个基准偏微分方程问题上——扩散方程、伯格斯方程、反应-扩散系统、达西流及二维纳维-斯托克斯方程——使用相同超参数、训练数据与评估协议进行对比,结果表明WLNO在所有问题上均优于LNO,尤其在具有强空间多尺度结构的问题(如含尖锐激波的伯格斯方程和含相干涡旋的纳维-斯托克斯方程)中表现突出,而在平滑与椭圆型问题中保持稳定。结果验证了基于小波的多尺度空间分解是拉普拉斯域算子学习的有效补充。
原文摘要 · Abstract (English)
This work introduces the Wavelet-Laplace Neural Operator (WLNO), a novel neural operator that fuses Haar wavelet multi-scale spatial decomposition with the Laplace-domain pole-residue formulation of the Laplace Neural Operator (LNO). While LNO captures transient and steady-state dynamics through learnable system poles and residues, it lacks an explicit mechanism for extracting spatially localized multi-scale features inherent in complex PDE solutions. WLNO addresses this by augmenting the LNO core with a parallel single-level Haar discrete wavelet transform (DWT) branch that decomposes the lifted feature map into four frequency subbands: approximation (LL), horizontal detail (LH), vertical detail (HL), and diagonal detail (HH) and applies independent learned $1\times1$ convolutions to each subband before reconstruction via the inverse DWT. The two branches are fused through a learnable sigmoid-gated weight $α_\mathrm{wav}$, initialized to give a small initial contribution to the wavelet branch, allowing the model to adaptively balance Laplace-domain dynamics against spatial multi-scale features throughout training. WLNO is evaluated against LNO on five benchmark PDE problems using identical hyperparameters, training data, and evaluation protocols: the diffusion equation, the Burgers equation, the reaction-diffusion system, Darcy flow, and the two-dimensional Navier-Stokes equation. WLNO consistently outperforms LNO on all five problems, with the most pronounced improvement on problems with strong spatial multi-scale structure, such as the Burgers equation with sharp shock fronts and the Navier-Stokes equation with coherent vortical structures, while remaining consistent across smoother and elliptic problems. These results demonstrate that wavelet-based multi-scale spatial decomposition is a principled and effective complement to Laplace-domain operator learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。