NeSTR: Neural S-Transform Reconstruction for Gaussian SPDEs

Abstract

We introduce Neural $S$-Transform Reconstruction (NeSTR) for Gaussian stochastic partial differential equations (SPDEs). The learned object is not the random solution itself, but its $S$-Transform on a finite dimensional subspace of the white noise test space. For Wick type SPDEs, NeSTR yields a deterministic parametric PDE for the restricted $S$-Transform, and the stochastic solution is recovered from the Taylor coefficients of this transform at the origin. Thus the neural approximation is tied directly to the Wiener chaos expansion of the solution. The main novelty of NeSTR is an inverse $S$-Transform reconstruction principle combined with finite mode learning: we prove consistency of the recovered chaos coefficients, separate noise mode, chaos truncation, deterministic solver, and neural approximation errors, and give a deterministic learning formulation for the restricted transform. Numerical tests on an additive stochastic heat equation and a multiplicative Wick heat equation show that polynomial spectral representations in the transform variables provide stable access to higher order chaos coefficients. The resulting reconstruction separates the equation dependent deterministic coefficient fields from a reusable Gaussian--Hermite basis. Consequently, NeSTR learns an analytic generating function whose derivatives recover the Wiener chaos coefficients, rather than learning stochastic trajectories directly.