Multi-Fidelity Physics-Informed Graph Neural Networks for 3D Gear Contact Stress Prediction Under Extreme Gradients
Abstract
1. Introduction
- •
- A physics-guided gated message-passing mechanism that selectively restricts information flow across steep stress gradients, reducing over-smoothing while preserving peak contact stress values on unstructured meshes.
- •
- A multi-fidelity graph representation that decouples computational graph resolution from physical mesh resolution, enabling peak-sensitive stress learning on a tractable graph size.
- •
- A scalable physics-constrained training scheme that replaces coordinate automatic differentiation with unstructured finite-difference gradient reconstruction, reducing receptive-field memory growth during full-graph training.
- •
- A separated evaluation of stress-prior-conditioned and geometry-only gating. The stress-prior-conditioned model serves as a benchmark for investigating the best achievable performance under ideal information, while the geometry-only gate provides the only label-free inference result reported here, with a peak-stress error of 4.1% (seed 42).
2. High-Fidelity FEM Data Generation and Multi-Fidelity Graph Construction
2.1. Automated Fem Generation and Adaptive Mesh Relaxation
2.2. Accelerated Graph Compilation via Hash-Mapped Topologies
2.3. Kdtree Spatial Indexing and Multi-Fidelity Distillation

3. Physics-Guided GatedSAGEConv Network
3.1. Over-Smoothing in Isotropic Aggregation
3.2. Physical Gating Mechanism
4. Physics-Constrained Optimization
4.1. Data-Driven Loss with Log-Stress Normalization
4.2. Memory Requirements of Autograd-Based Pinn
4.3. Unstructured Finite Difference Gradient Reconstruction
4.4. Projection-Induced Discontinuities and Residual Interpretation
4.5. Non-Contact Spatial Mask Loss
5. Temporal Curriculum Learning
6. Results and Discussion
6.1. Ablation Study and Baseline Comparison
6.2. Numerical Stress-Field Accuracy
Peak Stress Preservation
6.3. Inference Latency and Scope
6.4. Memory Profile
6.5. Projection-Strategy Sensitivity and Graph-Resolution Limits
6.6. Numerical Validation Scope and Experimental Requirements
7. Conclusions
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Yang, X.; Yin, S.; Chen, Y.; Zhang, Y.; Zhang, S.; Wu, Y. Numerical and experimental research of helical gear contact stress considering the influence of friction. Front. Mech. Eng. 2022, 8, 1078134. [Google Scholar] [CrossRef] [Scilit]
- Yang, Y.; Qiu, D.; Hu, N.; Hu, J.; Zhang, L.; Cheng, Z. Dynamic modeling and analysis of a split-torque transmission with a tooth crack fault. Nonlinear Dyn. 2026, 114, 148. [Google Scholar] [CrossRef] [Scilit]
- Yang, Y.; Hu, N.; Li, Y.; Cheng, Z.; Shen, G. Dynamic modeling and analysis of planetary gear system for tooth fault diagnosis. Mech. Syst. Signal Process. 2024, 207, 110946. [Google Scholar] [CrossRef] [Scilit]
- Sugunesh, A.P.; Mertens, A.J. A comprehensive study on Hertzian contact stress behaviour of engineering thermoplastic gears using 3D finite element analysis. Proc. Inst. Mech. Eng. Part C J. Mech. Eng. Sci. 2024, 238, 586–597. [Google Scholar] [CrossRef] [Scilit]
- Bonari, J.; Paggi, M.; Dini, D. A new finite element paradigm to solve contact problems with roughness. Int. J. Solids Struct. 2022, 253, 111643. [Google Scholar] [CrossRef] [Scilit]
- Gapp, S.; Otipka, D.; Maierhofer, J.; Daves, W.; Antretter, T. Efficient 3D finite element modeling of elasto-plastic Hertzian contact of two crossed cylinders with varying contact ellipses after high cycles. Tribol. Int. 2025, 211, 110825. [Google Scholar] [CrossRef] [Scilit]
- Bruzzone, F.; Fabbri, D.; Rosso, C. Machine learning surrogate models for Hertzian contact stress prediction in gear design: A comparative study of multiple approaches. Next Res. 2025, 2, 100940. [Google Scholar] [CrossRef] [Scilit]
- Gladstone, R.J.; Rahmani, H.; Suryakumar, V.; Meidani, H.; D’Elia, M.; Zareei, A. Mesh-based GNN surrogates for time-independent PDEs. Sci. Rep. 2024, 14, 3394. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Rusch, T.K.; Bronstein, M.M.; Mishra, S. A survey on oversmoothing in graph neural networks. arXiv 2023, arXiv:2303.10993. [Google Scholar] [CrossRef] [Scilit]
- Li, H.; Miao, Y.; Sharif Khodaei, Z.; Aliabadi, M.H. Finite-PINN: A physics-informed neural network with finite geometric encoding for solid mechanics. J. Mech. Phys. Solids 2025, 203, 106222. [Google Scholar] [CrossRef] [Scilit]
- He, W.; Li, J.; Kong, X.; Deng, L. Multi-level physics informed deep learning for solving partial differential equations in computational structural mechanics. Commun. Eng. 2024, 3, 151. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Oono, K.; Suzuki, T. Graph neural networks exponentially lose expressive power for node classification. In Proceedings of the International Conference on Learning Representations (ICLR), Addis Ababa, Ethiopia, 26–30 April 2020; Available online: https://openreview.net/forum?id=S1ldO2EFPr (accessed on 1 January 2026).
- Li, Z.; Kovachki, N.; Azizzadenesheli, K.; Liu, B.; Bhattacharya, K.; Stuart, A.; Anandkumar, A. Fourier neural operator for parametric partial differential equations. In Proceedings of the International Conference on Learning Representations (ICLR), Virtual Event, Austria, 3–7 May 2021; Available online: https://openreview.net/forum?id=c8P9NQVtmnO (accessed on 1 January 2026).
- Lu, L.; Jin, P.; Pang, G.; Zhang, Z.; Karniadakis, G.E. Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators. Nat. Mach. Intell. 2021, 3, 218–229. [Google Scholar] [CrossRef] [Scilit]
- Pathak, J.; Subramanian, S.; Harrington, P.; Raja, S.; Chattopadhyay, A.; Mardani, M.; Kurth, T.; Hall, D.; Li, Z.; Azizzadenesheli, K.; et al. FourCastNet: A global data-driven high-resolution weather model using adaptive Fourier neural operators. arXiv 2022, arXiv:2202.11214. [Google Scholar] [CrossRef] [Scilit]
- Bai, J.; Jeong, H.; Batuwatta-Gamage, C.P.; Xiao, S.; Wang, Q.; Rathnayaka, C.M.; Alzubaidi, L.; Liu, G.-R.; Gu, Y. An introduction to programming Physics-Informed Neural Network-based computational solid mechanics. Int. J. Comput. Methods 2023, 20, 2350013. [Google Scholar] [CrossRef] [Scilit]
- Jagtap, A.D.; Kawaguchi, K.; Karniadakis, G.E. Adaptive activation functions accelerate convergence in deep and physics-informed neural networks. J. Comput. Phys. 2020, 404, 109136. [Google Scholar] [CrossRef] [Scilit]
- Shin, Y.; Darbon, J.; Karniadakis, G.E. On the convergence of physics informed neural networks for linear second-order elliptic and parabolic type PDEs. Commun. Comput. Phys. 2020, 28, 2042–2074. [Google Scholar] [CrossRef] [Scilit]
- Pfaff, T.; Fortunato, M.; Sanchez-Gonzalez, A.; Battaglia, P.W. Learning mesh-based simulation with graph networks. In Proceedings of the International Conference on Learning Representations (ICLR), Virtual Event, Austria, 3–7 May 2021; Available online: https://openreview.net/forum?id=roNqYL0_XP (accessed on 1 January 2026).
- Li, Z.; Kovachki, N.; Azizzadenesheli, K.; Liu, B.; Bhattacharya, K.; Stuart, A.; Anandkumar, A. Multipole graph neural operator for parametric partial differential equations. Adv. Neural Inf. Process. Syst. 2020, 33, 6755–6766. [Google Scholar]
- Defferrard, M.; Bresson, X.; Vandergheynst, P. Convolutional neural networks on graphs with fast localized spectral filtering. Adv. Neural Inf. Process. Syst. 2016, 29, 3844–3852. [Google Scholar]
- Chen, J.; Ma, T.; Xiao, C. FastGCN: Fast learning with graph convolutional networks via importance sampling. In Proceedings of the International Conference on Learning Representations (ICLR), Vancouver, BC, Canada, 30 April–3 May 2018; Available online: https://openreview.net/forum?id=rytstxWAW (accessed on 1 January 2026).
- Penwarden, M.; Zhe, S.; Narayan, A.; Kirby, R.M. Multifidelity modeling for physics-informed neural networks (PINNs). J. Comput. Phys. 2022, 451, 110844. [Google Scholar] [CrossRef] [Scilit]
- Zhang, H.; Jiang, L.; Chu, X.; Wen, Y.; Li, L.; Liu, J.; Xiao, Y.; Wang, L. Combining physics-informed graph neural network and finite difference for solving forward and inverse spatiotemporal PDEs. Comput. Phys. Commun. 2025, 308, 109462. [Google Scholar] [CrossRef] [Scilit]
- Goswami, S.; Bora, A.; Yu, Y.; Karniadakis, G.E. Physics-Informed Deep Neural Operator Networks. In Machine Learning in Modeling and Simulation: Methods and Applications; Rabczuk, T., Bathe, K.-J., Eds.; Springer: Cham, Switzerland, 2023; pp. 219–254. [Google Scholar] [CrossRef] [Scilit]
- Ezemba, J.; McComb, C.; Tucker, C. Neural network surrogate modeling for stochastic finite element method using three-dimensional graph representations: A comparative study. J. Mech. Des. 2026, 148, 011704. [Google Scholar] [CrossRef] [Scilit]
- Black, N.; Najafi, A.R. Learning finite element convergence with the Multi-fidelity Graph Neural Network. Comput. Methods Appl. Mech. Eng. 2022, 397, 115120. [Google Scholar] [CrossRef] [Scilit]
- Chakraborty, S. Transfer learning based multi-fidelity physics informed deep neural network. J. Comput. Phys. 2021, 426, 109942. [Google Scholar] [CrossRef] [Scilit]
- Tripura, T.; Thakur, A.; Chakraborty, S. Multi-fidelity wavelet neural operator surrogate model for time-independent and time-dependent reliability analysis. Probabilistic Eng. Mech. 2024, 77, 103672. [Google Scholar] [CrossRef] [Scilit]
- Zhang, T.; Xiao, Z.; Xia, Y. Physics-Informed Multi-Fidelity Networks for Solving Discontinuous Problems. In Proceedings of the AIAA SCITECH 2026 Forum, AIAA Paper 2026-0587, Orlando, FL, USA, 12–16 January 2026. [Google Scholar] [CrossRef] [Scilit]
- Würth, T.; Freymuth, N.; Zimmerling, C.; Neumann, G.; Kärger, L. Physics-informed MeshGraphNets (PI-MGNs): Neural finite element solvers for non-stationary and nonlinear simulations on arbitrary meshes. Comput. Methods Appl. Mech. Eng. 2024, 428, 117102. [Google Scholar] [CrossRef] [Scilit]
- Kneifl, J.; Fehr, J.; Brunton, S.L.; Kutz, J.N. Multi-hierarchical surrogate learning for explicit structural dynamical systems using graph convolutional neural networks. Comput. Mech. 2025, 75, 1115–1135. [Google Scholar] [CrossRef] [Scilit]
- Barwey, S.; Kim, H.; Maulik, R. Interpretable A-posteriori error indication for graph neural network surrogate models. Comput. Methods Appl. Mech. Eng. 2025, 433, 117509. [Google Scholar] [CrossRef] [Scilit]
- Shao, X.; Liu, Z.; Zhang, S.; Zhao, Z.; Hu, C. PIGNN-CFD: A physics-informed graph neural network for rapid predicting urban wind field defined on unstructured mesh. Build. Environ. 2023, 232, 110056. [Google Scholar] [CrossRef] [Scilit]
- Guevara Garban, M.R.; Chemisky, Y.; Clément, M.; Prulière, É. Physics-Informed Graph Neural Networks to Reconstruct Local Fields Considering Finite Strain Hyperelasticity. Int. J. Numer. Methods Eng. 2025, 126, e70193. [Google Scholar] [CrossRef] [Scilit]
- Liu, K.; Ma, L. MeshODENet: A Graph-Informed Neural Ordinary Differential Equation Neural Network for Simulating Mesh-Based Physical Systems. J. Appl. Mech. 2026, 93, 051005. [Google Scholar] [CrossRef] [Scilit]
- Liu, Z.; Liu, Y.; Yan, X.; Liu, W.; Guo, S.; Zhang, C.-A. AsPINN: Adaptive symmetry-recomposition physics-informed neural networks. Comput. Methods Appl. Mech. Eng. 2024, 432, 117405. [Google Scholar] [CrossRef] [Scilit]
- Hu, H.; Qi, L.; Chao, X. Physics-informed Neural Networks (PINN) for computational solid mechanics: Numerical frameworks and applications. Thin-Walled Struct. 2024, 205, 112495. [Google Scholar] [CrossRef] [Scilit]
- Tong, Z.; Chen, R. Physics-informed spatiotemporal neural network for unsteady propeller wake prediction via least squares finite-difference framework. Phys. Fluids 2026, 38, 025132. [Google Scholar] [CrossRef] [Scilit]
- Gao, H.; Zahr, M.J.; Wang, J.-X. Physics-informed graph neural Galerkin networks: A unified framework for solving PDE-governed forward and inverse problems. Comput. Methods Appl. Mech. Eng. 2022, 390, 114502. [Google Scholar] [CrossRef] [Scilit]
- Hildebrand, S.; Klinge, S. Comparison of neural FEM and neural operator methods for applications in solid mechanics. Neural Comput. Appl. 2024, 36, 16657–16682. [Google Scholar] [CrossRef] [Scilit]
- Kaewnuratchadasorn, C.; Wang, J.; Kim, C.-W. Physics-informed neural operator solver and super-resolution for solid mechanics. Comput.-Aided Civ. Infrastruct. Eng. 2024, 39, 3435–3451. [Google Scholar] [CrossRef] [Scilit]
- Jha, P.K. Residual-based error corrector operator to enhance accuracy and reliability of neural operator surrogates of nonlinear variational boundary-value problems. Comput. Methods Appl. Mech. Eng. 2024, 419, 116595. [Google Scholar] [CrossRef] [Scilit]
- Zhang, Z. MODNO: Multi-Operator learning with Distributed Neural Operators. Comput. Methods Appl. Mech. Eng. 2024, 431, 117229. [Google Scholar] [CrossRef] [Scilit]
- Yang, J.; Liu, X.; Diao, Y.; Chen, X.; Hu, H. Adaptive task decomposition physics-informed neural networks. Comput. Methods Appl. Mech. Eng. 2024, 418, 116561. [Google Scholar] [CrossRef] [Scilit]
- Li, K. MultiPINN: Multi-head enriched physics-informed neural networks for differential equations solving. Neural Comput. Appl. 2024, 36, 11371–11395. [Google Scholar] [CrossRef] [Scilit]
- Zhang, Y.; Yang, Y.; Hu, Y.; Guo, Z. Multi-Fidelity Data and Prior-Enhanced Physics-Informed Neural Networks for Multi-Parameter Identification of Prestressed Concrete Beams with Unquantifiable Noise. Appl. Sci. 2026, 16, 608. [Google Scholar] [CrossRef] [Scilit]





