PINNForge: Execution-Grounded Evolutionary Design of Physics-Informed Neural Networks
Organizations: Nankai University · East China Normal University · Hanyang University
Abstract
Physics-informed neural networks (PINNs) require coordinated choices across representation, architecture, sampling, constraints, loss construction, and optimization, yet effective configurations can vary substantially across partial differential equations (PDEs). Existing automated design methods search over these choices, but training outcomes are still used primarily to rank candidates rather than to inform subsequent designs. We develop PINNForge, an execution-grounded large language model (LLM)-driven evolutionary framework that treats observed PINN training behavior as a cross-generation design signal. PINNForge initializes diverse configurations from PDE-related prior knowledge, executes them, retains globally strong configurations, and feeds their execution evidence and accumulated run-level experience back to the LLM. The LLM then revises and recombines coupled design components or explores new combinations within a generate--execute--evaluate--evolve loop. Across 25 PDE benchmarks, PINNForge achieves the lowest mean MSE on 24 tasks among RoPINN, PINNsFormer, PINNsAgent, and PINNForge. Removing knowledge guidance, execution feedback, or evolutionary search increases the median task-wise MSE ratio to 3.74, 12.10, and 10.10, respectively, relative to full PINNForge.
Figures & tables
| PDE | RoPINN | PINNsFormer | PINNsAgent | PINNForge |
|---|---|---|---|---|
| burgers_1d | 7.31E-04 6.13E-05 | 1.08E-03 8.21E-06 | 6.52E-05 1.55E-05 | 6.50E-05 4.42E-07 |
| burgers_2d | 2.40E-01 3.12E-02 | 2.65E-01 6.83E-02 | 1.98E-01 1.66E-02 | 1.95E-01 1.31E-02 |
| gray_scott_2d | 8.64E-03 4.53E-05 | 1.57E-02 1.32E-03 | 4.36E-03 3.08E-05 | 3.60E-03 2.65E-04 |
| heat_2d_complex_geometry | 4.00E-04 9.10E-05 | 2.78E-03 5.77E-05 | 1.84E-03 9.95E-04 | 4.45E-04 9.72E-05 |
| heat_2d_long_time | 2.56E+00 4.53E-01 | 3.83E+00 1.64E-01 | 4.32E+00 2.78E+00 | 1.13E+00 4.21E-02 |
| heat_2d_multiscale | 2.75E-05 7.52E-06 | 5.57E-05 9.93E-06 | 3.51E-05 2.20E-05 | 1.19E-05 2.16E-06 |
| Variant | Knowl. guidance | Exec. feedback | Parent evolution | Median ratio |
|---|---|---|---|---|
| w/o Knowledge Guidance | ||||
| w/o Execution Feedback to LLM | ||||
| w/o Evolutionary Search | ||||
| PINNForge |
| Method | Selection protocol | Budget |
|---|---|---|
| RoPINN | Best of 5 independently initialized 20K-step candidates | 100K |
| PINNsFormer | Best of 5 independently initialized 20K-step candidates | 100K |
| PINNsAgent | Best of 5 independently initialized 20K-step candidates | 100K |
| PINNForge | 53 1K LF + Top-3 10K HF | 83K |
| PDE | G0 | Best LF | Final HF | G0 HF | LF HF |
|---|---|---|---|---|---|
| burgers_1d | 3.64E-03 | 2.05E-04 | 6.52E-05 | ||
| heat_2d_multiscale | 3.02E-02 | 9.14E-05 | 1.01E-05 | ||
| poisson_5d | 3.96E-04 | 1.56E-07 | 5.65E-08 | ||
| navier_stokes_2d_classic | 1.86E-03 | 1.61E-05 | 2.35E-06 |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Design dimension | Representative executable choices |
|---|---|
| Representation | Coordinate transforms, Fourier or periodic features, and multiscale representations supported by the implementation. |
| Architecture | Network depth and width, activation functions, residual connections, and other supported topology-related choices. |
| Sampling | Collocation-point allocation, sampling distributions, and adaptive or localized resampling mechanisms. |
| Constraints | Soft constraints, exact constraint transforms when available, and supported periodic or interface handling. |
| Loss | PDE-residual construction, loss weighting, normalization, causal weighting, and other supported physical-loss handling. |
| Optimization | Optimizer choice, learning rate, gradient handling, scheduling, and staged optimization. |
| Context | Generation 0 | Generation | Purpose |
|---|---|---|---|
| Target PDE description | Provides the governing equation, computational domain, IC/BC, and problem characteristics | ||
| Prior knowledge | Provides PINN design rationale relevant to the current PDE | ||
| Allowed search space | Restricts the network, sampling, constraint, loss, and optimization components that may be generated | ||
| Training / resource budget | Restricts parameter count, sampling scale, and training budget | ||
| Role / search-strategy condition | Specifies the design focus of the current candidate | ||
| Global Top-3 AlgorithmSpec | – | Provides validated high-performing parent configurations |
| Generation-0 role | Primary design emphasis |
|---|---|
| Robust reference | Construct a conservative reference design that prioritizes broadly applicable mechanisms and stable training. |
| Constraint / causality | Emphasize physical-constraint enforcement and, when relevant, temporal or causal structure. |
| Localized sampling | Emphasize local structure, collocation-point placement, and adaptive or localized refinement of difficult regions. |
| Representation | Emphasize network input or feature representation and its interaction with the solver architecture. |
| Residual / loss | Emphasize PDE-residual construction, physical-loss formulation, and loss-handling choices. |
| Optimization | Emphasize optimizer selection, scheduling, gradient behavior, and numerical training stability. |
| Evolution strategy | Primary evolution direction |
|---|---|
| Refine | Use the current strongest candidate as the primary parent, preserve effective existing mechanisms, and make limited adjustments targeting its main shortcomings. |
| Physics-Guided | Adjust sampling-, constraint-, and loss-related designs primarily according to the observed PDE residual and initial/boundary-condition errors. |
| Architecture-Guided | Adjust network representation, architecture, and related training configurations according to representation capacity, model capacity, and optimization behavior. |
| Synthesis | Combine complementary, well-performing design mechanisms from multiple Global Top-3 parents into a new complete configuration. |
| Novelty | Explore design combinations not yet covered by the current high-performing candidates or the existing search trajectory. |
| Variant | Prior Knowledge | Run-scoped Posterior | Execution Feedback to LLM | Parent-based Evolution |
|---|---|---|---|---|
| w/o Knowledge Guidance | ||||
| w/o Execution Feedback to LLM | ||||
| w/o Evolutionary Search | ||||
| PINNForge |
| Component | Parent Configuration | Execution Evidence | Design Decision | Child Configuration |
|---|---|---|---|---|
| Architecture | Modified MLP, ; no residual connections | Completed the full 1,000-step training without NaN or solution collapse | Preserve. The architecture was numerically stable in the observed run | Keep Modified MLP, |
| Constraint | Exact hard transform burgers_1d_initial_dirichlet_exact | IC error , BC error | Preserve. Do not reallocate search emphasis to constraints that are already well satisfied | Keep the exact hard transform |
| Sampling | 10,000 Sobol interior points + RAD; add 500 each time; cap at 15,000; residual exponent 2.5 | Persistent residual hotspot at , | Modify. Strengthen local refinement while preserving global Sobol coverage | Add 600 each time; raise the cap to 16,000; keep the remaining RAD settings unchanged |
| Optimization | Adam 800 steps, lr , one-cycle; L-BFGS 200 steps, history 100 | Optimization remains numerically stable, but clear residual error remains at the end of training | Explore. Treat optimization changes as accompanying exploration rather than misclassifying a stable optimizer as a failure | Increase Adam lr to , switch to step decay; L-BFGS history 120 |
| Metric | Parent G6-C4 | Child G7-C4 |
|---|---|---|
| Solution MSE | ||
| PDE residual | ||
| IC error | ||
| BC error |
| Generation | Candidate | Parent(s) | Evolution Mode | Main Inherited / Modified Components |
|---|---|---|---|---|
| 0 | G0-C2 | – | Independent Initialization | 4 64 modified MLP, tanh , soft constraints, Sobol + RAD, Adam + L-BFGS |
| 1 | G1-C3 | G0-C2, G0-C6, G0-C4 | Multi-Parent Synthesis | Combine RAD sampling, a deeper network topology, and optimization control; expand to a 6 96 modified MLP |
| 2 | G2-C1 | G1-C3 | Single-Parent Refinement | Preserve the 6 96 network; increase the RAD exponent and interior-point budget |
| 3 | G3-C4 | G2-C1, G2-C4 | Multi-Parent Synthesis | Introduce an exact hard IC/BC transform; use fixed weighting; increase the RAD cap |
| 4 | G4-C4 | G3-C4, G3-C2 | Multi-Parent Synthesis | Preserve hard constraints and Sobol + RAD; widen the network to 6 128; add one-cycle Adam and L-BFGS recovery |
| 5 | G5-C4 | G4-C4, G4-C3 | Multi-Parent Synthesis | Adjust the Adam schedule, RAD capacity, and L-BFGS history |
| Setting | Value |
|---|---|
| Generation-0 candidates | 8 |
| Candidates per later generation | 5 |
| Min. / max. completed generations | 4 / 10 |
| Parent population | Global Top-3 |
| Low-fidelity training | 1K steps |
| High-fidelity training | 10K steps |
| Generation-0 direction | Temperature |
|---|---|
| Robust reference | 0.20 |
| Constraint / causality | 0.35 |
| Localized sampling | 0.40 |
| Representation | 0.45 |
| Residual / loss | 0.45 |
| Optimization | 0.40 |
| Evolution strategy | Normal | Escape |
|---|---|---|
| Refine | 0.20 | 0.35 |
| Physics-Guided | 0.36 | 0.51 |
| Architecture-Guided | 0.42 | 0.57 |
| Synthesis | 0.47 | 0.62 |
| Novelty | 0.58 | 0.73 |
| Component | Configuration |
|---|---|
| GPU | NVIDIA GeForce RTX 4090 |
| NVIDIA Driver | 550.163.01 |
| CUDA | 12.4 |
| Python | 3.12 |
| PyTorch | 2.6.0+cu124 |
| Precision | FP32 |
| PDE Problem | RoPINN | PINNsFormer | PINNsAgent | PINNForge | Rank | PINNacle Best |
|---|---|---|---|---|---|---|
| burgers_1d | 7.31E-04 6.13E-05 | 1.08E-03 8.21E-06 | 6.52E-05 1.55E-05 | 6.50E-05 4.42E-07 | 1 | 7.90E-05 |
| burgers_2d | 2.40E-01 3.12E-02 | 2.65E-01 6.83E-02 | 1.98E-01 1.66E-02 | 1.95E-01 1.31E-02 | 1 | 1.09E-01 |
| gray_scott_2d | 8.64E-03 4.53E-05 | 1.57E-02 1.32E-03 | 4.36E-03 3.08E-05 | 3.60E-03 2.65E-04 | 1 | 4.32E-03 |
| heat_2d_complex_geometry | 4.00E-04 9.10E-05 | 2.78E-03 5.77E-05 | 1.84E-03 9.95E-04 | 4.45E-04 9.72E-05 | 2 | 8.53E-04 |
| heat_2d_long_time | 2.56E+00 4.53E-01 | 3.83E+00 1.64E-01 | 4.32E+00 2.78E+00 | 1.13E+00 4.21E-02 | 1 | 1.13E+00 |
| heat_2d_multiscale | 2.75E-05 7.52E-06 | 5.57E-05 9.93E-06 | 3.51E-05 2.20E-05 | 1.19E-05 2.16E-06 | 1 | 5.27E-05 |
| PDE | Characteristic | w/o Knowledge Guidance | w/o Execution Feedback to LLM | w/o Evolutionary Search | PINNForge |
|---|---|---|---|---|---|
| burgers_1d | Nonlinear / shock | 7.42E-05 6.31E-05 | 6.32E-04 7.82E-05 | 2.78E-04 3.24E-05 | 6.50E-05 4.42E-07 |
| burgers_2d | 2D nonlinear transport | 2.44E-01 2.86E-02 | 6.71E-01 8.37E-02 | 1.67E+00 2.31E-01 | 1.95E-01 1.31E-02 |
| gray_scott_2d | Coupled reaction–diffusion | 5.29E-03 4.91E-04 | 5.36E-03 5.62E-03 | 6.23E-03 6.87E-04 | 3.60E-03 2.65E-04 |
| heat_2d_multiscale | Multiscale dynamics | 3.81E-05 4.28E-06 | 2.08E-04 2.74E-04 | 1.30E-03 1.66E-03 | 1.19E-05 2.16E-06 |
| heat_2d_varying_coefficient | Variable coefficient | 2.18E-03 2.03E-03 | 2.57E-03 2.71E-04 | 4.15E-03 4.86E-04 | 1.59E-03 4.77E-04 |
| heat_2d_complex_geometry | Complex geometry | 1.90E-03 2.41E-03 | 1.59E+00 2.14E-01 | 1.97E-02 2.77E-03 | 4.45E-04 9.72E-05 |
| Method | Selection Protocol | Budget |
|---|---|---|
| RoPINN | Best of 5 independently initialized 20K candidates | 100K |
| PINNsFormer | Best of 5 independently initialized 20K candidates | 100K |
| PINNsAgent | Best of 5 independently initialized 20K candidates | 100K |
| PINNForge | 53 1K LF + Top-3 10K HF | 83K |
| PDE | Generation-0 Best MSE | Best LF MSE | Final HF MSE | G0 Best LF | G0 Final HF |
|---|---|---|---|---|---|
| burgers_1d | 3.64E-03 | 2.05E-04 (G9) | 6.52E-05 | 17.76 | 55.81 |
| burgers_2d | 2.49E-01 | 2.13E-01 (G5) | 1.78E-01 | 1.17 | 1.39 |
| gray_scott_2d | 4.37E-03 | 4.03E-03 (G9) | 4.07E-03 | 1.08 | 1.07 |
| heat_2d_complex_geometry | 1.28E-02 | 3.26E-03 (G8) | 4.63E-04 | 3.92 | 27.64 |
| heat_2d_long_time | 1.13E+00 | 1.10E+00 (G5) | 1.08E+00 | 1.03 | 1.05 |
| heat_2d_multiscale | 3.02E-02 | 9.14E-05 (G9) | 1.01E-05 | 330.42 | 2,986.95 |
Physics-Informed ML
Physics-informed machine learning combines data-driven models with physical laws to solve, predict, and infer scientific systems.
See all 830 papers