PiGRAND: Physics-informed Graph Neural Diffusion for Intelligent Additive Manufacturing

Benjamin Uhrich; Tim Häntschel; Erhard Rahm

cs.LGMar 16, 2026

PiGRAND: Physics-informed Graph Neural Diffusion for Intelligent Additive Manufacturing

Authors: Benjamin Uhrich, Tim Häntschel, Erhard Rahm

Organizations: Center for Scalable Data Analytics and Artificial Intelligence Dresden/Leipzig, Germany. · Leipzig University, Leipzig, Germany.

Abstract

A comprehensive understanding of heat transport is essential for optimizing various mechanical and engineering applications, including 3D printing. Recent advances in machine learning, combined with physics-based models, have enabled a powerful fusion of numerical methods and data-driven algorithms. This progress is driven by the availability of limited sensor data in various engineering and scientific domains, where the cost of data collection and the inaccessibility of certain measurements are high. To this end, we present PiGRAND, a Physics-informed graph neural diffusion framework. In order to reduce the computational complexity of graph learning, an efficient graph construction procedure was developed. Our approach is inspired by the explicit Euler and implicit Crank-Nicolson methods for modeling continuous heat transport, leveraging sub-learning models to secure the accurate diffusion across graph nodes. To enhance computational performance, our approach is combined with efficient transfer learning. We evaluate PiGRAND on thermal images from 3D printing, demonstrating significant improvements in prediction accuracy and computational performance compared to traditional graph neural diffusion (GRAND) and physics-informed neural networks (PINNs). These enhancements are attributed to the incorporation of physical principles derived from the theoretical study of partial differential equations (PDEs) into the learning model. The PiGRAND code is open-sourced on GitHub: https://github.com/bu32loxa/PiGRAND

Figures & tables

Figure 1: Graph Construction Algorithm - Step 1: Generating point cloud, Step 2: Pruning method, Step 3: Delaunay triangulation, Step 4: Alpha shape, Step 5: Simplicial 3-complex.

Figure 2: Flowchart - Graph construction and physics-informed graph neural diffusion

Figure 3: Geometry of Printed Objects - Photographs of high quality components, strongly deformed component P9 and CAD models

P	$h_{0}$ [mm]	$\alpha$ [°]	$l_{1}$ [mm]	$l_{0}$ [mm]	$h_{1}$ [mm]	Material
4	10	70	20	6.7	18.27	Stainless Steel
7	10	70	25	8.3	22.94	Stainless Steel
9	10	50	25	8.3	9.95	Stainless Steel
7M	10	70	25	8.3	22.94	Nickel-based Alloy

Table 1: Comparison of the measures of the 3D-printed objects P4, P7, P9 and P7M

Figure 4: Temporal-Spatial Graph - graph model development for layer 100, 250, 500 (rows). Left column: point cloud, middle column: Simplicial Complex, right column: Alpha Shape

Panel of Figure 5 — Figure 5: 4D-Heat Transport Prediction - Heat transport evolution for layer 100, 250, 350, 500. As time progresses, the component’s temperature increases, reaching a maximum at the upper portion and subsequently declining towards the base.

Panel of Figure 6 — Figure 6: Heat Transport Prediction and Error Plots – A comparison of the heat transport prediction for print layer 232 ( $t=7$ s) with a PINN, GRAND, PiGRAND and the real measurement data.

Figure 7: Relative Error - Comparison of

\epsilon_{r}

between GRAND, PiGRAND and PINN over the entire printing layer 232 for each timepoint.

	PINN	GRAND	PiGRAND
$\frac{1}{N_{T}}\sum_{k=0}^{N_{T}}\epsilon_{r}$	0.041	0.003	0.001

Table 2: Comparison of the mean value of

\epsilon_{r}

between PINN, GRAND and PiGRAND, where

N_{T}

is the number of time points during printing layer 232.

