---
title: Neural Diffeomorphic Neural Operator
url: https://www.emergentmind.com/topics/neural-diffeomorphic-neural-operator-ndno
type: topic
---

# Neural Diffeomorphic Neural Operator

Neural Diffeomorphic–Neural Operator (NDNO) denotes a family of operator-learning frameworks that extend neural operators from a single fixed spatial domain to families of varying geometries by composing the learned operator with diffeomorphic maps to and from a common reference domain. In the 2024 formulation, this idea appears as the “diffeomorphism neural operator (DNO),” whose central move is to transform operator learning on many physical domains into operator learning on one generic domain through precomputed diffeomorphisms; subsequent work develops “diffeomorphic latent neural operator” and learned diffeomorphic-embedding variants that retain the same pullback–operator–pushforward structure while changing how the diffeomorphism and latent operator are represented [2402.12475].

## 1. Definition, scope, and nomenclature

In its canonical form, NDNO addresses the problem of learning a single solution operator over a collection of smooth domains \(\{\Omega_i\}_{i=1}^n\) and parameter fields \(p_i(x)\), where the target is the PDE solution \(u_i(x)\). Classical neural operators, including the Fourier Neural Operator (FNO) and DeepONet-style models, are described as learning on one fixed spatial domain \(\Omega\); when \(\Omega_i\) changes, performance does not generally transfer without retraining or extensive multi-geometry data. NDNO removes this fixed-domain restriction by introducing smooth invertible maps \(\varphi_i:\Omega_i\to\Omega_{\rm ref}\), pulling all inputs and outputs back to a shared reference domain, training one latent or reference-domain operator there, and then pushing predictions forward to the original geometry [2402.12475].

Two terminological lines are visible in the literature. One uses “diffeomorphism neural operator” for the reference-domain construction on varying domains and parameters. Another uses “diffeomorphic latent neural operator” for a data-efficient DeepONet-style formulation in which geometry and physics are encoded jointly on the reference domain. A later mechanics-focused paper uses “neural diffeomorphic-neural operator” for an end-to-end learned diffeomorphic embedding plus neural operator pipeline on 3D point clouds [2411.18014]. Taken together, these papers suggest that NDNO names a methodological family rather than one uniquely standardized architecture.

The common premise is geometric factorization. Instead of asking a neural operator to learn shape variability implicitly, NDNO externalizes geometry into \(\varphi_i\) and reserves the neural operator for the transformed field problem. This is the essential distinction between NDNO and fixed-domain operator learning.

## 2. Core mathematical construction

Let \(\Omega_{\rm ref}\) be a fixed reference domain, such as \([0,1]^d\). For a physical domain \(\Omega_i\) and parameter fields \(p_i=\{a_i(x),f_i(x),\dots\}\), the pullback to the reference configuration is
\[
\bar p_i(x):=p_i(\varphi_i^{-1}(x)),\qquad x\in\Omega_{\rm ref}.
\]
NDNO then learns a reference-domain operator \(\bar{\mathcal T}_\theta\) such that
\[
\bar u_i=\bar{\mathcal T}_\theta[\bar p_i],
\qquad
u_i(x)=\bar u_i(\varphi_i(x)).
\]
Equivalently,
\[
u_i(x)\approx \bigl[\varphi_i^{-1}\circ \bar{\mathcal T}_\theta\circ \varphi_i\bigr](p_i)(x)
=\bigl(\bar{\mathcal T}_\theta[\bar p_i]\bigr)(\varphi_i(x)).
\]
Training is posed through an empirical \(L^2\)-loss over \((p_i,u_i)\) pairs, with an exact reference-domain expression weighted by \(\lvert\det D\varphi_i^{-1}\rvert\); in practice, point-wise sampling on a uniform grid in \(\Omega_{\rm ref}\) together with Monte-Carlo yields an unbiased estimate [2402.12475].

