Divergence-Free Transport Noise Layer
Implementation & benchmark of arXiv:2607.15140 — Absence of blow-up in the 3D Navier-Stokes equations with transport noise
Source paper: Absence of blow-up in the 3D Navier-Stokes equations with transport noise arXiv:2607.15140 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Inject Stratonovich transport noise into intermediate spatial feature maps instead of adding independent elementwise Gaussian noise. Choose divergence-free vector fields whose covariance is approximately isotropic, so the corresponding Itô correction acts like a tunable Laplacian and preferentially suppresses unstable high-frequency feature components.
Formulas
Mathematical statement
The paper studies the stochastic Navier-Stokes transport term, where $u$ is a divergence-free velocity field, $P$ is pressure, $\sigma_{k,\alpha}$ are smooth divergence-free vector fields, $W^{k,\alpha}_t$ are independent Brownian motions, and $\circ dW^{k,\alpha}_t$ denotes Stratonovich integration. The transferable operator is $T_jh=(\sigma_j\cdot\nabla)h$, applied to a feature map $h$, with $\nabla\cdot\sigma_j=0$. Use the adapted feature SDE $dh=b_\theta(h,t)dt+\sqrt{2\kappa}\sum_{j=1}^mT_jh\circ dW^j_t$, where $b_\theta$ is the ordinary neural-network drift and $\kappa\ge0$ is the noise strength. Its Itô form adds the deterministic correction $\kappa\sum_jT_j^2h\,dt$. If $\sum_j\sigma_j(x)\sigma_j(x)^\top\approx I$ and first-order coefficient terms cancel, then $\sum_jT_j^2\approx\Delta$, so the correction damps Fourier mode $\xi$ at rate approximately $\kappa|\xi|^2$. The paper's localized estimate also motivates monitoring feature moments: $(\mathbf E|v^N(t_0,x_0)|^r)^{1/r}\lesssim(\mathbf E\|F\|_{L^p(I;L^q(B))^r})^{1/r}+(\mathbf E\|G\|_{L^p(I;L^q(B))^r})^{1/r}+(\mathbf E\|v^N\|_{\underline L^2(N^{-1}Q)}^r)^{1/r}+\mathscr C_Q[\pi^N,(\widetilde\pi^N_{k,\alpha})_{k,\alpha}]$, where $F,G$ are forcing terms, $v^N$ is a localized solution, and $\mathscr C_Q$ is a pressure contribution. In the network, replace pressure by ordinary drift and use empirical feature and gradient moments as stability diagnostics.
Implementation notes
1. Integration point: insert the layer after a convolution and normalization in a small CNN, operating on $h\in\mathbb R^{B\times C\times H\times W}$. Use periodic padding for the first experiment to avoid boundary artifacts. Construct $m=4$ or $8$ smooth divergence-free fields on the grid. A simple choice is constant fields $\sigma_1=(1,0)$ and $\sigma_2=(0,1)$; augment them with low-frequency fields generated from stream functions $\psi_j$ via $\sigma_j=(\partial_y\psi_j,-\partial_x\psi_j)$, which guarantees $\nabla\cdot\sigma_j=0$. 2. Pseudocode: sample independent $\varepsilon_j\sim\mathcal N(0,1)$; compute centered finite differences $D_xh,D_yh$; set $T_jh=\sigma_{j,x}D_xh+\sigma_{j,y}D_yh$; compute $T_j^2h$ by applying the same operator to $T_jh$; return $h+\sqrt{2\kappa\Delta t}\sum_j\varepsilon_jT_jh+\kappa\Delta t\sum_jT_j^2h$. This is Euler-Maruyama for the Itô form and implements the paper's Stratonovich transport noise plus correction. Start with $\Delta t=1$ and clip $\kappa$ so the estimated diffusion CFL number is below $0.25$. 3. Compute vector fields analytically and estimate effective diffusion by applying the operator to Fourier basis images; do not learn $\sigma_j$ in the MVP. Compare stochastic-only, correction-only, and combined variants. Measure high/low-frequency feature-energy ratio, activation variance, gradient-norm tails, corruption accuracy, and clean accuracy. 4. First experiment: train ResNet-18 on CIFAR-10. Insert layers after stages 1 and 2, and compare baseline, dropout, additive Gaussian activation noise, explicit Gaussian blur, and transport noise at matched wall-clock cost. Sweep $\kappa\in\{0.001,0.003,0.01,0.03\}$. Success means lower worst-case gradient and activation moments and improved CIFAR-10-C high-frequency corruption accuracy without reducing clean accuracy by more than 0.3 percentage points.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.