Panel of Figure 8 — Figure 8: Comparison of explicit Euler and implicit Crank-Nicolson inspired graph diffusion network - Left top: Comparison of $\epsilon_{r}$ for 500 print layers, Right top: Comparison of $\mathcal{L}_{\text{Energy}}$ for 500 print layers. Left bottom: Max temperatures for print layers 3, 4, 5.

	$\frac{1}{N_{L}}\sum_{k=0}^{N_{L}}\epsilon_{r}$	$\frac{1}{N_{L}}\sum_{k=0}^{N_{L}}\hat{\mathcal{L}}_{\text{Energy}}$
Euler	0.010	24.473
Crank-Nicolson	0.006	23.449

Table 3: Comparison of the mean of

\epsilon_{r}

and

\mathcal{L}_{Energy}

between Euler and Crank-Nicolson inspired Graph Neural Diffusion, where

N_{L}

is the number of print layers.

regularization	weight	$\mathcal{L}_{\text{data}}$	$\mathcal{L}_{\varphi}$	$\mathcal{L}_{\psi}$	$\mathcal{L}_{\text{energy}}$
	high	1.085	1.134	0.375	0.000
all	normal	1.128	1.211	0.175	0.076
	low	1.111	1.206	0.231	0.588
	high	1.107	1.205	0.238	0.367
$\mathcal{L}_{\varphi},\,\mathcal{L}_{\psi},\,\mathcal{L}_{\text{heat}}$	normal	1.109	1.205	0.238	0.335
	low	1.108	1.205	0.269	0.667

Table 4: Model fit for the foundation model using different sets of weighted regularization functions for training. ’High weights’ uses a factor of

10

for the regularization terms in the total loss function, ’low weights’ uses a factor of

0.1

regularization	weight	$\mathcal{L}_{\text{data}}$	$\mathcal{L}_{\varphi}$	$\mathcal{L}_{\psi}$	$\mathcal{L}_{\text{energy}}$
	high	1.067	1.021	0.133	0.000
all	normal	1.065	0.979	0.127	0.008
	low	1.063	0.981	0.114	0.291
	high	1.063	0.979	0.112	0.029
$\mathcal{L}_{\varphi},\,\mathcal{L}_{\psi},\,\mathcal{L}_{\text{heat}}$	normal	1.063	0.979	0.108	0.024
	low	1.063	0.981	0.104	0.283

Table 5: Model fit for P4 using different sets of weighted regularization functions for training. ’High weights’ uses a factor of

10

for the regularization terms in the total loss function, ’low weights’ uses a factor of

0.1

Panel of Figure 9 — Figure 9: Inference and transfer learning - Left: Inference for pyramid P4, print layer 380, Right: Inference for pyramid P9, print layer 200, Bottom: Transfer learned heat transport for pyramid P7M, print layer 232.

Figure 10: A comparison of the P7 foundation model inspired by Crank-Nicolson, the P4 and P9 inference and the P7M transfer-learned model. - Left: Comparison of

\epsilon_{r}

for 500 print layers, Right: Comparison of

\mathcal{L}_{\text{Energy}}

for 500 print layers.

Panel of Figure 11 — Figure 11: Performance of PiGRAND - Left: The total solving time of the PiGRAND foundation model is compared with the inference times of the other printed components P4, P9 and the transfer-learned temperature distribution of P7M, Right: The average computational effort associated with the prediction of the temperature distribution of a single printed layer, contrasting the performance of a transfer-learned PINN with our proposed PiGRAND, Bottom: The total solving time of PiGRAND (Crank-Nicolson) compared to a FVM solver and a vanilla PINN for one printing benchmark layer

Physics-Informed GNNs

Physics-informed graph neural networks encode equations, conservation laws, and boundary conditions into graph models for scientific prediction and simulation.

See all 40 papers

Oct 12, 2026

pi-Net: Empirical Latent Representations for Differentiable Physics Heads in Data-Scarce Systems

