PGD-NO: A Neural Operator with Precomputed Geometry Decomposition for 3D Million-scale Physics Simulations

arXiv cs.LG Papers

Summary

PGD-NO is a neural operator that precomputes geometry decomposition to achieve linear memory scalability, enabling high-fidelity physics simulations on meshes exceeding 10 million nodes and overcoming the single-node memory bottleneck.

arXiv:2607.08025v1 Announce Type: new Abstract: While neural PDE solvers have demonstrated significant potential for accelerating engineering simulations, existing architectures remain constrained by high memory consumption and the single node bottleneck, where the maximum processable mesh resolution is strictly limited by the VRAM of a single compute unit. To address these challenges, we propose PGD-NO, a neural operator with Precomputed Geometry Decomposition, that relocates the computational overhead of geometric encoding to a deterministic pre-computation phase. By utilizing an iterative geometry decomposition algorithm to extract geometry tokens, our model decouples feature extraction from solution querying. This architecture enables linear memory scalability, allowing high fidelity learning on meshes exceeding 10 million nodes, a scale where existing architectures typically encounter memory exhaustion. PGD-NO demonstrates competitive predictive accuracy across diverse industrial benchmarks and provides intrinsic interpretability through attention mechanisms. By effectively overcoming traditional mesh-size constraints, PGD-NO offers a robust and efficient solution for the next generation of large-scale, high-fidelity industrial design applications.
Original Article
View Cached Full Text

Cached at: 07/10/26, 06:17 AM