The same pullback–pushforward notation appears in the latent formulation. If \(u_i:\Omega_i\to\mathbb R^d\) and \(\varphi:\Omega_i\to\Omega_{\rm ref}\), then
\[
(\varphi^*u_i)(\xi)=u_i(\varphi^{-1}(\xi)),
\qquad
(\varphi_*u^0)(x)=u^0(\varphi(x)).
\]
Under this coordinate change, the differential operator changes as well. For the Laplacian,
\[
\Delta_x\,u_i(x)
=
\frac{1}{\lvert\det J_\varphi\rvert}
\nabla_\xi\!\cdot\!\Bigl(\lvert\det J_\varphi\rvert\,J_\varphi^{-T}J_\varphi^{-1}\,\nabla_\xi u^0(\xi)\Bigr),
\quad \xi=\varphi(x).
\]
This makes explicit that the choice of \(\varphi_i\) influences the regularity and structure of the transformed problem, not merely its discretization [2411.18014].

A distinctive diagnostic introduced in the original DNO work is the Domain Diffeomorphism Similarity (DDS), defined by normalized cross-correlation of pull-back coordinate maps. Empirically, lower DDS correlates with higher prediction error, and the proposed use of DDS is to assess generalization to an unseen domain before deployment [2402.12475].

## 3. Diffeomorphic mappings and architectural realizations

In the original two-stage DNO pipeline, the diffeomorphism is a preprocessing object. In 2D, \(\varphi_i:\Omega_i\to\Omega_{\rm ref}\) is obtained by solving a harmonic map \(\Delta f=0\) with Dirichlet boundary conditions; practically, this is a sparse linear system on a triangle mesh whose cotangent-Laplace discretization enforces conformal bijectivity. In 3D, volume parameterization, including transfinite interpolation, maps each part into a bounding cuboid with a proof of positive Jacobian ensuring smooth invertibility. After \(\varphi_i\) is computed, a uniform grid is sampled on \(\Omega_{\rm ref}\), pulled back through \(\varphi_i^{-1}\), and used to interpolate parameter fields into a regular array. A standard FNO is then applied on the reference domain, with lifting \(P\), spectral layers using FFT and iFFT, and projection \(Q\); the reported experimental setting typically used 6 Fourier layers, 4 fully-connected layers in \(P\) and \(Q\), and grids of resolution \(128\times128\) in 2D or \(128^3\) in 3D [2402.12475].

The diffeomorphic latent formulation replaces the pure FNO pipeline with a DeepONet-style architecture. It combines a geometry encoder \(g_{\rm geo}(\alpha_i)\), a physical-condition encoder \(g_{\rm phys}(v_i^0)\), and a spatial trunk \(h(\xi)\), then forms the latent solution as
\[
\hat u_i^0(\xi)=\sum_{j=1}^s w_j\bigl(g_{\rm geo}(\alpha_i),g_{\rm phys}(v_i^0)\bigr)\,h_j(\xi).
\]
The mapping \(\varphi_i\) may be constructed by Schwarz–Christoffel conformal maps, LDDMM, or discrete OT. A central claim of this work is that maps respecting PDE invariances can substantially reduce data requirements; the extreme case highlighted is conformal invariance of the Laplacian [2411.18014].

A later 3D mechanics formulation learns \(\varphi\) rather than precomputing it. There, \(\phi:\Omega\to\hat\Omega\) is realized by a registration neural network acting on source and target point clouds. The total mapping loss is
\[
L_{\rm map}(\phi)=L_{\rm sim}(\phi;S,T)+L_{\rm diffeo}(\phi),
\]
where \(L_{\rm sim}\) uses the Sinkhorn distance, \(L_{\rm inv}=\sum_i \mathrm{ReLU}[-\det J\phi(x_i)]\) penalizes folding or inversion, and \(L_{\rm smooth}=\sum_i \|J\phi(x_i)-I\|_F^2\) discourages overly wiggly mappings. The reported architecture uses two DGCNN backbones, a cross-attention Transformer, and an MLP projection module, while the operator on \(\hat\Omega\) is a Geo-FNO-style network with a lifting MLP, 4 Fourier layers, 12 retained modes, and channel width 64 [2509.12237].

## 4. Reported empirical performance