Developing reliable predictive models for multi-physics systems often forces a compromise between computationally demanding physical simulators and data-hungry, opaque machine learning (ML) algorithms. While incorporating dimensional analysis into directed computation graphs offers a structurally sound alternative to explicitly solving partial differential equations (PDEs), such formulations typically yield underdetermined systems that act as imperfect simulators. We introduce pi-Net, a physics-informed latent variable model that resolves these imperfect simulators. The hybrid pi-Net architecture consists of an encoder that maps inputs into physically normalized latent space, which are then decoded into final predictions by a differentiable physics head. Evaluated on a real-world, data-scarce Gas Metal Arc Welding (GMAW) case study, the framework achieves an R2 of 0.864 with only 60 experimental samples. The results demonstrate that pi-Net significantly outperforms purely data-driven baselines in data-scarce regimes while providing causal, process-level explainability.

Akshay Dhalpe, Arnab Chattopadhyay, Shahriar Bakrani Balani +2

Tampere University · University of Helsinki

Physics-Informed ML Surrogate Modeling Latent Variable Models Data-Efficient Learning Manufacturing

Ask PDF

Jun 6, 2026

Overcoming the Limits of Finite Difference Method; Physics-Informed Neural Network for Noisy High-Dimensional Heat Diffusion

High-dimensional transient heat diffusion under noisy boundary conditions exposes a fundamental limitation of classical numerical methods: accuracy degrades catastrophically where physical noise is unavoidable. This paper presents a Physics-Informed Neural Network (PINN) framework as a systematic solution to this problem across one, two, and three spatial dimensions, establishing clear operational regimes that redefine solver selection in noisy thermal systems. Under 20% boundary noise in 3D, PINN sustains approximately 91% accuracy while Finite Difference Method (FDM) collapses to 36%, a clear decisive advantage. This is further confirmed in a physical copper thermal system, where PINN reduces boundary reconstruction error by 3.3 times under realistic noise conditions. This noise resilience is accompanied by a dimensionality-driven efficiency crossover: PINN requires fewer spacetime nodes than FDM in 3D while achieving superior accuracy, exposing the true cost of classical discretization at scale. These findings reframe solver selection: the decisive axis is not accuracy alone, but noise exposure and dimensionality jointly. When noise and dimensionality are both high, the classical solver paradigm is insufficient; this work provides the foundation to justify PINN as the operational standard in such regimes.

Shreesh Bhattarai, Harish Chandra Bhandari

Department of Mathematics, Kathmandu University Dhulikhel, Nepal

PDE Solving Physics-Informed ML

Ask PDF

May 20, 2026

Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment

Physics-informed diffusion models typically enforce PDE constraints only on final outputs, leaving intermediate representations unconstrained and prone to shortcut learning under shifted boundary conditions. We introduce REPA-P, a teacher-free, architecture-agnostic framework that aligns intermediate features with physical states using first-principles residuals. REPA-P attaches lightweight

1{\times}1

projection heads to selected layers, decodes hidden activations into physical quantities, and applies PDE residual losses during training. These heads are discarded at inference, introducing zero overhead. Across four PDE tasks, including Darcy flow, topology optimization, electrostatic potential, and turbulent channel flow, REPA-P accelerates convergence by up to

2{\times}

, reduces physics residuals by up to

66.4\%

, and improves out-of-distribution robustness by up to

49.3\%

, with consistent gains on both U-Net and Diffusion Transformer backbones. Ablations show that supervising a small set of intermediate layers captures most benefits and complements output-level physics losses. Code is available at https://github.com/Hxxxz0/REPA-P.

Haozhe Jia, Pengyu Yin, Wenshuo Chen +6

The Hong Kong University of Science and Technology (Guangzhou), Guangzhou, China · Shandong University, Jinan, China · Data61/CSIRO, Australia +5

Physics-Informed Diffusion Models Physics-Informed Generative Modeling PDE Surrogate Modeling Physics-Informed ML Shortcut Learning Diffusion Models

Ask PDF

Abstract

Physics-Informed GNNs

Explore similar work

pi-Net: Empirical Latent Representations for Differentiable Physics Heads in Data-Scarce Systems

Overcoming the Limits of Finite Difference Method; Physics-Informed Neural Network for Noisy High-Dimensional Heat Diffusion

Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment