Auxiliary Finite-Difference Residual-Gradient Regularization for PINNs
Organizations: Computational Sciences Laboratory, Department of Mechanical Engineering, University of Cyprus, 1 University Avenue, Aglantzia, 2109, Nicosia, Cyprus
Abstract
Physics-informed neural networks (PINNs) are often selected by a single scalar loss even when the quantity of interest is more specific. We study a hybrid design in which the governing PDE residual remains automatic-differentiation (AD) based, while finite differences (FD) appear only in a weak auxiliary term that penalizes gradients of the sampled residual field. The FD term regularizes the residual field without replacing the PDE residual itself. We examine this idea in two stages. Stage 1 is a controlled Poisson benchmark comparing a baseline PINN, the FD residual-gradient regularizer, and a matched AD residual-gradient baseline. Stage 2 transfers the same logic to a three-dimensional annular heat-conduction benchmark (PINN3D), where baseline errors concentrate near a wavy outer wall and the auxiliary grid is implemented as a body-fitted shell adjacent to the wall. In Stage 1, the FD regularizer reproduces the main effect of residual-gradient control while exposing a trade-off between field accuracy and residual cleanliness. In Stage 2, the shell regularizer improves the application-facing quantities, namely outer-wall flux and boundary-condition behavior. Across seeds 0-5 and 100k epochs, the most reliable tested configuration is a fixed shell weight of 5e-4 under the Kourkoutas-beta optimizer regime: relative to a matched run without the shell term, it reduces the mean outer-wall BC RMSE from 1.22e-2 to 9.29e-4 and the mean wall-flux RMSE from 9.21e-3 to 9.63e-4. Adam with beta2=0.999 becomes usable when the initial learning rate is reduced to 1e-3, although its shell benefit is less robust than under Kourkoutas-beta. Overall, the results support a targeted view of hybrid PINNs: an auxiliary-only FD regularizer is most valuable when it is aligned with the physical quantity of interest, here the outer-wall flux.
Figures & tables
| Inherited benchmark ingredients | New in this manuscript |
| Steady PINN3D annular heat-conduction workload, wavy outer-wall geometry, six-term weighted objective, reference FD solution, and the base MLP/optimizer context from the prior PINN3D and Kourkoutas-β work [ 8 , 9 ] . | An auxiliary-only FD regularizer applied to the sampled AD residual field ; the controlled Stage-1 Poisson mechanism study; the body-fitted outer-shell deployment in Stage 2; seedwise wall-reference audits; and the optimizer and learning-rate (LR) sensitivity checks used to interpret the Stage-2 result carefully. |
| Item | Stage-2 protocol used in this paper |
|---|---|
| Physical domain | Steady cylindrical annulus with wavy outer wall , with , , and |
| Boundary conditions | Inner wall and inlet at unit temperature; outlet Neumann; prescribed flux on the outer wall; periodicity in |
| Base objective | Six weighted terms: PDE, inner, inlet, outlet, outer, and periodic/theta contributions inherited from the PINN3D benchmark [ 8 ] |
| Model | Fully connected 16-layer MLP of width 128 with SiLU activations and scalar temperature output |
| Training cloud | 4000 interior points; 2000 points on each of inner, inlet, outlet, and outer boundaries; 400 periodic point pairs |
| Validation cloud | 1024 interior points; 512 points on each boundary; 200 periodic point pairs |
| model | best val ( ) | rel ( ) | rel ( ) | residual RMSE ( ) | RMSE |
|---|---|---|---|---|---|
| PINN off | |||||
| FD fixed | |||||
| FD linear | |||||
| AD fixed | |||||
| AD linear |
| model | final loss ( ) | wall BC RMSE ( ) | RMSE ( ) | RMSE ( ) |
|---|---|---|---|---|
| OFF | ||||
| shell fixed |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| direction | comparison | relative change | max pointwise change |
|---|---|---|---|
| axial | |||
| axial | — | ||
| axial | — | ||
| radial ( ) | |||
| radial ( ) | |||
| radial ( ) |
| optimizer | init. LR | arm | final loss | wall BC RMSE | RMSE | RMSE |
|---|---|---|---|---|---|---|
| Kourkoutas-β | OFF | |||||
| Kourkoutas-β | fixed shell | |||||
| Kourkoutas-β | OFF | |||||
| Kourkoutas-β | fixed shell | |||||
| Adam95 | OFF | |||||
| Adam95 | fixed shell |
| seed | final loss | wall BC RMSE | RMSE | RMSE | ||||
|---|---|---|---|---|---|---|---|---|
| OFF | fixed shell | OFF | fixed shell | OFF | fixed shell | OFF | fixed shell | |
| 0 | ||||||||
| 1 | ||||||||
| 2 | ||||||||
| 3 | ||||||||
| 4 | ||||||||
| seed | final loss | wall BC RMSE | RMSE | RMSE | ||||
|---|---|---|---|---|---|---|---|---|
| OFF | fixed shell | OFF | fixed shell | OFF | fixed shell | OFF | fixed shell | |
| 0 | ||||||||
| 1 | ||||||||
| 2 | ||||||||
| 3 | ||||||||
| 4 | ||||||||
| FD weight | best val | rel | rel | residual RMSE | RMSE |
|---|---|---|---|---|---|
| off | 2.178e-03 | 3.708e-03 | 1.170e-02 | 5.187e-02 | 2.029e+00 |
| 1e-5 | 2.142e-03 | 3.691e-03 | 1.163e-02 | 5.139e-02 | 2.009e+00 |
| 1e-4 | 1.895e-03 | 3.619e-03 | 1.119e-02 | 4.804e-02 | 1.870e+00 |
| 5e-4 | 1.260e-03 | 3.401e-03 | 1.051e-02 | 3.905e-02 | 1.510e+00 |
| 1e-3 | 9.767e-04 | 3.316e-03 | 9.882e-03 | 3.390e-02 | 1.324e+00 |
| FD weight | grid strategy | best val | rel | rel | residual RMSE | RMSE |
|---|---|---|---|---|---|---|
| 1e-3 | fixed-safe | 5.366e-04 | 2.489e-03 | 8.193e-03 | 2.257e-02 | 1.222e+00 |
| 1e-3 | cycle4 | 6.893e-04 | 5.196e-03 | 8.315e-03 | 2.559e-02 | 1.191e+00 |
| 1e-3 | jitter4 | 6.676e-04 | 3.565e-03 | 8.277e-03 | 2.490e-02 | 1.205e+00 |
| 1e-2 | fixed-safe | 3.983e-04 | 3.723e-03 | 1.123e-02 | 1.217e-02 | 7.689e-01 |
| 1e-2 | cycle4 | 7.197e-04 | 5.224e-03 | 1.166e-02 | 2.190e-02 | 7.235e-01 |
| 1e-2 | jitter4 | 1.457e-03 | 4.258e-03 | 1.155e-02 | 3.540e-02 | 8.042e-01 |
Neural PDE Solvers
Neural PDE solvers use trainable models to approximate solutions or solution operators for partial differential equations, often with physical constraints.
See all 268 papers