The original DNO paper evaluates the framework on Darcy flow, incompressible pipe flow, airfoil flow, and 3D mechanics. The reported benchmarks are summarized below [2402.12475].

| Benchmark | Setting | Reported result |
|---|---|---|
| Darcy flow | 200 pentagon-shaped domains; test on scaled pentagons and unseen hexagons | \(L^2\)-error below \(1\times10^{-1}\) on scaled pentagons; unseen hexagons \(\sim 1.2\times10^{-1}\) |
| Pipe flow | Pipe cross-sections with 2–6 baffles; inlet velocity in \([1,4]\) | Velocity-field \(L^2\)-errors in \([4\times10^{-2},1\times10^{-1}]\); stable over temporal snapshots |
| Airfoil flow | 5 airfoils; Reynolds \(0.5\)–\(5\times10^6\); angle-of-attack \(\pm 22.5^\circ\) | Five unseen airfoil/flow conditions with \(L^2\) errors \(\lesssim 5\times10^{-3}\) |
| 3D mechanics | 12 part variants; train on first 10 | Test \(L^2\) error \(=4.1\times10^{-2}\); last 2 variants \(=4.4\times10^{-2}\); new topology \(=7.3\times10^{-2}\) |

These experiments are used to support two claims: first, that one learned operator can be reused across domains after diffeomorphic transfer; second, that DDS tracks domain-shift risk, since the paper reports a strong monotonic trend between DDS and prediction error on the Darcy task [2402.12475].

The latent-data-efficient formulation studies the 2D Laplace equation on random doubly-connected domains mapped to a fixed annulus. Using relative \(L^2\) error on test domains, the reported results are 0.26% error with 80 samples for conformal NDNO without a geometry branch, 2.56% error with 400 samples for LDDMM NDNO with 10 PCA modes, and 22.4% error with 400 samples for discrete OT NDNO. The error-versus-training-size curves are reported to show that conformal NDNO is one order of magnitude more data-efficient [2411.18014].

The learned-embedding 3D mechanics variant reports both accuracy and speed. For frame parts, NDNO gives avg. max-error \(=0.0452\,\mathrm{mm}\) and RMSE \(=0.0140\,\mathrm{mm}\) on main-direction prediction, compared with Geo-FNO max \(=0.0603\,\mathrm{mm}\) and RMSE \(=0.0188\,\mathrm{mm}\). For C-beam parts, NDNO reports max \(=0.0273\,\mathrm{mm}\) and RMSE \(=0.0076\,\mathrm{mm}\), compared with Geo-FNO max \(=0.0597\,\mathrm{mm}\) and RMSE \(=0.0090\,\mathrm{mm}\). Reported inference for a new part is \(\approx 0.02\,\mathrm{s}\) for registration, \(\approx 0.01\,\mathrm{s}\) for operator evaluation, and \(\approx 30\,\mathrm{ms}\) overall on a single GPU, which the paper states is \(10^3\)–\(10^4\times\) faster than a mesh-based FEA solve [2509.12237].

## 5. Relation to broader diffeomorphic operator learning

A broader geometric line of work uses diffeomorphisms not only to normalize varying domains, but also to represent the evolution operator itself. In “Diffeomorphic Neural Operator Learning,” the learned object is a lift
\[
\mathcal K_\theta:U\to \mathrm{Diff}^1(\Omega),
\]
and the one-step evolution is defined by pull-back,
\[
\Phi_{\Delta t}[u]=u\circ\bigl(\mathcal K_\theta(u)\bigr)^{-1}.
\]
This converts semigroup time evolution into group composition in \(\mathrm{Diff}(\Omega)\). The reported consequences are hard enforcement of relabeling symmetry, exact preservation of masses of level-sets, extrema, Sobolev norms, and invariants of ideal transport up to interpolation error, together with non-diffusivity. On a Zalesak slotted cylinder test, the paper reports almost machine-precision transport and mass-conservation error \(\approx 10^{-7}\); on 2D incompressible Euler turbulence, it reports stability up to \(T=10\), enstrophy-error \(\sim 10^{-5}\), and recovery of the \(k^{-3}\) cascade on \(512^2\) and \(1024^2\) grids [2508.06690].

