arXiv:2605.15806cs.LG2026-05

提出新型神经算子,直接学习随机偏微分方程的均值与协方差结构。

Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorization

论文配图:Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorization
图 1 · 摘自论文原文
  • 基于Doob-Meyer分解,将解分解为可预测漂移与零均值鞅部分
  • 在1D和2D任务中,Wasserstein距离降低达120倍,推理速度提升3倍
  • 适合需要高效不确定性量化且对分辨率不变性的场景

神经算子在确定性问题上表现优异,但在处理随机偏微分方程时会退化为条件均值,丢失方差和尾部结构,而传统方法如蒙特卡洛模拟或拼接生成模型又牺牲了单次推断效率与分辨率不变性。为此,本文基于Doob-Meyer定理,将任意半鞅分解为可预测漂移与零均值鞅,构建了马丁格尔神经算子(MNO)。MNO直接从初值映射到终端分布的均值与协方差,由漂移型均值和构造上正半定的低秩因子 $B_ϕ$ 构成。实验采用高斯残差形式,在1D SPDE、粗糙波动率及2D算子任务中,$ϕ^4$ 场论下Wasserstein距离减少120倍,随机伯格斯方程下减少68倍,训练时间相近时比条件扩散基线快约3倍。在2D任务中,零样本分辨率迁移能力接近FNO,湍流模拟表现良好,但对准确定系统如Gray-Scott仍失败。

原文摘要 · Abstract (English)

Neural operators excel as deterministic surrogates, but inevitably collapse to the conditional mean when applied to stochastic PDEs, discarding the variance and tail structure upon which uncertainty quantification depends. Recovering this structure typically requires Monte Carlo rollouts or grafted generative models, both of which surrender the one-shot efficiency and resolution invariance that define the operator paradigm. To resolve this, we draw on the Doob-Meyer theorem, which establishes that any semimartingale fundamentally decomposes into a predictable drift and an unpredictable, zero-mean martingale. Translating this theorem into an architectural prior, we introduce the Martingale Neural Operator (MNO). MNO maps an initial condition directly to the conditional mean and covariance of the terminal law, parameterized by a drift-like mean and a low-rank factor $B_ϕ$ with $B_ϕ^\top B_ϕ$ positive semi-definite by construction. For our experiments, we use a Gaussian residual instantiation. Across 1D SPDEs, rough volatility, and 2D operator tasks, MNO reduces Wasserstein distance by up to $120\times$ on $ϕ^4$ field theory and $68\times$ on stochastic Burgers, evaluating $\sim 3\times$ faster than a conditional diffusion baseline at matched wall-clock training budgets. On 2D tasks, MNO is comparable to FNO on zero-shot resolution transfer and turbulent flow, while quasi-deterministic systems such as Gray-Scott remain a failure mode.

神经算子随机方程不确定性量化概率建模

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