| Gear and Material Setting | Value | Loading and Contact Setting | Value | Mesh and Dataset Setting | Value |
|---|---|---|---|---|---|
| Gear type | Spur/helical | Torque range | 100–480 N·m | FEM cases | 5000 cases |
| Module | 2.0–4.5 | Misalignment range | 0–0.30° | Train: validation: test | 3500:750:750 |
| Teeth number | 18–120 | Contact formulation | Augmented Lagrange | Element type | SOLID187 |
| Pressure angle | 20° | Friction coefficient | 0.10 | Contact-region mesh size | 0.5 mm |
| Helix angle | 0–25° | Penetration tolerance | <1.0 × 10−3 mm | Far-field mesh size | 1.0 mm |
| Face width | 18–74 mm | Contact stiffness | 1.0 (program-controlled factor) | Typical high-fidelity nodes | 1 × 106 |
| Material grade | 42CrMo4 | Solver | ANSYS 2026 R1 Mechanical | Typical coarse graph nodes | 5 × 105 |
| Young’s modulus | 206 GPa | Hardware | 96 GB GPU | ||
| Poisson’s ratio | 0.3 |
| Module | Parameter | Value |
|---|---|---|
| Network architecture | Backbone | Gated GraphSAGE |
| Graph layers/hidden channels | 5/128 | |
| Activation/normalization | SiLU/LayerNorm | |
| Regularization | Dropout/gradient clipping | Dropout = 0.1; global gradient norm clipped at 10.0 |
| Optimization | Optimizer | AdamW |
| Learning rate/weight decay | 1.0 × 10−3/1.0 × 10−5 | |
| LR schedule | Warm-up + cosine restarts | |
| Training | Batch size/epochs | 1 graph/100 |
| Train: validation: test | 70%:15%:15% | |
| Loss | Stress scaling | sgn(s) ln(1 + |s|/10 MPa), componentwise |
| Physics/non-contact weights | λphys,max = 0.5; λmask = 0.02 | |
| Curriculum | Physics schedule | 0 for epochs 1–20; ramp 21–29; full 30–100 |
| Sampling | Physics nodes | 10%, capped at 8192 |
| Reproducibility | Primary seed/additional full-model seeds | 42/37, 55, 73, 91 |
| Projection Strategy | Coarse-Graph NMSE | Peak-Stress Error | Complementary-Energy Error |
|---|---|---|---|
| Distance-weighted average | 1.18 × 10−4 | 3.1% | 2.8% |
| Maximum projection | 8.87 × 10−5 | 2.3% | 2.1% |
| Variant | Changed Component | NMSE | R2 | Peak-Stress Error | Peak VRAM |
|---|---|---|---|---|---|
| Ours-Full (conditioned) | All components; projected stress prior; five-seed mean ± SD | (9.1 ± 0.4) × 10−5 | 0.985 ± 0.001 | 2.5 ± 0.2% | 47.6 GB |
| Ours-GeometryGate | Stress terms removed from gate | Not reported | Not reported | 4.1% (seed 42) | Not reported |
| Ours-NoGate | Physical edge gate removed | 1.74 × 10−4 | 0.964 | 15.6% | 46.8 GB |
| Ours-AutoGrad | AD equilibrium residual; full-graph OOM | Not available | Not available | Not available | >100 GB projected; OOM |
| Ours-WeightedAvg | Distance-weighted projection | 1.18 × 10−4 | 0.978 | 3.1% | 47.3 GB |
| Ours-NoCurriculum | Full physics weight from epoch 1 | 1.46 × 10−4 | 0.971 | 5.4% | 47.8 GB |
| Ours-NoMask | Spatial mask loss removed | 9.6 × 10−5 | 0.983 | 2.5% | 47.4 GB |
| Method | Training/Input Representation | NMSE | R2 | Peak-Stress Error | Forward/Solve Time | Parameters | FLOPs |
|---|---|---|---|---|---|---|---|
| High-fidelity FEM | Native dense FEM mesh | Reference | 1.000 | Reference | 4.5 h | 0 | — |
| 5-layer GCN | Coarse graph, isotropic aggregation | 3.82 × 10−4 | 0.944 | 17.6% | 35 ms | 111,560 | 1.28 × 1011 |
| GCN + Prior | Coarse graph, isotropic, with prior | 1.42 × 10−4 | 0.976 | 5.8% | 36 ms | 112,200 | 1.31 × 1011 |
| GraphSAGE | Coarse graph, mean aggregation | 2.61 × 10−4 | 0.961 | 14.2% | 39 ms | 194,120 | 2.00 × 1011 |
| Point-cloud PINN | Coordinate MLP with AD residual | 1.93 × 10−4 | 0.970 | 8.9% | 210 ms | 268,550 | 2.69 × 1011 |
| Proposed (conditioned) | Multi-fidelity gated graph + LSFD | 8.7 × 10−5 | 0.985 | 2.3% | 42 ms conditioned pass | 196,680 | 2.39 × 1011 |
| Split | Cases | NMSE | R2 | Median |Error| | 95th Percentile |Error| | Peak-Stress Error |
|---|---|---|---|---|---|---|
| Training | 3500 | 8.2 × 10−5 | 0.987 | 3.1 MPa | 11.4 MPa | 2.1% |
| Validation | 750 | 8.9 × 10−5 | 0.986 | 3.3 MPa | 12.1 MPa | 2.4% |
| Test | 750 | (9.1 ± 0.4) × 10−5 | 0.985 ± 0.001 | 3.5 ± 0.2 MPa | 12.7 ± 0.5 MPa | 2.5 ± 0.2% |
| Solver Framework | Gradient Computation | Hardware | Peak Memory | Forward/Solve Time |
|---|---|---|---|---|
| Traditional FEM | Analytical FEM residual | 64-core HPC cluster | Not reported | 4.5 h |
| Standard point-cloud PINN | Reverse-mode AD | 96 GB GPU | >100 GB projected; OOM | Not available |
| Proposed PINN–GNN (conditioned) | Unstructured LSFD | 96 GB GPU | 47.6 GB measured | 42 ms conditioned pass |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Zeng, J.; Li, Z.; Lin, Q. Multi-Fidelity Physics-Informed Graph Neural Networks for 3D Gear Contact Stress Prediction Under Extreme Gradients. Processes 2026, 14, 2706. https://doi.org/10.3390/pr14172706
Zeng J, Li Z, Lin Q. Multi-Fidelity Physics-Informed Graph Neural Networks for 3D Gear Contact Stress Prediction Under Extreme Gradients. Processes. 2026; 14(17):2706. https://doi.org/10.3390/pr14172706
Chicago/Turabian StyleZeng, Jinchao, Zicheng Li, and Qizhe Lin. 2026. "Multi-Fidelity Physics-Informed Graph Neural Networks for 3D Gear Contact Stress Prediction Under Extreme Gradients" Processes 14, no. 17: 2706. https://doi.org/10.3390/pr14172706
APA StyleZeng, J., Li, Z., & Lin, Q. (2026). Multi-Fidelity Physics-Informed Graph Neural Networks for 3D Gear Contact Stress Prediction Under Extreme Gradients. Processes, 14(17), 2706. https://doi.org/10.3390/pr14172706