# A Neural Operator with Precomputed Geometry Decomposition for 3D Million-scale Physics Simulations
Source: [https://arxiv.org/html/2607.08025](https://arxiv.org/html/2607.08025)
###### Abstract

While neural PDE solvers have demonstrated significant potential for accelerating engineering simulations, existing architectures remain constrained by high memory consumption and the ”single\-node bottleneck,” where the maximum processable mesh resolution is strictly limited by the VRAM of a single compute unit\. To address these challenges, we propose PGD\-NO, a neural operator with Precomputed Geometry Decomposition, that relocates the computational overhead of geometric encoding to a deterministic pre\-computation phase\. By utilizing an iterative geometry decomposition algorithm to extract ”geometry tokens,” our model decouples feature extraction from solution querying\. This architecture enables linear memory scalability, allowing high\-fidelity learning on meshes exceeding 10 million nodes—a scale where existing architectures typically encounter memory exhaustion\. PGD\-NO demonstrates competitive predictive accuracy across diverse industrial benchmarks and provides intrinsic interpretability through attention mechanisms\. By effectively overcoming traditional mesh\-size constraints, PGD\-NO offers a robust and efficient solution for the next generation of large\-scale, high\-fidelity industrial design applications\. Our codes and datasets are available at[https://github\.com/WeihengZ/PGD\-NO](https://github.com/WeihengZ/PGD-NO)\.

Neural Operator, computational physics, industry\-scale

## 1Introduction

The numerical solution of partial differential equations \(PDEs\) is a cornerstone of computational science and engineering, facilitating evaluations\(Jagotaet al\.,[2013](https://arxiv.org/html/2607.08025#bib.bib2)\), sensitivity analysis\(Hisada,[1995](https://arxiv.org/html/2607.08025#bib.bib61)\), and uncertainty quantification\(Zhong and Meidani,[2023](https://arxiv.org/html/2607.08025#bib.bib37)\)in industrial design\. Traditional discretization\-based methods—such as the finite element method \(FEM\)\(Jagotaet al\.,[2013](https://arxiv.org/html/2607.08025#bib.bib2)\)and finite volume method \(FVM\)\(Moukalledet al\.,[2015](https://arxiv.org/html/2607.08025#bib.bib22)\)—are highly refined, offering significant accuracy and robustness across diverse domains\. Nevertheless, these conventional approaches often involve prohibitive computational costs, particularly for large\-scale, high\-fidelity simulations\(Luet al\.,[2021a](https://arxiv.org/html/2607.08025#bib.bib25)\)\. To mitigate these demands, deep learning architectures have been investigated as efficient surrogates for numerical methods, called neural PDE solvers\(Luet al\.,[2021b](https://arxiv.org/html/2607.08025#bib.bib36); Liet al\.,[2021](https://arxiv.org/html/2607.08025#bib.bib26); Wuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib53)\)\. By training on simulation data, these neural solvers learn to approximate the mapping between inputs and outputs of the traditional methods, thereby significantly accelerating the manufacturing and design process\.

Industrial applications typically involve large, complex geometries, requiring models to efficiently capture the underlying physical features\(Wuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib53)\)\. DeepONet\(Luet al\.,[2021b](https://arxiv.org/html/2607.08025#bib.bib36)\)was initially applied to industrial design because many geometries are governed by low\-dimensional parameter vectors for manufacturing purposes\. However, recent studies indicate that the performance of DeepONet diminishes as geometric complexity increases\(Liet al\.,[2021](https://arxiv.org/html/2607.08025#bib.bib26)\)\. Simultaneously, various deep learning models have been proposed for freeform geometries\. Although these methods show promise in benchmark problems, they struggle with real\-world industrial tasks involving millions of mesh points\(Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\)\. This is because training on such massive datasets requires significant GPU memory, which often exceeds the capacity of standard hardware\.

![Refer to caption](https://arxiv.org/html/2607.08025v1/x1.png)

Figure 1:Architectural comparison and memory scaling\. \(Left\) Encoding mechanisms: GINO utilizes grid\-based mapping, and Transolver\+\+ computes physics tokens via weighted nodal sums, whereas PGD\-NO employs pre\-computed geometry tokens to bypass intensive encoding\. \(Right\) Peak GPU memory on a GH200 system: PGD\-NO maintains the lowest footprint, enabling high\-resolution simulations \(up to 10M nodes\) that exceed the capacity of traditional grid\-based or graph\-based neural operators\.Existing neural PDE solvers for freeform geometries can be categorized into graph\-based, grid\-based, and point\-based models, depending on how they represent PDE solutions on a mesh\. Graph\-based models\(Anandkumaret al\.,[2020](https://arxiv.org/html/2607.08025#bib.bib8); Velascoet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib35); Liet al\.,[2020](https://arxiv.org/html/2607.08025#bib.bib27); Gladstoneet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib14)\)may fail to capture complex geometries if they rely on sparse mesh graphs; conversely, they can rapidly exhaust GPU memory if they utilize radius\-based methods to synthesize edges\. Grid\-based models\(Sahebiet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib45)\), such as GINO\(Liet al\.,[2023b](https://arxiv.org/html/2607.08025#bib.bib32)\), are also memory\-intensive, as capturing complex geometries requires mapping point features onto high\-resolution grids\. In contrast, point\-based methods\(Kashefi and Mukerji,[2022](https://arxiv.org/html/2607.08025#bib.bib13); Park and Kang,[2026](https://arxiv.org/html/2607.08025#bib.bib42); Lee and Oh,[2024](https://arxiv.org/html/2607.08025#bib.bib58)\), such as Transolver\(Wuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib53)\), utilize a physics\-attention mechanism to identify underlying ”physics tokens,” demonstrating that complex geometric features can be captured directly from point clouds\. Recently, Transolver\+\+\(Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\)has extended this capability to million\-scale problems through an optimized parallelism framework\.

Despite its advancements, Transolver\+\+ still faces limitations when scaled to high\-fidelity PDE solving tasks\. In practice, we observe significant bottlenecks in training efficiency when mesh sizes exceed 10 million nodes\. While Transolver\+\+ can theoretically scale via parallel computation, this comes at the cost of increased communication overhead\. Although intra\-node communication remains efficient due to NVLink\(Weiet al\.,[2023](https://arxiv.org/html/2607.08025#bib.bib23)\)architecture, scaling beyond a single compute node necessitates data transfers between compute nodes via standard networking\. This inter\-node communication introduces substantial latency, significantly increasing the duration of each training epoch\. Consequently, the practical mesh\-size capacity of Transolver\+\+ remains constrained by the resources of a single compute node\. As illustrated in Figure[1](https://arxiv.org/html/2607.08025#S1.F1), even utilizing the latest 4\-way GH200 architecture with 360GB of total memory, a Transolver\+\+ model with a reduced number of layers can barely accommodate 10M nodes for CFD simulation learning\.

We consider that the GPU memory limitations of existing methods stem from their reliance on learning\-based encoding process to map complex geometries into hidden representations\. In contrast, geometric features can be pre\-computed using deterministic algorithms, which significantly reduces GPU memory consumption by bypassing the encoding process\. Motivated by this observation, this work explores an alternative paradigm for developing neural PDE solvers\. We introduce a geometry decomposition algorithm for feature extraction and propose a simple but effective decoding architecture to predict PDE solutions based on these pre\-computed features\. Hence, our primary contributions in this work are as follows:

- •We introduce a deterministic hierarchical decomposition algorithm to extract “geometry tokens,” demonstrating that PDE solutions can be effectively learned using non\-learnable geometric primitives\.
- •We propose the Pre\-computed Geometry Decomposition Neural Operator \(PGD\-NO\), which decouples geometry encoding from solution querying, eliminating inter\-node communication for million\-scale mesh processing on standard hardware\.
- •We evaluate our model against current state\-of\-the\-art methods across five industrial benchmarks, including a novel, self\-generated dataset of up to 12 million nodes, where our approach achieves superior performance\.
- •We further demonstrate that PGD\-NO enables efficient training via random sampling while providing enhanced interpretability for its physical predictions

Conflict of Interest DisclosureThe authors J\.B\. and V\.O\. are employed by Dassault Systèmes, and W\.Z\. is currently employed by Dassault Systèmes\. This paper evaluates multiple datasets, including one provided by Dassault Systèmes\.

## 2Related Works

### 2\.1Neural PDE Solver

Inspired by DeepONet, branch\-trunk operators leverages the universal operator approximation theorem to map infinite\-dimensional function spaces\(Luet al\.,[2021a](https://arxiv.org/html/2607.08025#bib.bib25)\)\. These architectures consist of a branch network encoding input functions and a trunk network processing output coordinates\(Luet al\.,[2021b](https://arxiv.org/html/2607.08025#bib.bib36)\)\. While extensions have addressed the curse of dimensionality\(Mandlet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib38)\), uncertainty quantification\(Loneet al\.,[2026](https://arxiv.org/html/2607.08025#bib.bib39)\), and computational efficiency\(Zhanget al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib40); Qiuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib41)\), and specialized variants like GANO\(Zhong and Meidani,[2025](https://arxiv.org/html/2607.08025#bib.bib15)\)and Point\-DeepONet\(Park and Kang,[2026](https://arxiv.org/html/2607.08025#bib.bib42)\)incorporate geometric information, these models generally rely on fixed sensor locations or global pooling\. Consequently, despite improvements in expressiveness\(Zhong and Meidani,[2024](https://arxiv.org/html/2607.08025#bib.bib7)\)and temporal encoding\(Heet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib6)\), branch\-trunk architectures remain fundamentally limited in their ability to handle truly freeform, complex geometries\.

Graph\-based neural operators \(GNOs\)\(Anandkumaret al\.,[2020](https://arxiv.org/html/2607.08025#bib.bib8)\)are designed for irregular domains by utilizing message\-passing frameworks\. Key developments such as MGNO\(Liet al\.,[2020](https://arxiv.org/html/2607.08025#bib.bib27)\)and LNO\(Wang and Wang,[2024](https://arxiv.org/html/2607.08025#bib.bib28)\)enable the capture of long\-range interactions and expressive latent mappings\. To generalize across arbitrary geometries, models like CORAL\(Serranoet al\.,[2023](https://arxiv.org/html/2607.08025#bib.bib19)\), EA\-GNN\(Gladstoneet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib14)\), and PDE\-GCN\(Eliasofet al\.,[2021](https://arxiv.org/html/2607.08025#bib.bib33)\)have introduced coordinate\-based features and PDE\-aware message passing\. However, a significant drawback of these methods is the memory overhead required to store and process edge information; as the mesh density increases, the memory cost of the graph connectivity can quickly exceed the capacity of modern hardware\.

Grid\-based operators, primarily led by the Fourier Neural Operator \(FNO\)\(Liet al\.,[2021](https://arxiv.org/html/2607.08025#bib.bib26)\), capture global dependencies in the frequency domain with high efficiency\. Scalable variants like D\-FNO\(Li and Ye,[2025](https://arxiv.org/html/2607.08025#bib.bib46)\), alongside geometry\-aware models like GINO\(Liet al\.,[2023b](https://arxiv.org/html/2607.08025#bib.bib32)\)and DAFNO\(Liuet al\.,[2023](https://arxiv.org/html/2607.08025#bib.bib47)\), attempt to adapt spectral methods to irregular topologies\. While these models provide significant speedups\(Luet al\.,[2022](https://arxiv.org/html/2607.08025#bib.bib51)\)and incorporate physics\-informed constraints\(Liet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib50)\), they rely on mapping unstructured data to latent regular grids\. In large\-scale industrial applications, these grids must maintain an extremely high resolution to avoid significant information loss, which becomes computationally prohibitive\.

Point\-based operators offer the greatest flexibility for nonuniform resolutions and complex geometries by operating directly on spatial point clouds\. Architectures such as GNOT\(Haoet al\.,[2023](https://arxiv.org/html/2607.08025#bib.bib1)\)and Transolver\(Wuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib53)\)utilize physics\-aware attention to identify underlying tokens, while frameworks like Universal Physics Transformers \(UPT\)\(Alkinet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib54)\)and GFN\(Morrisonet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib57)\)ensure resolution invariance\. Although these methods mitigate numerical artifacts\(Zhenget al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib55); Yuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib56)\)and achieve better scalability\(Liet al\.,[2023a](https://arxiv.org/html/2607.08025#bib.bib52)\), a clear limitation persists: while their memory complexity scales linearly with the number of nodes, the maximum node capacity remains strictly capped by the GPU memory limits of a single compute node\.

### 2\.2Geometry Decomposition

Traditional geometric decomposition methods\(Jermannet al\.,[2006](https://arxiv.org/html/2607.08025#bib.bib63)\)often rely on deterministic algorithms such as K\-means clustering on mesh faces\(Mamou and Ghorbel,[2009](https://arxiv.org/html/2607.08025#bib.bib64)\), hierarchical clustering based on curvature, or spectral clustering utilizing the Laplace\-Beltrami operator\(Lévy,[2006](https://arxiv.org/html/2607.08025#bib.bib65)\)to identify salient features\. Advanced techniques such as convex decomposition\(Lien and Amato,[2008](https://arxiv.org/html/2607.08025#bib.bib66)\)aim to break complex, non\-convex volumes into a set of simpler, nearly convex parts to facilitate more efficient physical simulations\. In the context of industrial design, primitive\-based segmentation is frequently employed to decompose geometries into standard shapes \(e\.g\., planes, cylinders, or spheres\) that correspond to manufacturing parameters\(Elrefaieet al\.,[2024a](https://arxiv.org/html/2607.08025#bib.bib9); Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\)\. While these classical segmentation methods offer high interpretability without requiring training data, their integration into end\-to\-end neural PDE solvers remains largely unexplored\. Furthermore, achieving further reductions in computational complexity requires a method to identify a sparse yet representative set of geometry tokens—a capability currently absent in existing geometric segmentation algorithms

## 3Method

### 3\.1Geometry Token Extraction

The proposed workflow transforms a raw 3D mesh into a set of discrete, semantically meaningful physics tokens by combining graph topology with geometric feature detection\.

Step \(1\) Graph Construction:The process begins by transforming the raw 3D mesh into a structured topological networkG=\(V,E\)G=\(V,E\), whereVVis the set of mesh vertices andEEis the set of edges\.

Step \(2\) Sharp Edge Identification:To determine where a mesh should naturally break, the algorithm identifies sharp features using geometric gradients\. The procedure evaluates the dihedral angleθi​j\\theta\_\{ij\}between the normals𝐧a\\mathbf\{n\}\_\{a\}and𝐧b\\mathbf\{n\}\_\{b\}of adjacent faces sharing edgeei​je\_\{ij\}\. If the angle satisfies:θi​j=arccos⁡\(𝐧a⋅𝐧b\)\>θthreshold\\theta\_\{ij\}=\\arccos\(\\mathbf\{n\}\_\{a\}\\cdot\\mathbf\{n\}\_\{b\}\)\>\\theta\_\{\\text\{threshold\}\}, the nodesv∈ei​jv\\in e\_\{ij\}are added to a node set of sharp edgesVsharpV\_\{\\text\{sharp\}\}, whereθthreshold\\theta\_\{\\text\{threshold\}\}is the angle threshold\. This acts as a structural shattering mechanism\(Szusziket al\.,[2026](https://arxiv.org/html/2607.08025#bib.bib67)\), where prominent ridges are marked for disconnection\.

Step \(3\) Graph Partition:Once the sharp edges are identified, the partition is executed by creating a subgraphG′=G​\[V∖Vsharp\]G^\{\\prime\}=G\[V\\setminus V\_\{\\text\{sharp\}\}\], where all sharp edge boundary are removed\. This operation of node removal causes the graph to fracture into a set of independent connected components𝒞=\{C1,C2,…,Ck\}\\mathcal\{C\}=\\\{C\_\{1\},C\_\{2\},\\dots,C\_\{k\}\\\}\. To achieve a hierarchical refinement of the mesh, the process is performed iteratively across a sequence of thresholds\. We introduce a threshold decay mechanism,θthresholdi\+1=α​θthresholdi\\theta^\{i\+1\}\_\{\\text\{threshold\}\}=\\alpha\\theta^\{i\}\_\{\\text\{threshold\}\}with a hyper\-parameterα∈\(0,1\)\\alpha\\in\(0,1\), which progressively increases the sensitivity of sharp edge detection\. This decay allows the algorithm to capture coarse structural parts at higher angles before isolating finer geometric details at lower angles\. Any subgraphCiC\_\{i\}that satisfies the size constraint\|Ci\|\>min\_size\|C\_\{i\}\|\>\\text\{min\\\_size\}is preserved, thereby preventing the generation of excessively small geometry tokens that could undermine computational efficiency\.

Step \(4\) Graph Merge:To prevent the creation of excessive noise, the algorithm employs a merging logic based on proximity\. For a small fragmentGsmallG\_\{\\text\{small\}\}where\|Vsmall\|<threshold\|V\_\{\\text\{small\}\}\|<\\text\{threshold\}, the algorithm identifies the boundary set∂Vsmall\\partial V\_\{\\text\{small\}\}and expands it byhhhops using the adjacency relationAdj​\(v\)\\text\{Adj\}\(v\)\. The merge occurs when an expanded setVsmall\(h\)V\_\{\\text\{small\}\}^\{\(h\)\}intersects with another larger graph node setVlargeV\_\{\\text\{large\}\}\. Using this strategy, small graphs are absorbed into the most relevant neighbor, ensuring that the final set of tokens remains computationally manageable\.

Step \(5\) Geometry token matrix:In the last step, we create the mapping matrixS∈ℝK×NS\\in\\mathbb\{R\}^\{K\\times N\}representing the relationship between geometry tokens and mesh nodes, whereKKis the number of geometry tokens andNNis the number of mesh vertices:

Sk​i=\{1\|Ck\|if​vi∈Ck,0otherwise\.S\_\{ki\}=\\begin\{cases\}\\frac\{1\}\{\|C\_\{k\}\|\}&\\text\{if \}v\_\{i\}\\in C\_\{k\},\\\\ 0&\\text\{otherwise\}\.\\end\{cases\}\(1\)
The details of the geometry token creation algorithm is shown in Algorithm[1](https://arxiv.org/html/2607.08025#alg1)\.

Algorithm 1Iterative Geometry Decomposition1:Input:Mesh PolyData

𝒫\\mathcal\{P\}, Coordinates

𝐗\\mathbf\{X\}, Angle thresholds

Θ=\{θ1,θ2,…,θm\}\\Theta=\\\{\\theta\_\{1\},\\theta\_\{2\},\\dots,\\theta\_\{m\}\\\}, Min graph size ratio

ρ\\rho
2:1\. Graph construction:

3:Build adjacency graph

G=\(V,E\)G=\(V,E\)from mesh connectivity

𝒫\\mathcal\{P\}
4:Initialize set of subgraphs to split:

𝒮=\{G\}\\mathcal\{S\}=\\\{G\\\}
5:2\. Iterative partition with sharp edges:

6:for

θ∈Θ\\theta\\in\\Thetado

7:

Vs​h​a​r​p=\{v∈V∣dihedral angle at​v\>θ\}V\_\{sharp\}=\\\{v\\in V\\mid\\text\{dihedral angle at \}v\>\\theta\\\}
8:

𝒮next=∅\\mathcal\{S\}\_\{\\text\{next\}\}=\\emptyset
9:for

g∈𝒮g\\in\\mathcal\{S\}do

10:

\{𝒞1,…,𝒞k\}=Graph\-Split​\(g∖Vs​h​a​r​p\)\\\{\\mathcal\{C\}\_\{1\},\\dots,\\mathcal\{C\}\_\{k\}\\\}=\\text\{Graph\-Split\}\(g\\setminus V\_\{sharp\}\)
11:

𝒮next=𝒮next∪\{𝒞i∣\|𝒞i\|\>ρ⋅\|V\|\}\\mathcal\{S\}\_\{\\text\{next\}\}=\\mathcal\{S\}\_\{\\text\{next\}\}\\cup\\\{\\mathcal\{C\}\_\{i\}\\mid\|\\mathcal\{C\}\_\{i\}\|\>\\rho\\cdot\|V\|\\\}
12:endfor

13:

𝒮=𝒮next\\mathcal\{S\}=\\mathcal\{S\}\_\{\\text\{next\}\}
14:if

𝒮=∅\\mathcal\{S\}=\\emptysetthen

15:break

16:endif

17:endfor

18:3\. Graph remerge

19:

Vu​n​a​s​s​i​g​n​e​d=V∖⋃g∈𝒮V​\(g\)V\_\{unassigned\}=V\\setminus\\bigcup\_\{g\\in\\mathcal\{S\}\}V\(g\)
20:

𝒮=𝒮∪Graph\-Split​\(G​\[Vu​n​a​s​s​i​g​n​e​d\]\)\\mathcal\{S\}=\\mathcal\{S\}\\cup\\text\{Graph\-Split\}\(G\[V\_\{unassigned\}\]\)
21:Sort

𝒮\\mathcal\{S\}by size descending:

\{𝒮l​a​r​g​e,𝒮s​m​a​l​l\}\\\{\\mathcal\{S\}\_\{large\},\\mathcal\{S\}\_\{small\}\\\}
22:for

gs​m​a​l​l∈𝒮s​m​a​l​lg\_\{small\}\\in\\mathcal\{S\}\_\{small\}do

23:Perform

hh\-hop BFS expansion in

GG:

Ve​x​p​a​n​d​e​d=BFS​\(V​\(gs​m​a​l​l\),h\)V\_\{expanded\}=\\text\{BFS\}\(V\(g\_\{small\}\),h\)
24:if

Ve​x​p​a​n​d​e​d∩V​\(gl​a​r​g​e\)≠∅V\_\{expanded\}\\cap V\(g\_\{large\}\)\\neq\\emptysetthen

25:Merge

V​\(gs​m​a​l​l\)V\(g\_\{small\}\)into the first intersecting

V​\(gl​a​r​g​e\)V\(g\_\{large\}\)
26:endif

27:endfor

28:4\. Token encoding:

29:Construct sparse matrix

𝐌∈ℝ\|𝒮\|×\|V\|\\mathbf\{M\}\\in\\mathbb\{R\}^\{\|\\mathcal\{S\}\|\\times\|V\|\}
30:

Si​j=\{1/\|V​\(𝒮i\)\|if​vj∈𝒮i0otherwise\\text\{S\}\_\{ij\}=\\begin\{cases\}1/\|V\(\\mathcal\{S\}\_\{i\}\)\|&\\text\{if \}v\_\{j\}\\in\\mathcal\{S\}\_\{i\}\\\\ 0&\\text\{otherwise\}\\end\{cases\}
31:Output:Sparse segmentation matrix

𝐒\\mathbf\{S\}

### 3\.2Model Architecture

Upon the generation of the sparse segmentation matrix𝐒∈ℝS×N\\mathbf\{S\}\\in\\mathbb\{R\}^\{S\\times N\}, the model integrates these precomputed geometry tokens into a deep learning framework to process the input point cloud\. The input mesh, represented as a point cloud ofNNpoints with spatial coordinates inℝN×3\\mathbb\{R\}^\{N\\times 3\}, is first projected into a high\-dimensional feature space𝐗∈ℝN×C\\mathbf\{X\}\\in\\mathbb\{R\}^\{N\\times C\}\. These features are then aggregated into the token space by computing the matrix product of the segmentation matrix and the point features,𝐒𝐗∈ℝS×C\\mathbf\{SX\}\\in\\mathbb\{R\}^\{S\\times C\}\. This operation effectively down\-samples the dense point information into a compact latent token representation of the geometry\.

![Refer to caption](https://arxiv.org/html/2607.08025v1/x2.png)

Figure 2:PGD\-NO architecture and layer variants\. \(Top\) Model pipeline: a pre\-computed segmentation matrix extracts geometry tokens from the point cloud, which are refined by stacked layers to capture multi\-scale physics\. \(Bottom Left\) Layer configurations: Variants 1 and 2 use multi\-head attention to project features from tokens back to nodes, while Variant 3 uses a transposed segmentation matrix for high\-efficiency surface mapping\.The core of the architecture consists ofMMsuccessive Geometric Token Layers designed to refine the interaction between global geometric tokens and local point features\. As shown in the model architecture diagram in Figure[2](https://arxiv.org/html/2607.08025#S3.F2), the aggregated geometry token features serve as a global context that is fed back into the nodal features with an Attention Layer, which enables the network to propagate information from semantically distinct regions back to individual points\. Following the iterative refinement within the Geometric Token Layers, the final high\-dimensional point features are passed through a concluding MLP to produce the target predictions\.

The proposed framework introduces three architectural alternatives for the geometric token layer, varying in how they reconstruct point\-wise features from the latent space\. In the first variant, a multi\-head cross\-attention mechanism directly maps token features back to point features by utilizing the input point cloud as the Query \(QQ\) and the geometry tokens as the Key \(KK\) and Value \(VV\)\. The second variant introduces an additional Multi\-Head Attention \(MHA\) refinement step on the token features prior to the cross\-attention mapping, allowing for more complex interactions within the latent representation\. The third variant offers a high\-efficiency alternative specifically for surface prediction by transposing the segmentation matrix to map features directly back to each mesh node\. However, since these geometry tokens are precomputed based on specific mesh nodes, this third architecture is restricted to surface tasks and cannot be extended to volume node prediction\.

### 3\.3Model Complexity Analysis

To establish a baseline for performance, we evaluate the computational and memory complexity of our proposed architecture against Transolver\+\+, which represents the best state\-of\-the\-art neural PDE solver framework in terms of efficiency\. Our analysis considers a 3D PDE problem defined byNsN\_\{s\}surface nodes, which characterize the geometry, andNvN\_\{v\}volume nodes, representing the spatial coordinates where the Partial Differential Equation \(PDE\) solutions are computed\.

The system’s computational complexity is primarily governed by operations performed on volume node features\. While our model maintains the same order of magnitude as Transolver\+\+, it achieves a 50% reduction in computational overhead \(O​\(4​Nv\)O\(4N\_\{v\}\)versusO​\(8​Nv\)O\(8N\_\{v\}\)\) by streamlining the attention mechanism\. Furthermore, by decoupling geometry extraction from solution querying, our framework significantly enhances memory efficiency\. This architecture allows volume nodes to be partitioned across multiple GPUs, shifting the primary memory bottleneck to surface node operations \(O​\(4​Ns\)O\(4N\_\{s\}\)\), a substantial improvement over theO​\(8​Nv\)O\(8N\_\{v\}\)bottleneck inherent in Transolver\+\+\. More details about this analysis can be found in Appendix[A](https://arxiv.org/html/2607.08025#A1)\.

![Refer to caption](https://arxiv.org/html/2607.08025v1/x3.png)

Figure 3:Visualization of 3D industrial benchmarks\. \(Top\) Geometry tokens generated via our iterative decomposition algorithm for representative samples\. \(Bottom\) Corresponding ground\-truth physics fields\. These benchmarks cover a wide spectrum of complexities—from smooth surfaces to irregular, multi\-component geometries—and diverse prediction tasks, including internal volume, surface\-based, and external fluid domain patterns\.

## 4Experiment

Data: Given that two\-dimensional problems have been extensively explored in the literature of Neural PDE solvers, this study focuses exclusively on assessing model performance within the context of three\-dimensional, real\-world industrial applications\. The evaluation is conducted across the following five benchmarks: Heat Sink\(Zhong,[2025](https://arxiv.org/html/2607.08025#bib.bib60)\), JEB\(Honget al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib12)\), DrivAerNet\+\+\(Elrefaieet al\.,[2024b](https://arxiv.org/html/2607.08025#bib.bib11)\), Aircraft\(Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\), and CFD\-VOL proposed by us\. The datasets used in this study encompass a diverse range of surface geometries, each presenting unique structural challenges for the ML model\. The heat sink dataset serves as a baseline case, consisting of geometries with sharp edges formed by combinations of 3D rectangular volumes\. This dataset verifies the model performance when clear geometry tokens are present, as the algorithm can easily decompose these surfaces into flat facets\. In contrast, the JEB dataset consists of non\-parametric geometries created by generative models\. These organic shapes pose a significant challenge for the model in extracting representative geometry tokens\. The aircraft dataset introduces the challenge of representing smooth surfaces that lack distinct sharp edges\. Finally, the DrivAer\+\+ dataset includes vehicle models with a high number of individual components, serving as a benchmark for the model to handle complex, multi\-component assemblies\.

Prediction tasks also vary across these datasets, presenting different field\-specific challenges\. The heat sink and JEB datasets require the model to predict volumetric data, whereas the DrivAer\+\+ dataset focuses on surface\-based predictions\. The aircraft dataset similarly involves surface prediction but adds the complexity of handling parametric controls within the physics simulation\. Finally, the proposed CFD\-VOL dataset presents the unique challenge of predicting physics features in the external domain surrounding the geometry\. This requires the model to accurately capture fluid patterns and field interactions outside the physical boundaries of the object\.

Our geometry decomposition algorithm is applied exclusively to the manifold surface\. We keep all the extracted tokens of heat sink dataset and the largestMM=128 tokens for all the other datasets\. Empirical benchmarks across all datasets indicate that the pre\-processing time remains under 15 seconds per sample with CPU parallel computation\. This negligible overhead ensures that the deterministic encoding phase does not become a bottleneck, maintaining the overall efficiency of the PGD\-NO pipeline for rapid industrial design iterations\. The visualization of the geometry tokens and predicted physics features are shown in Figure[3](https://arxiv.org/html/2607.08025#S3.F3), and more extracted geometry tokens can be visualized in Appendix[D](https://arxiv.org/html/2607.08025#A4)

Baseline: We used the following model architectures for comparison: PointNet\(Kashefi and Mukerji,[2022](https://arxiv.org/html/2607.08025#bib.bib13)\), MeshGraphNet\(Pfaffet al\.,[2021](https://arxiv.org/html/2607.08025#bib.bib68)\), Graph Neural Operator\(Anandkumaret al\.,[2020](https://arxiv.org/html/2607.08025#bib.bib8)\), GALERKIN\(Cao,[2021](https://arxiv.org/html/2607.08025#bib.bib24)\), Geometry\-informed Neural Operator\(Liet al\.,[2023b](https://arxiv.org/html/2607.08025#bib.bib32)\), General Neural Operator Transformer\(Haoet al\.,[2023](https://arxiv.org/html/2607.08025#bib.bib1)\), Transolver\(Wuet al\.,[2024](https://arxiv.org/html/2607.08025#bib.bib53)\), Transolver\+\+\(Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\), and AB\-UPT\(Alkinet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib34)\)\. The details of the hyper\-parameter choice of our baseline methods are documented in Appendix[E](https://arxiv.org/html/2607.08025#A5)\. To accommodate memory\-constrained baselines, the mesh is decomposed into sub\-domains for independent inference, then concatenated to reconstruct the global prediction, following the implementations of\(Luoet al\.,[2025](https://arxiv.org/html/2607.08025#bib.bib62)\)\.

Table 1:Comparative analysis of relative error \(%\) across 3D industrial benchmarks\. Bold values denote the best performance\. ”\-” represents cases where the mesh resolution exceeds the model’s memory capacity or the architecture does not support the specific prediction task\. In addition to global field errors, we report the prediction accuracy for critical physical measurements \(Quantities of Interest\) specific to each engineering design problem\.Metric: To quantify the accuracy of our model predictions, we employ theL2L^\{2\}relative error as the primary metric\. This provides a holistic assessment of the model’s performance over the entire spatial domain\. However, global error metrics can sometimes obscure localized physical phenomena that are critical for engineering applications\. Therefore, we supplement our analysis by reporting the prediction accuracy for specific critical quantities of interest \(QoIs\) tailored to each design problem\. More details of the metric are clarified in Appendix[F](https://arxiv.org/html/2607.08025#A6)\.

### 4\.1Main Results

The experimental results, summarized in Table[1](https://arxiv.org/html/2607.08025#S4.T1), demonstrate that the proposed PGD\-NO model consistently outperforms or achieves a similar level of accuracy as established baselines across all five 3D industrial benchmarks\. Our architecture achieves the lowest relative errors in both volumetric field predictions and critical global scalar metrics\. The performance improvement of PGD\-NO over the second\-best baseline is most significant on the heat sink dataset\. This indicates that when the algorithm can perfectly decompose the geometry into representative tokens, the model is able to learn the underlying partial differential equations \(PDEs\) much more effectively\. Our results on the JEB dataset also indicate that the algorithm is capable of handling distributionally diverse data\. However, the improvement over other models in this case is more limited, which suggests a performance plateau when working with such irregular geometries\.

Comparing PGD\-NO\-v1 and v2 reveals that adding an attention layer to geometry tokens primarily benefits datasets like JEB, where complex geometries are harder to partition precisely; otherwise, the extra complexity may introduce noise and degrade performance\. Conversely, PGD\-NO\-v3 hierarchical mapping—which omits long\-range dependencies—yields the best results on the Aircraft dataset, suggesting long\-range feature relationship are less effective than local relationship for the simple and smooth surfaces\. Finally, PGD\-NO strong performance on the CFD\-VOL dataset demonstrates that surface\-derived geometry tokens sufficiently capture the latent representations required for high\-fidelity fluid simulations\. More results are documented in Appendix[C](https://arxiv.org/html/2607.08025#A3), including the computational and memory efficiency of different models and comparison with grid\-pooling tokens\. The error map comparison between our model and Transolver\+\+ are provided in Figure[4](https://arxiv.org/html/2607.08025#S4.F4), which is further clarified in Appendix[G](https://arxiv.org/html/2607.08025#A7)\.

![Refer to caption](https://arxiv.org/html/2607.08025v1/x4.png)

Figure 4:Error map comparison between Transolver\+\+ and PGD\-NO\. Geometric references on the left provide spatial context for the residuals\. For the Heat Sink and CFD\-VOL datasets, we visualize the direct physical fields due to their specific structural or external domain requirements\. For all other benchmarks, error maps are overlaid onto the extracted geometry tokens to demonstrate localized model accuracy across distinct structural regions\.![Refer to caption](https://arxiv.org/html/2607.08025v1/x5.png)

Figure 5:Interpretability analysis via multi\-layer nodal importance mapping\. \(Left\) Geometry reference showing the pre\-computed tokens and the specific query location \(red circle\)\. \(Right\) Corresponding nodal importance maps of different geometry token layers, with red arrows highlighting the regions of peak attention\.
### 4\.2Larger Meshes

To evaluate the scalability of our proposed model on large\-scale meshes, we increased the resolution of the CFD\-VOL dataset using random spatial sampling and 3D interpolation\. As shown in Table[2](https://arxiv.org/html/2607.08025#S4.T2), we then compared the performance of Transolver\+\+ and PGD\-NO\-v1 on this up\-sampled data\.

Table 2:Comparative relative error \(%\) on CFD\-VOL across varying mesh resolutions and spatial sampling ratios\.While Transolver\+\+ fails to process the full 60M mesh due to memory constraints, PGD\-NO\-v1 maintains high accuracy even at this scale\. To address the challenges of huge dense point clouds, we also utilize a spatial random sampling strategy during training\. By computing the loss gradient on a subset of points rather than the entire mesh, we significantly accelerate the training process and reduce peak memory requirements\. Our results demonstrate that, Compared with Transolver\+\+, this stochastic sampling approach does not significantly degrade the gradient quality of our model, as the model continues to deliver high prediction accuracy\. Notably, as the total point cloud size increases, the performance gap caused by random sampling is further reduced\.

### 4\.3Ablation Studies

Ablation studies on the CFD\-VOL dataset \(Table[3](https://arxiv.org/html/2607.08025#S4.T3)\) compare the depth sensitivity of Transolver\+\+ and PGD\-NO\-v1\. While Transolver\+\+ shows low sensitivity to increasing layers—likely because its adaptive physics tokens capture complex geometry tokens efficiently in shallower architectures\. In contrast, the performance of our model decreases significantly if the number of Geometry Token Layers is insufficient\. However, as the number of layers increases, our model demonstrates a superior ability to capture complex geometric features, leading to a consistent and significant enhancement in prediction accuracy\. This suggests that the iterative refinement of geometry tokens is vital for improving our model performance\.

Table 3:ComparativeL2L\_\{2\}relative error \(%\) on CFD\-VOL across varying model depths\.We conduct an additional ablation study of number of tokens using the JEB dataset, as it contains the most complex and organic geometries in our benchmark\. As shown in Table[4](https://arxiv.org/html/2607.08025#S4.T4), while performance degrades significantly with an insufficient number of tokens, marginal gains diminish as the token count scales up, which indicates that our proposed method can provide stable performance as sufficient geometry tokens are extracted\.

Table 4:ComparativeL2L\_\{2\}relative error \(%\) on JEB across varying number of extracted geometry tokens\.
### 4\.4Model Explanation

Our model produces granular, node\-wise importance maps by redistributing cross\-attention weights from the Geometric Token Layers to their constituent nodes—a diagnostic capability absent in existing baseline models\. In Figure[5](https://arxiv.org/html/2607.08025#S4.F5), we visualize the nodal attention score distribution relative to the predicted location of maximum stress, with the peak attention region highlighted by a red arrow\. The visualization reveals several key behaviors of the model\. First, the model focuses on localized tokens near the predicted region before attending to larger, distant tokens\. Generally, nodes on the left side of the bracket, which are closer to the region of interest, receive higher attention scores than those on the right\. Notably, the model prioritizes the constraint region in Layer 2 and the loading region in Layer 6\. This aligns with physical intuition, as these areas define the boundary conditions that govern the stress distribution\. Each layer targets distinct spatial locations, indicating that the model successfully integrates multi\-scale geometric details to inform its final prediction\. These high\-attention regions can ultimately serve as a design heuristic, guiding engineers toward critical structural areas that require optimization\.

## 5Conclusion

We introduce the Pre\-computed Geometry Decomposition Neural Operator \(PGD\-NO\), a paradigm that mitigates GPU memory bottlenecks in high\-fidelity PDE simulations by shifting geometric encoding to a deterministic, pre\-computed step\. By extracting ”geometry tokens” that effectively represent latent physical features, our model achieves linear scalability and handles massive meshes—up to 100 million nodes without inter\-GPU communication overhead\. Our results across five diverse industrial benchmarks demonstrate that PGD\-NO can achieve similar accuracy as state\-of\-the\-art baselines and provides enhanced interpretability\.

Limitations: While PGD\-NO matches state\-of\-the\-art accuracy and offers enhanced interpretability via attention\-based importance scores, combining our structural decomposition with local learnable refinements can be further investigated to better represent irregular, smooth topologies while maintaining the efficiency of decoupled volume\-surface querying\. Additionally, integrating recent advanced point cloud sampling techniques into our framework could further boost computational efficiency\. Finally, because PGD\-NO relies on precomputed geometry tokens, it exhibits robust invariance to underlying geometry representation\. This decoupled nature makes our approach a promising candidate for building scientific foundation models capable of handling diverse data sources with varying geometric design principles and meshing formats—an avenue we intend to investigate thoroughly in the future\.

## Impact Statement

This paper presents work aimed at advancing the efficiency of neural operators for 3D physics deep learning\. Our method enables large\-scale physical simulations on complex industrial geometries that were previously computationally prohibitive, offering a generalizable surrogate modeling framework applicable to most current industry\-scale modeling problems\. Beyond accelerating industrial design and manufacturing, our focus on 3D geometry learning provides valuable insights for research into the development of geometry generation models and other frameworks centered on 3D geometries\.

## References

- B\. Alkin, M\. Bleeker, R\. Kurle, T\. Kronlachner, R\. Sonnleitner, M\. Dorfer, and J\. Brandstetter \(2025\)AB\-UPT: scaling neural CFD surrogates for high\- fidelity automotive aerodynamics simulations via anchored\- branched universal physics transformers\.Transactions on Machine Learning Research\.Note:External Links:ISSN 2835\-8856,[Link](https://openreview.net/forum?id=nwQ8nitlTZ)Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- B\. Alkin, A\. Fürst, S\. Schmid, L\. Gruber, M\. Holzleitner, and J\. Brandstetter \(2024\)Universal physics transformers: a framework for efficiently scaling neural operators\.Advances in Neural Information Processing Systems37,pp\. 25152–25194\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1)\.
- A\. Anandkumar, K\. Azizzadenesheli, K\. Bhattacharya, N\. Kovachki, Z\. Li, B\. Liu, and A\. Stuart \(2020\)Neural operator: graph kernel network for partial differential equations\.InICLR 2020 workshop on integration of deep neural models and differential equations,Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- S\. Bock and M\. Weiß \(2019\)A proof of local convergence for the adam optimizer\.In2019 international joint conference on neural networks \(IJCNN\),pp\. 1–8\.Cited by:[Appendix E](https://arxiv.org/html/2607.08025#A5.p3.2)\.
- S\. Cao \(2021\)Choose a transformer: fourier or galerkin\.Advances in neural information processing systems34,pp\. 24924–24940\.Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- M\. Eliasof, E\. Haber, and E\. Treister \(2021\)Pde\-gcn: novel architectures for graph neural networks motivated by partial differential equations\.Advances in neural information processing systems34,pp\. 3836–3849\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1)\.
- M\. Elrefaie, A\. Dai, and F\. Ahmed \(2024a\)Drivaernet: a parametric car dataset for data\-driven aerodynamic design and graph\-based drag prediction\.InInternational Design Engineering Technical Conferences and Computers and Information in Engineering Conference,Vol\.88360,pp\. V03AT03A019\.Cited by:[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1)\.
- M\. Elrefaie, F\. Morar, A\. Dai, and F\. Ahmed \(2024b\)Drivaernet\+\+: a large\-scale multimodal car dataset with computational fluid dynamics simulations and deep learning benchmarks\.Advances in Neural Information Processing Systems37,pp\. 499–536\.Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p1.1)\.
- R\. J\. Gladstone, H\. Rahmani, V\. Suryakumar, H\. Meidani, M\. D’Elia, and A\. Zareei \(2024\)Mesh\-based gnn surrogates for time\-independent pdes\.Scientific reports14\(1\),pp\. 3394\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1)\.
- Z\. Hao, Z\. Wang, H\. Su, C\. Ying, Y\. Dong, S\. Liu, Z\. Cheng, J\. Song, and J\. Zhu \(2023\)Gnot: a general neural operator transformer for operator learning\.InInternational Conference on Machine Learning,pp\. 12556–12569\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- J\. He, S\. Kushwaha, J\. Park, S\. Koric, D\. Abueidda, and I\. Jasiuk \(2024\)Sequential deep operator networks \(s\-deeponet\) for predicting full\-field solutions under time\-dependent loads\.Engineering Applications of Artificial Intelligence127,pp\. 107258\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- T\. Hisada \(1995\)Recent progress in nonlinear fem\-based sensitivity analysis\.JSME international journal\. Ser\. A, Mechanics and material engineering38\(3\),pp\. 301–310\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1)\.
- S\. Hong, Y\. Kwon, D\. Shin, J\. Park, and N\. Kang \(2025\)Deepjeb: 3d deep learning\-based synthetic jet engine bracket dataset\.Journal of Mechanical Design147\(4\)\.Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p1.1)\.
- V\. Jagota, A\. P\. S\. Sethi, and K\. Kumar \(2013\)Finite element method: an overview\.Walailak Journal of Science and Technology \(WJST\)10\(1\),pp\. 1–8\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1)\.
- C\. Jermann, G\. Trombettoni, B\. Neveu, and P\. Mathis \(2006\)Decomposition of geometric constraint systems: a survey\.International Journal of Computational Geometry & Applications16\(05n06\),pp\. 379–414\.Cited by:[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1)\.
- A\. Kashefi and T\. Mukerji \(2022\)Physics\-informed pointnet: a deep learning solver for steady\-state incompressible flows and thermal fields on multiple sets of irregular geometries\.Journal of Computational Physics468,pp\. 111510\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- S\. Lee and T\. Oh \(2024\)Inducing point operator transformer: a flexible and scalable architecture for solving pdes\.pp\. 153–161\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1)\.
- B\. Lévy \(2006\)Laplace\-beltrami eigenfunctions towards an algorithm that” understands” geometry\.pp\. 13–13\.Cited by:[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1)\.
- K\. Li and W\. Ye \(2025\)D\-fno: a decomposed fourier neural operator for large\-scale parametric partial differential equations\.Computer Methods in Applied Mechanics and Engineering436,pp\. 117732\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1)\.
- Z\. Li, D\. Shu, and A\. Barati Farimani \(2023a\)Scalable transformer for pde surrogate modeling\.Advances in Neural Information Processing Systems36,pp\. 28010–28039\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1)\.
- Z\. Li, N\. Kovachki, K\. Azizzadenesheli, B\. Liu, A\. Stuart, K\. Bhattacharya, and A\. Anandkumar \(2020\)Multipole graph neural operator for parametric partial differential equations\.Advances in Neural Information Processing Systems33,pp\. 6755–6766\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1)\.
- Z\. Li, N\. B\. Kovachki, and K\. Azizzadenesheli \(2021\)Burigede liu, kaushik bhattacharya, andrew stuart, and anima anandkumar\. fourier neural operator for parametric partial differential equations\.InInternational Conference on Learning Representations,Vol\.2,pp\. 8\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1),[§1](https://arxiv.org/html/2607.08025#S1.p2.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1)\.
- Z\. Li, N\. Kovachki, C\. Choy, B\. Li, J\. Kossaifi, S\. Otta, M\. A\. Nabian, M\. Stadler, C\. Hundt, K\. Azizzadenesheli,et al\.\(2023b\)Geometry\-informed neural operator for large\-scale 3d pdes\.Advances in Neural Information Processing Systems36,pp\. 35836–35854\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- Z\. Li, H\. Zheng, N\. Kovachki, D\. Jin, H\. Chen, B\. Liu, K\. Azizzadenesheli, and A\. Anandkumar \(2024\)Physics\-informed neural operator for learning partial differential equations\.ACM/JMS Journal of Data Science1\(3\),pp\. 1–27\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1)\.
- J\. Lien and N\. M\. Amato \(2008\)Approximate convex decomposition of polyhedra and its applications\.Computer Aided Geometric Design25\(7\),pp\. 503–522\.Cited by:[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1)\.
- N\. Liu, S\. Jafarzadeh, and Y\. Yu \(2023\)Domain agnostic fourier neural operators\.Advances in Neural Information Processing Systems36,pp\. 47438–47450\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1)\.
- S\. N\. Lone, S\. De, and R\. Nayek \(2026\)α\\alpha\-VI deeponet: a prior\-robust variational bayesian approach for enhancing deeponets with uncertainty quantification\.Computer Methods in Applied Mechanics and Engineering449,pp\. 118552\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- L\. Lu, P\. Jin, G\. Pang, Z\. Zhang, and G\. E\. Karniadakis \(2021a\)Learning nonlinear operators via deeponet based on the universal approximation theorem of operators\.Nature machine intelligence3\(3\),pp\. 218–229\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- L\. Lu, P\. Jin, G\. Pang, Z\. Zhang, and G\. E\. Karniadakis \(2021b\)Learning nonlinear operators via deeponet based on the universal approximation theorem of operators\.Nature machine intelligence3\(3\),pp\. 218–229\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1),[§1](https://arxiv.org/html/2607.08025#S1.p2.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- L\. Lu, X\. Meng, S\. Cai, Z\. Mao, S\. Goswami, Z\. Zhang, and G\. E\. Karniadakis \(2022\)A comprehensive and fair comparison of two neural operators \(with practical extensions\) based on fair data\.Computer Methods in Applied Mechanics and Engineering393,pp\. 114778\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p3.1)\.
- H\. Luo, H\. Wu, H\. Zhou, L\. Xing, Y\. Di, J\. Wang, and M\. Long \(2025\)Transolver\+\+: an accurate neural solver for PDEs on million\-scale geometries\.pp\. 41432–41449\.External Links:[Link](https://proceedings.mlr.press/v267/luo25o.html)Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p2.1),[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1),[§4](https://arxiv.org/html/2607.08025#S4.p1.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- K\. Mamou and F\. Ghorbel \(2009\)A simple and efficient approach for 3d mesh approximate convex decomposition\.pp\. 3501–3504\.Cited by:[§2\.2](https://arxiv.org/html/2607.08025#S2.SS2.p1.1)\.
- L\. Mandl, S\. Goswami, L\. Lambers, and T\. Ricken \(2025\)Separable physics\-informed deeponet: breaking the curse of dimensionality in physics\-informed machine learning\.Computer Methods in Applied Mechanics and Engineering434,pp\. 117586\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- O\. M\. Morrison, F\. Pichi, and J\. S\. Hesthaven \(2024\)GFN: a graph feedforward network for resolution\-invariant reduced operator learning in multifidelity applications\.Computer Methods in Applied Mechanics and Engineering432,pp\. 117458\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1)\.
- F\. Moukalled, L\. Mangani, and M\. Darwish \(2015\)The finite volume method\.InThe finite volume method in computational fluid dynamics: An advanced introduction with OpenFOAM® and Matlab,pp\. 103–135\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1)\.
- J\. Park and N\. Kang \(2026\)Point\-deeponet: predicting nonlinear fields on non\-parametric geometries under variable load conditions\.Neural Networks,pp\. 108560\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- T\. Pfaff, M\. Fortunato, A\. Sanchez\-Gonzalez, and P\. W\. Battaglia \(2021\)Learning mesh\-based simulation with graph networks\.Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- Y\. Qiu, N\. Bridges, and P\. Chen \(2024\)Derivative\-enhanced deep operator network\.Advances in Neural Information Processing Systems37,pp\. 20945–20981\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- A\. Sahebi, M\. Procaccini, and R\. Giorgi \(2025\)Hashgrid: an optimized architecture for accelerating graph computing on fpgas\.Future Generation Computer Systems162,pp\. 107497\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1)\.
- L\. Serrano, L\. Le Boudec, A\. Kassaï Koupaï, T\. X\. Wang, Y\. Yin, J\. Vittaut, and P\. Gallinari \(2023\)Operator learning with neural fields: tackling pdes on general geometries\.Advances in Neural Information Processing Systems36,pp\. 70581–70611\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1)\.
- C\. Szuszik, R\. Szatmári, G\. Pál, and F\. Kun \(2026\)From segmentation to shattering: structural transitions in the breakup of brittle rings\.Chaos, Solitons & Fractals202,pp\. 117573\.Cited by:[§3\.1](https://arxiv.org/html/2607.08025#S3.SS1.p3.8)\.
- M\. Velasco, K\. O’Hare, B\. Rychtenberg, and S\. Villar \(2024\)Graph neural networks and non\-commuting operators\.Advances in neural information processing systems37,pp\. 95662–95691\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p3.1)\.
- T\. Wang and C\. Wang \(2024\)Latent neural operator for solving forward and inverse pde problems\.Advances in Neural Information Processing Systems37,pp\. 33085–33107\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p2.1)\.
- Y\. Wei, Y\. C\. Huang, H\. Tang, N\. Sankaran, I\. Chadha, D\. Dai, O\. Oluwole, V\. Balan, and E\. Lee \(2023\)9\.3 nvlink\-c2c: a coherent off package chip\-to\-chip interconnect with 40gbps/pin single\-ended signaling\.In2023 IEEE International Solid\-State Circuits Conference \(ISSCC\),pp\. 160–162\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p4.1)\.
- H\. Wu, H\. Luo, H\. Wang, J\. Wang, and M\. Long \(2024\)Transolver: a fast transformer solver for PDEs on general geometries\.InProceedings of the 41st International Conference on Machine LearningProceedings of the AAAI Conference on Artificial IntelligenceProceedings of the 42nd International Conference on Machine Learning2009 16th IEEE international conference on image processing \(ICIP\)IEEE International Conference on Shape Modeling and Applications 2006 \(SMI’06\)International Conference on Learning Representations \(ICLR\),R\. Salakhutdinov, Z\. Kolter, K\. Heller, A\. Weller, N\. Oliver, J\. Scarlett, F\. Berkenkamp, A\. Singh, M\. Fazel, D\. Hsu, S\. Lacoste\-Julien, F\. Berkenkamp, T\. Maharaj, K\. Wagstaff, and J\. Zhu \(Eds\.\),Proceedings of Machine Learning ResearchProceedings of Machine Learning Research, Vol\.23538267,pp\. 53681–53705\.External Links:[Link](https://proceedings.mlr.press/v235/wu24r.html)Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1),[§1](https://arxiv.org/html/2607.08025#S1.p2.1),[§1](https://arxiv.org/html/2607.08025#S1.p3.1),[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1),[§4](https://arxiv.org/html/2607.08025#S4.p4.1)\.
- Y\. Yu, N\. Liu, F\. Lu, T\. Gao, S\. Jafarzadeh, and S\. A\. Silling \(2024\)Nonlocal attention operator: materializing hidden knowledge towards interpretable physics discovery\.Advances in Neural Information Processing Systems37,pp\. 113797–113822\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1)\.
- Z\. Zhang, C\. Moya, L\. Lu, G\. Lin, and H\. Schaeffer \(2025\)Deeponet as a multi\-operator extrapolation model: distributed pretraining with physics\-informed fine\-tuning\.Journal of Computational Physics,pp\. 114537\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- J\. Zheng, W\. Li, N\. Xu, J\. Zhu, and X\. Zhang \(2024\)Alias\-free mamba neural operator\.Advances in Neural Information Processing Systems37,pp\. 52962–52995\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p4.1)\.
- W\. Zhong and H\. Meidani \(2023\)Pi\-vae: physics\-informed variational auto\-encoder for stochastic differential equations\.Computer Methods in Applied Mechanics and Engineering403,pp\. 115664\.Cited by:[§1](https://arxiv.org/html/2607.08025#S1.p1.1)\.
- W\. Zhong and H\. Meidani \(2024\)Physics\-informed discretization\-independent deep compositional operator network\.Computer Methods in Applied Mechanics and Engineering431,pp\. 117274\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- W\. Zhong and H\. Meidani \(2025\)Physics\-informed geometry\-aware neural operator\.Computer Methods in Applied Mechanics and Engineering434,pp\. 117540\.Cited by:[§2\.1](https://arxiv.org/html/2607.08025#S2.SS1.p1.1)\.
- W\. Zhong \(2025\)Cited by:[§4](https://arxiv.org/html/2607.08025#S4.p1.1)\.

## Appendix AModel complexity analysis

In the Transolver\+\+ framework, each physics\-based attention layer requires approximately eightO​\(N\)O\(N\)operations per volume node\. Specifically, for every block, the model must perform aninput projectionof nodal features to the internal head dimension, followed bytwo separate linear mappingsto determine the Gumbel\-Softmax temperature and the slice membership weights\. The membership assignment itself necessitates aSoftmax operationacross the entire node set\. The core interaction involvestwo Einstein summation steps: one to aggregateNNnodes into a reduced set of geometry tokens, and another to redistribute the processed token features back to the originalNNcoordinates\. Finally, the layer concludes with anoutput linear projectionand a subsequentMLP passthat maps the hidden representation back to the nodal feature space\. In contrast, our first two proposed attention layer architectures, as illustrated in Figure[2](https://arxiv.org/html/2607.08025#S3.F2)\(B\), optimize this process by requiring only three primary operations: query projection, attention score computation, and the linear operation for final output in a Multi\-Head Attention layer\. This architectural effectively reduces the computational overhead by more than 50% while maintaining representative capacity\.

Regarding memory complexity, our model shifts the primary bottleneck from the volume nodes to the surface nodes\. While Transolver\+\+ operates on the entire point cloud holistically, resulting in a memory footprint dominated by the number of volume nodes \(O​\(8​Nv\)O\(8N\_\{v\}\)\), our framework decouples geometry feature extraction from solution querying\. Because volume nodes in our model serve strictly as query points for PDE evaluation and do not participate in surface feature encoding, the volume point cloud can be partitioned and processed independently\. In an extreme parallelization scenario, each volume node could theoretically be processed by a discrete unit\. Consequently, the memory bottleneck resides within the surface node operations of the Geometry Token Layer\. With a memory complexity ofO​\(3​Ns\)O\(3N\_\{s\}\), and utilizing hardware such as the NVIDIA GH200, our framework supports efficient parallel training for high\-fidelity meshes containing up to 80 million surface nodes\.

## Appendix BDataset description

The details of the physics simulation are clarified:

- •Heat Sink \(Thermal Management\):In the semiconductor industry, precise thermal management is essential to prevent hardware failure due to overheating\. This simulation models steady\-state heat dissipation from a CPU through a parameterized geometry\. The design is parameterized by𝐩∈ℝ4\\mathbf\{p\}\\in\\mathbb\{R\}^\{4\}\(fin height, thickness, spacing, and count\) mapped to a 10k\-node 3D coordinate mesh𝒳\\mathcal\{X\}\. The output is the steady\-state temperature fieldT​\(𝐱\)T\(\\mathbf\{x\}\), governed by the Poisson equation for heat conduction: −∇⋅\(κ​∇T\)=Q\-\\nabla\\cdot\(\\kappa\\nabla T\)=Qwhereκ\\kappais the thermal conductivity andQQis the heat source\. Boundary conditions specify forced convectionq=h​\(T−T∞\)q=h\(T\-T\_\{\\infty\}\)on the lateral surfaces and free convection on the top surface\.
- •JEB \(Aerospace Structural Engineering\):Reducing component weight while maintaining structural integrity is a core objective in aerospace manufacturing\. This dataset utilizes 2,138 freeform 3D bracket designs where the input is a diverse set of 100k\-scale meshesℳ\\mathcal\{M\}with fixed bolt\-hole constraints\. The output is the displacement vector field𝐮​\(𝐱\)\\mathbf\{u\}\(\\mathbf\{x\}\)and the stress distribution, derived from the linear elasticity equilibrium equation: ∇⋅σ\+𝐟=0\\nabla\\cdot\\sigma\+\\mathbf\{f\}=0whereσ\\sigmais the stress tensor and𝐟\\mathbf\{f\}represents the applied vertical loading of 3\.5 kN\. This setup tests the model’s ability to solve for structural stability under prescribed mechanical loads\.
- •DrivAer\+\+ \(Automotive Aerodynamics\):Minimizing aerodynamic drag is critical for automotive OEMs to meet fuel efficiency and emission standards\. The input consists of high\-fidelity surface meshes of realistic car designs, including complex features like wheels and underbodies\. The output is the surface pressureppand wall shear stressτw\\tau\_\{w\}fields, which are solutions to the Reynolds\-Averaged Navier\-Stokes \(RANS\) equations for turbulent flow: ρ​\(𝐮⋅∇\)​𝐮=−∇p\+μ​∇2𝐮\+∇⋅τR\\rho\(\\mathbf\{u\}\\cdot\\nabla\)\\mathbf\{u\}=\-\\nabla p\+\\mu\\nabla^\{2\}\\mathbf\{u\}\+\\nabla\\cdot\\tau^\{R\}whereτR\\tau^\{R\}is the Reynolds stress tensor\. Boundary conditions include a defined inlet velocity and no\-slip conditions on the vehicle surface\.
- •Aircraft \(Transonic Aerodynamics\):Aircraft design requires evaluating aerodynamic performance across varying flight conditions to ensure stability and efficiency\. The input features include a 300,000\-point surface mesh alongside flow conditions: Mach numberMM, angle of attackα\\alpha, and sideslip angleβ\\beta\. The output is a multi\-variable state vector including fluid densityρ\\rho, pressurepp, and velocity components\(u,v,w\)\(u,v,w\), governed by the compressible Navier\-Stokes equations: ∂ρ∂t\+∇⋅\(ρ​𝐮\)=0\\frac\{\\partial\\rho\}\{\\partial t\}\+\\nabla\\cdot\(\\rho\\mathbf\{u\}\)=0This benchmark captures the complex physics of transonic flow, specifically the formation and movement of shock waves on the airframe surfaces\.
- •CFD\-VOL \(Industrial Fluid Mechanics\):Full volumetric fluid analysis is required for comprehensive engineering tasks such as HVAC design and internal flow modeling\. The output of the CFD simulation of the fluid part is the 3D velocity field𝐮​\(𝐱\)\\mathbf\{u\}\(\\mathbf\{x\}\)on a 12M\-node\-scale mesh and pressure fieldp​\(𝐱\)p\(\\mathbf\{x\}\)throughout the entire volume, governed by the incompressible Navier\-Stokes and continuity equations: ρ​\(𝐮⋅∇\)​𝐮=−∇p\+μ​∇2𝐮\+𝐟,∇⋅𝐮=0\\rho\(\\mathbf\{u\}\\cdot\\nabla\)\\mathbf\{u\}=\-\\nabla p\+\\mu\\nabla^\{2\}\\mathbf\{u\}\+\\mathbf\{f\},\\quad\\nabla\\cdot\\mathbf\{u\}=0This setup focuses on the extreme scalability and local conservation required in large\-scale industrial fluid simulations\.

## Appendix CAdditional results

In addition to accuracy comparison, we provide a detailed comparison of training times and memory usage in Table[5](https://arxiv.org/html/2607.08025#A3.T5)\. Entries with ”\-” indicate memory requirements exceeding the a 4\-way GH200 capacity \(approx\. 360GB\)\.

Table 5:Comparison of Training Times and Memory Usage Across DatasetsTo demonstrate the importance of our geometry decomposition for model performance, we compare it against a baseline where geometry latent tokens are instead generated via mean\-pooling of the point cloud features\. Specifically, this baseline partitions the point cloud into uniform grid cells, and compute the average of the point features inside each cell and treat it as one geometry token\. Then we use the same decoder structure as our PGD\-NO\-v1 to generate predictions\. As shown in the Table[6](https://arxiv.org/html/2607.08025#A3.T6), this grid\-pooling approach consistently underperforms across all datasets in terms ofL2L\_\{2\}relative error when compared to our proposed geometry decomposition \(Surface token\)\.

Table 6:Comparison between surface tokens and grid\-pooling tokens
## Appendix DExtracted geometry token visualizations

![Refer to caption](https://arxiv.org/html/2607.08025v1/x6.png)

Figure 6:The extracted tokens of different samples in Heat Sink dataset are presented\.![Refer to caption](https://arxiv.org/html/2607.08025v1/x7.png)

Figure 7:The extracted tokens of different samples in JEB dataset are presented\.![Refer to caption](https://arxiv.org/html/2607.08025v1/x8.png)

\(a\)Top view![Refer to caption](https://arxiv.org/html/2607.08025v1/x9.png)

\(b\)Bottom viewFigure 8:The extracted tokens of different samples in JEB dataset are presented\.![Refer to caption](https://arxiv.org/html/2607.08025v1/x10.png)

\(a\)Front view![Refer to caption](https://arxiv.org/html/2607.08025v1/x11.png)

\(b\)Back viewFigure 9:The extracted tokens of different samples in DrivAerNet\+\+ dataset are presented\.![Refer to caption](https://arxiv.org/html/2607.08025v1/x12.png)

Figure 10:The bottom view of the extracted tokens of different samples in drivAerNet\+\+ dataset are presented\.
## Appendix EImplementation details and baseline comparison

To ensure a rigorous and fair comparison, all models are implemented within a unified experimental framework that standardizes architectural depth, hidden dimensions, and spatial resolution\. Based on a combination of original implementations and GPU memory constraints, we set the default depth to 8 layers for point\-based models and 4 layers for graph\-based models\. Hidden dimensions are assigned by model family: 128 for point\-based networks \(e\.g\., PointNet, Galerkin, GNOT, Transolver, Transolver\+\+, and PGD\-NO\), 32 for graph\-based operators \(e\.g\., GNO, MeshGraphNet\), and 16 for grid\-based models \(e\.g\., GINO\)\. GELU is employed as the default activation function across all architectures\. For point cloud inputs, graph connectivity is constructed using the KDTree algorithm, while grid\-based models utilize a maximum hidden graph resolution of 100 voxels to maintain sufficient density\.

To handle the extreme disparity in mesh scales—ranging from thousands to 2\.5 million points—we implement a scale\-aware adaptation strategy\. For million\-scale datasets such as DrivAerNet\+\+ and Aircraft, where most baselines encounter memory limitations, we adopt a consistent subsampling approach\. Inputs are reduced to 50k mesh points, with manifold structure preserved via K\-Nearest Neighbors \(KNN\) reconstruction\. During training, a random subsampling strategy is applied at the start of each epoch to increase data diversity while maintaining a constant step count\. For evaluation, we aggregate multiple overlapping subsampled predictions into a single, high\-fidelity reconstructed output\. This ensures that even memory\-constrained models can be evaluated on industry\-scale geometries without compromising structural integrity\.

To isolate architectural efficacy from optimization artifacts, all models are trained in a standardized environment using NVIDIA GH200 GPUs \(100GB\)\. We split each dataset into 60% training, 10% validation, and 30% testing sets\. Training is conducted for 200 epochs with a batch size of 1, utilizing the Adam optimizer\(Bock and Weiß,[2019](https://arxiv.org/html/2607.08025#bib.bib44)\)\(β1=0\.9,β2=0\.999\\beta\_\{1\}=0\.9,\\beta\_\{2\}=0\.999\) and a weight decay of10−410^\{\-4\}\. The base learning rate of 0\.001 is managed by a OneCycleLR scheduler to facilitate stable convergence\. In instances where a configuration exceeds available GPU memory, we reduce the hidden dimension to the largest size that fits the constraint\.

By unifying these training\-side variables, we ensure that the reportedL2L\_\{2\}relative errors and Quantities of Interest \(QoI\) reflect the intrinsic predictive power of the model architectures rather than variations in hyperparameter tuning\. Each training step computes gradients based on a single graph to maintain consistency across different geometric complexities\. This rigorous protocol allows us to highlight the performance gains specifically attributable to the structural design of PGD\-NO\.

## Appendix FMetric

To evaluate the predictive performance of our model across diverse physical domains, we employ both a global field reconstruction metric and task\-specific engineering measurements\. This dual\-layered evaluation ensures that the model captures the continuous solution space of the governing Partial Differential Equations \(PDEs\) while maintaining precision in the critical scalar values used for industrial design\.

The primary metric for quantifying prediction accuracy is theL2L^\{2\}relative error over the entire field\. Specifically, we predict temperature for the heat sink dataset, stress for the JEB dataset, pressure for the Aircraft and DrivAerNet\+\+ datasets, and velocity in the X\-direction for the CFD\-VOL dataset\. To validate the model for specific design tasks, we also evaluate several critical physical quantities\. For the heat sink dataset, we extract the temperature distributions for both the chip and the heat sink\. We specifically monitor the average chip temperature, as it serves as a direct indicator of heat retention and cooling efficiency\. For the JEB dataset, we focus on the accuracy of the maximum predicted stress to ensure the design remains within safe operating limits\. In aerodynamic applications, we evaluate the DrivAer dataset using the drag coefficient \(CdC\_\{d\}\) to measure wind resistance\. For the aircraft dataset, we calculate the lift coefficient \(ClC\_\{l\}\) to determine flight viability\. Finally, for the volumetric CFD data, a single scalar value is often insufficient to characterize the fluid pattern\. Consequently, we evaluate the fluid field predictions on the middle cross\-section \(vmidv\_\{\\text\{mid\}\}\), as this plane contains the most significant information regarding flow structures and boundary layer interactions\.

### F\.1Field prediction error

To assess the global accuracy of the predicted physical fields across the entire spatial domain, we utilize theRelativeL2L\_\{2\}Error\. This metric provides a normalized measure of the discrepancy between the model predictiony^\\hat\{y\}and the ground\-truth fieldyy, allowing for consistent comparison across different physical units and scales\. It is defined as:

Relative​L2=‖y^−y‖2‖y‖2=∑i=1n\(y^i−yi\)2∑i=1nyi2\.\\text\{Relative \}L\_\{2\}=\\frac\{\\\|\\hat\{y\}\-y\\\|\_\{2\}\}\{\\\|y\\\|\_\{2\}\}=\\frac\{\\sqrt\{\\sum\_\{i=1\}^\{n\}\(\\hat\{y\}\_\{i\}\-y\_\{i\}\)^\{2\}\}\}\{\\sqrt\{\\sum\_\{i=1\}^\{n\}y\_\{i\}^\{2\}\}\}\.\(2\)This error is computed individually for each state variable in the output vector\.

### F\.2Quantities of Interest \(QoI\)

Beyond field\-wide errors, we evaluate specificQuantities of Interest \(QoI\)that serve as the primary performance indicators in engineering design loops\. These scalars are often derived through the integration or extremum searching of the predicted fields:

- •Heat Sink \(Average Chip TemperatureT¯chip\\bar\{T\}\_\{\\text\{chip\}\}\):The critical thermal performance metric is the average temperature at the contact interface between the heat source and the cooling fins: T¯chip=1Abase​∫AbaseT​\(𝐱\)​𝑑A\\bar\{T\}\_\{\\text\{chip\}\}=\\frac\{1\}\{A\_\{\\text\{base\}\}\}\\int\_\{A\_\{\\text\{base\}\}\}T\(\\mathbf\{x\}\)dA\(3\)
- •JEB \(Maximum von Mises Stressσmax\\sigma\_\{\\text\{max\}\}\):For structural safety, we identify the peak stress concentration within the geometry to predict potential material yielding or structural failure: σmax=max𝐱∈ℳ⁡σv​m​\(𝐱\)\\sigma\_\{\\text\{max\}\}=\\max\_\{\\mathbf\{x\}\\in\\mathcal\{M\}\}\\sigma\_\{vm\}\(\\mathbf\{x\}\)\(4\)
- •DrivAer\+\+ \(Drag CoefficientCDC\_\{D\}\):Automotive efficiency is measured by the drag coefficient, which requires integrating the predicted surface pressureppand wall shear stressτw\\tau\_\{w\}in the direction of the flow: CD=2ρ​v2​A​∫A\(p​𝐧⋅𝐢\+τw​𝐭⋅𝐢\)​𝑑AC\_\{D\}=\\frac\{2\}\{\\rho v^\{2\}A\}\\int\_\{A\}\(p\\mathbf\{n\}\\cdot\\mathbf\{i\}\+\\tau\_\{w\}\\mathbf\{t\}\\cdot\\mathbf\{i\}\)dA\(5\)
- •Aircraft \(Lift CoefficientCLC\_\{L\}\):Aerodynamic performance for flight stability is determined by the lift coefficient, derived by integrating the pressure distribution over the wing surface along the vertical axis𝐤\\mathbf\{k\}: CL=2ρ​v2​A​∫A\(p​𝐧⋅𝐤\)​𝑑AC\_\{L\}=\\frac\{2\}\{\\rho v^\{2\}A\}\\int\_\{A\}\(p\\mathbf\{n\}\\cdot\\mathbf\{k\}\)dA\(6\)
- •CFD\-VOL \(Cross\-Cut Velocity ErrorErrcut\\text\{Err\}\_\{\\text\{cut\}\}\):For large\-scale volumetric flow, we evaluate the relativeL2L\_\{2\}error of the velocity vector field𝐮\\mathbf\{u\}along specific 2D planes to ensure accurate capture of internal wake structures: Errcut=‖𝐮^plane−𝐮plane‖2‖𝐮plane‖2\\text\{Err\}\_\{\\text\{cut\}\}=\\frac\{\\\|\\hat\{\\mathbf\{u\}\}\_\{\\text\{plane\}\}\-\\mathbf\{u\}\_\{\\text\{plane\}\}\\\|\_\{2\}\}\{\\\|\\mathbf\{u\}\_\{\\text\{plane\}\}\\\|\_\{2\}\}\(7\)

## Appendix GError map clarification

The visual analysis of the error distribution across diverse datasets highlights that the PGD\-NO framework consistently achieves lower and more localized error patterns compared to the Transolver\+\+ baseline\. In thermal simulations like the Heat Sink, Transolver\+\+ exhibits notable error clusters along the fin interfaces, whereas PGD\-NO maintains a nearly uniform low\-error profile across the entire geometry\. This trend extends to complex structural and aerodynamic cases; for instance, in the JEB and Aircraft benchmarks, Transolver\+\+ shows significant error concentrations in high\-curvature regions and leading edges, while PGD\-NO successfully mitigates these artifacts through its more effective geometric tokenization\. In large\-scale fluid dynamics simulations, such as the DrivAerNet\+\+ surface and the corresponding velocity cross\-sections, PGD\-NO demonstrates superior precision in capturing boundary layer interactions and wake turbulences\. By decoupling geometric feature extraction from point\-wise querying, the model ensures that the prediction error remains minimal even in high\-gradient areas, proving its robustness for high\-fidelity industrial PDE modeling\.

Similar Articles

NoPA: Non-Parametric Online 3D Scene Graph Generation

Hugging Face Daily Papers

NoPA introduces a non-parametric distribution-based approach for real-time 3D scene graph generation, preserving geometric details using kernel density estimates and particle-based object representation, substantially outperforming current methods.

@AnimaAnandkumar: This is something I have been emphasizing since we started our work on Neural Operators. We very quickly went from simp…

X AI KOLs Following

Anima Anandkumar highlights that neural operators, despite simple benchmarks, have achieved massive speedups (10,000–million times) in hard real-world problems like high-resolution AI weather modeling (FourCastNet) and nuclear fusion turbulence, referencing a new paper showing learned solvers become more cost-effective as PDE tasks get harder.

Sequential Physics-Constrained Neural Operator Forward Modeling for the $\textit{Norne}$ Reservoir System

arXiv cs.LG

This paper presents a comprehensive mathematical framework for sequential surrogate modeling of three-phase black-oil reservoir dynamics using Fourier Neural Operators (FNO) and physics-informed variants (PINO), applied to the Norne benchmark reservoir. Theoretical contributions include functional-analytic formulation, covariate shift analysis, physics-constrained spectral stability, and truncated backpropagation gradient analysis.