# Wiener Chaos Expansion based Neural Operator for Singular Stochastic Partial Differential Equations

arXiv: https://arxiv.org/abs/2603.08219 · HTML: https://mlsanity.com/paper/2603.08219/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- Builds on 9 of its 33 references that are in the map.
- First posted on arXiv 9 Mar 2026.

## Builds on (most cited first, 9 of 9)

- [Fourier Neural Operator for Parametric Partial Differential Equations](https://mlsanity.com/paper/2010.08895/) (2020)
- [FiLM: Visual Reasoning with a General Conditioning Layer](https://mlsanity.com/paper/1709.07871/) (2017)
- [Neural Stochastic PDEs: Resolution-Invariant Learning of Continuous Spatiotemporal Dynamics](https://mlsanity.com/paper/2110.10249/) (2021)
- [Wavelet Diffusion Neural Operator](https://mlsanity.com/paper/2412.04833/) (2024)
- [Neural Operator with Regularity Structure for Modeling Dynamics Driven by SPDEs](https://mlsanity.com/paper/2204.06255/) (2022)
- [Solving stochastic partial differential equations using neural networks in the Wiener chaos expansion](https://mlsanity.com/paper/2411.03384/) (2024)
- [Expanding the Chaos: Neural Operator for Stochastic (Partial) Differential Equations](https://mlsanity.com/paper/2601.01021/) (2026)
- [Functional SDE approximation inspired by a deep operator network architecture](https://mlsanity.com/paper/2402.03028/) (2024)
- [Latent-Variable Learning of SPDEs via Wiener Chaos](https://mlsanity.com/paper/2602.11794/) (2026)

## Built on by (most cited first, 1 of 1)

- [Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorization](https://mlsanity.com/paper/2605.15806/) (2026)

## Abstract (arXiv)

In this paper, we explore how our recently developed Wiener Chaos Expansion (WCE)-based neural operator (NO) can be applied to singular stochastic partial differential equations, e.g., the dynamic boldsymbolPhi42 model simulated in the recent works. Unlike the previous WCE-NO which solves SPDEs by simply inserting Wick-Hermite features into the backbone NO model, we leverage feature-wise linear modulation (FiLM) to appropriately capture the dependency between the solution of singular SPDE and its smooth remainder. The resulting WCE-FiLM-NO shows excellent performance on boldsymbolPhi42, as measured by relative L2 loss, out-of-distribution L2 loss, and autocorrelation score; all without the help of renormalisation factor. In addition, we also show the potential of simulating boldsymbolPhi43 data, which is more aligned with real scientific practice in statistical quantum field theory. To the best of our knowledge, this is among the first works to develop an efficient data-driven surrogate for the dynamical boldsymbolPhi43 model.