NeurEPDiff is closely related at the level of diffeomorphic dynamics rather than domain transfer. It learns the EPDiff solution map from an initial velocity field to a trajectory of velocities in the LDDMM/geodesic-shooting setting. Its operator is built from global integral kernels and smooth activations in a Fourier-bandlimited representation, and the paper emphasizes resolution invariance: the same trained parameters can be used at \(64^d\), \(128^d\), or \(192^d\) without retraining. Reported CPU runtimes for EPDiff alone are \(0.045\,\mathrm{s}\), \(0.12\,\mathrm{s}\), and \(0.33\,\mathrm{s}\) for NeurEPDiff at \(64^3\), \(128^3\), and \(192^3\), versus \(0.37\,\mathrm{s}\), \(1.24\,\mathrm{s}\), and \(21.8\,\mathrm{s}\) for FLASH, and \(3.85\,\mathrm{s}\), \(30.76\,\mathrm{s}\), and \(103.09\,\mathrm{s}\) for LDDMM; average Dice scores on OASIS MRI are reported as 0.866, 0.883, and 0.878 for NeurEPDiff at those resolutions [2303.07115].

These related developments situate NDNO within a larger program of geometric operator learning in which diffeomorphism structure is not incidental preprocessing, but part of the learned representation.

## 6. Limitations, applications, and theoretical status

The principal limitations reported for the original NDNO/DNO framework are geometric rather than purely statistical. If two domains are only weakly diffeomorphic in the sense of low DDS, prediction error may grow. High-genus shapes or topological changes break simple harmonic or volume-parameterization approaches and require more sophisticated manifold charts. The map \(\varphi_i\) must also be precomputed and stored, so extremely fine meshes raise memory demands [2402.12475].

The same papers outline several extensions. These include incorporating group-equivariance, such as rotation or reflection, into the reference-domain operator; learning \(\varphi_i\) jointly with the operator in an end-to-end fashion; extending the method to time-dependent domains by a time-parameterized \(\varphi_i(t)\); and combining the framework with PINN-style physics-informed losses in data-scarce regimes. The reported application areas are design optimization in aerodynamics, real-time digital twins in manufacturing, multi-physics couplings in geosciences, and high-throughput materials screening [2402.12475].

A theoretical counterpart appears in work on Neural Shape Operator Surrogates. There, a family of admissible domains \(D_\mu\) is generated from one bounded Lipschitz reference domain \(D_0\) by an affine-parametric diffeomorphic encoder \(V_\mu\), and the PDE is pulled back to \(D_0\). Under holomorphy and uniform coercivity assumptions, the pull-backed parametric solution family is well posed, and the map \(\mu\mapsto \hat u_\mu\) is holomorphic. The operator surrogate is written
\[
N_N(\mu)=D\circ \phi_N\circ E(\mu),
\]
with linear encoder \(E\), ReLU network \(\phi_N\), and decoder \(D\). The main expression-rate result states that for any \(\delta>0\), there exists \(C\) such that
\[
\sup_{V\in C_r^s}\|S(V)-(D\circ\phi_N\circ E)(V)\|_{H^1(D_0)}
\le C\,N^{-(\min(s-1,t)-\delta)}.
\]
Harbrecht and Schwab explicitly interpret this construction as instantiating the NDNO paradigm, and their results provide uniform approximation and generalization guarantees across the admissible shape family [2604.18012].

A recurring misconception is that diffeomorphic transfer eliminates geometry dependence altogether. The reported evidence does not support that stronger claim. Rather, NDNO replaces raw geometry variation by a structured latent geometry problem: success depends on the quality, smoothness, invertibility, and PDE-compatibility of the chosen or learned diffeomorphism, and generalization remains sensitive to the transformed similarity of domains, as quantified in the original framework by DDS [2402.12475].

Source: https://www.emergentmind.com/topics/neural-diffeomorphic-neural-operator-ndno