Organizations: School of Computer Science and Technology, Chongqing University of Posts and Telecommunications, Chongqing 400065, China · Center for Scientific Intelligence Innovation, University of Science and Technology of China, Hefei 230026, China · Institute of Advanced Technology, University of Science and Technology of China, Hefei 230088, China
Efficient determination of the effective multiplication factor (keff) is an important computational task in reactor core neutronics analysis. Physics-informed neural networks (PINNs) incorporate neutron diffusion equations and boundary conditions into network training to efficiently determine the neutron flux distribution and keff. To further improve the efficiency of keff calculations using PINNs, a Joint Initialization Physics-Informed Neural Network (JI-PINN) is proposed in this work. In this method, a low-resolution approximate solution to the K-eigenvalue problem is used to construct a joint initial state for the flux network parameters and keff, and both are then jointly optimized under physical constraints. The proposed method was validated on a two-dimensional two-group two-material case, the IAEA 2D benchmark, a two-dimensional two-group four-material case, and a three-dimensional single-group case. For these test cases, the total computational time was reduced by 25.4%, 38.2%, 49.4%, and 28.9%, respectively, while comparable solution accuracy was maintained. The occurrence of anomalous results associated with marked deviations of keff from the reference value was also reduced. The proposed method provides a more efficient and robust initialization strategy for solving neutron diffusion K-eigenvalue problem with PINNs.
Figures & tables
Fig. 1: Overall framework of JI-PINN.
Fig. 2: Geometry and material distribution of the two-dimensional two-group two-material case.
Material
g
Dg(cm)
Σa(cm−1)
νΣf(cm−1)
Σ1→2(cm−1)
Material 1
1
1.268
0.007181
0.004609
0.02767
2
0.1902
0.07047
0.08675
−
Material 2
1
1.255
0.008252
0.004602
0.02533
2
0.211
0.1003
0.1091
−
TABLE I: Material parameters of the two-dimensional two-group two-material case
Experimental Variable
Parameter Value
L0
Ttotal(s)
εk
Nh
17
4.929×10−3
797.9
6.693×10−3
25
3.974×10−3
1575.3
2.532×10−3
33
5.946×10−3
1631.0
2.730×10−3
49
7.665×10−3
1688.4
2.825×10−3
Mpre
1000
4.699×10−1
8045.4
0.133×10−3
3000
7.413×10−3
1986.1
2.93×10−3
TABLE II: Results of the Hyperparameter Analysis for the Two-Dimensional Two-Group Two-Material Case
Method
Final keff
εk
Fast Group MSE
Thermal Group MSE
Ttotal(s)
Randomly Initialized PINN
1.037±0.039
(3.141±3.512)×10−2
(1.088±1.299)×10−1
(1.634±1.835)×10−2
3309.6±366.2
R2 -PINN
1.056±0.027
(1.252±2.532)×10−2
(2.276±5.791)×10−2
(3.389±8.454)×10−3
4033.4±806.2
JI-PINN
1.066±0.001
(2.428±1.511)×10−3
(5.615±3.246)×10−3
(8.330±4.892)×10−4
2467.7±931.5
TABLE III: Statistical Results of the Random Seed Experiments for the Two-Dimensional Two-Group Two-Material Case
Fig. 3: Solution results and computational time for different random seeds: (a) final keff ; (b) total computational time; (c) MSE of the neutron flux for the fast and thermal groups.
Fig. 4: Geometric layout of the IAEA 2D benchmark.
Region
g
Dg(cm)
Σa(cm−1)
νΣf(cm−1)
Σ1→2(cm−1)
Ω1
1
1.5
0.010
0.000
0.02
2
0.4
0.080
0.135
−
Ω2
1
1.5
0.010
0.000
0.02
2
0.4
0.085
0.135
−
Ω3
1
1.5
0.010
0.000
0.02
2
0.4
0.130
0.135
−
TABLE IV: Material parameters of the IAEA 2D Benchmark
Initialization Method
keff(0)
keff
L0
εk
Ttotal(s)
Random Initialization
1
1.029586
6.420×105
3.48×10−4
7398.5
1.029240
1.030006
6.420×105
6.02×10−5
7106.3
Flux Initialization
1
1.030104
8.858×103
1.55×10−4
5194.9
1.029240
1.029367
8.842×103
5.61×10−4
4573.0
TABLE V: Computational Results for the IAEA 2D Benchmark under Different Initialization Configurations
Fig. 5: Convergence histories of keff under different initialization configurations.
Fig. 6: Geometry and material distribution of the two-dimensional two-group four-material case.
Material
g
Dg(cm)
Σa(cm−1)
νΣf(cm−1)
Σ1→2(cm−1)
Material 1
1
1.255
0.008252
0.004602
0.02533
2
0.2110
0.100300
0.10910
−
Material 2
1
1.268
0.007181
0.004609
0.02767
2
0.1902
0.070470
0.08675
−
Material 3
1
1.259
0.008002
0.004663
0.02617
2
0.2091
0.083440
0.10210
−
TABLE VI: Material parameters of the two-dimensional two-group four-material case
Method
keff
εk
Ttotal(s)
Randomly Initialized PINN
0.796108
4.88×10−4
2757.6
R2 -PINN
0.795609
1.39×10−4
4578.0
JI-PINN
0.793601
2.66×10−3
1394.9
TABLE VII: Computational Results for the Two-Dimensional Two-Group Four-Material Case
Fig. 7: Convergence history of keff for the two-dimensional two-group four-material case.
Method
keff
εk
Flux MSE
Ttotal(s)
Randomly Initialized PINN
0.9505335
5.62×10−4
6.15×10−7
697.4
R2 -PINN
0.950510
5.37×10−4
8.17×10−7
612.6
JI-PINN
0.950660
6.95×10−4
9.49×10−7
496.1
TABLE VIII: Computational Results for the Three-Dimensional Single-Group Cubic Case
Fig. 8: Geometry of the three-dimensional single-group cubic case.
Fig. 9: Convergence history of keff for the three-dimensional single-group cubic case.
Neural PDE solvers use trainable models to approximate solutions or solution operators for partial differential equations, often with physical constraints.
Physics-Informed Neural Networks (PINNs) embed governing equations into deep learning, but enforce them only through loss residuals, leaving highly oscillatory wave behavior to be discovered by optimization. As a result, methods that achieve relative L2 errors below 10−3 on standard manufactured Helmholtz benchmarks can fail on practical radiation problems involving singular excitations, absorbing boundaries, and wave fields spanning tens of wavelengths. Architectural physics embedding addresses this limitation by factorizing the field into analytically derived oscillatory kernels and learnable envelopes. However, the kernel dictionary must be manually constructed and scales with the number of elementary units, growing exponentially with the depth of hierarchically structured systems such as antenna arrays and metasurfaces. We propose PE-EK-PINN (Physics Embedded with Evolving Kernels), which treats physics kernels as reusable learned representations rather than fixed analytical inputs. A converged subsystem field is frozen and promoted to an evolved kernel, whose transformed copies are reused to represent higher-level configurations without deriving new governing equations. The resulting hierarchy makes the peak number of active kernels independent of system size and reduces cumulative training cost from O(N) to O(logN). Experiments on dipole arrays, composite line-source geometries, and cross arrays demonstrate the dramatic training cost reduction, while achieving a reduced or comparable relative L2 error. One notable example is PE-EK-PINN solves a 256-dipole array more than 30 times faster than direct PE-PINN.
Huiwen Zhang, Feng Ye, Chu Ma
Department of Electrical & Computer Engineering Madison, WI, 53706, USA
Severe accident analysis using system-level codes such as MELCOR is indispensable for nuclear safety assessment, yet the computational cost of repeated simulations poses a significant bottleneck for parametric studies and uncertainty quantification. Existing surrogate models accelerate these analyses but depend on large volumes of simulation data, while physics-informed neural networks (PINNs) enable data-free training but must be retrained for every change in problem parameters. This study addresses both limitations by developing the Parameterized PINNs coupled with FDM (P2F) method, a node-assigned hybrid framework for MELCOR's Control Volume Hydrodynamics/Flow Path (CVH/FP) module. In the P2F method, a parameterized Node-Assigned PINN (NA-PINN) accepts the water-level difference, initial velocity, and time as inputs, learning a solution manifold so that a single trained network serves as a data-free surrogate for the momentum conservation equation across all flow paths without retraining. This PINN is coupled with a finite difference method (FDM) solver that advances the mass conservation equation at each time step, ensuring exact discrete mass conservation while replacing the iterative nonlinear momentum solve with a single forward pass. Verification on a six-tank gravity-driven draining scenario yields a water level mean absolute error of 7.85×10−5 m and a velocity mean absolute error of 3.21×10−3 m/s under the nominal condition with Δt=1.0 s. The framework maintains consistent accuracy across time steps ranging from 0.2 to 1.0 s and generalizes to five distinct initial conditions, all without retraining or simulation data. This work introduces a numerical coupling methodology for integrating parameterized PINNs with FDM within a nuclear thermal-hydraulic system code framework.
Jeesuk Shin, Donggyun Seo, Sihyeong Yu +1
Division of Advanced Nuclear Engineering, Pohang University of Science and Technology, Pohang-si, Republic of Korea · Mueunjae School of Undergraduate Studies, Pohang University of Science and Technology, Pohang-si, Republic of Korea · Department of Quantum System Engineering, Jeonbuk National University, Jeonju-si, Republic of Korea
Amortizing physics-informed neural networks (PINNs) across related PDEs requires describing each equation to a reusable solver. Coefficient vectors encode numerical parameters in predefined slots, leaving operator and cross-field assignments implicit. We make these relationships explicit in an operator graph, with nodes for fields, derivatives, terms, and residuals and coefficients retained as term attributes. A graph hypernetwork generates diagonal codes that initialize a meta-trained factorized PINN for each target equation. Meta-training and target-specific adaptation use governing equations and prescribed conditions without solution labels. We compare coefficient-vector, DeepSets-based term-set, and graph conditioning by solution accuracy within a fixed adaptation budget. In scalar convection-diffusion-reaction problems, both term-based descriptors improve high-reaction accuracy, with similar performance. In two-field Fisher-KPP, meta-training sees uncoupled and one-way systems; after 3,000 adaptation steps on unseen two-way coupling, the graph's mean final error is 35.7% below the term set and 67.7% below the coefficient vector. In a fixed-structure capacitively coupled plasma model, the coefficient vector performs best. These results support extending coefficient conditioning with explicit equation relationships for physics-based solver adaptation.