Inverse Critical Experiment Design via Gradient Optimization and a Multigroup Attention-Based Neural Network Architecture
The paper uses a neural surrogate and gradient optimization to design critical experiments that better match advanced nuclear systems.
Intelligence analysis by GPT-5.4 Mini

The authors frame critical experiment design as an inverse problem: find geometries that maximize neutronic similarity to a target system. Their approach combines an OpenMC-trained neural model, multigroup attention pooling, and gradient-based search across the design space.
The paper is like using a smart map to arrange building blocks so a test setup looks as close as possible to a real nuclear system. A computer learns which layouts match best, then quickly tries many choices until it finds better ones.
Analysis
What the paper does
The paper addresses a validation problem for advanced nuclear reactor designs and fuel concepts. It says critical experiments need high neutronic similarity to the target technology, and that similarity is measured with the correlation coefficient $c_k$. The abstract notes that a value of at least 0.9 is generally needed for an experiment to count as sufficiently similar.
The method
To design experiments in reverse, the authors train a deep neural network on OpenMC-calculated sensitivity vectors for grid-based critical experiment geometries. The model pairs a U-Net convolutional encoder-decoder with a new multigroup attention pooling layer. According to the abstract, that attention mechanism is meant to capture different spatial dependencies across sensitivity groups, and it performs better than traditional pooling while also being more interpretable.
Because the surrogate model is differentiable, the authors can use gradient-based optimization over the full combinatorial design space. In practical terms, the method directly changes the material assignment at each position in the geometry grid to search for layouts that maximize $c_k$.
Reported results
The approach is applied to validation of the TN-Americas TN-LC transportation cask with HALEU fuel, where the paper says existing critical experiment coverage is limited. The optimization procedure produces experiment geometries with $c_k$ scores of 0.97757, 0.81324, and 0.93276 for three configurations of interest. The abstract presents this as evidence that deep learning plus gradient optimization can speed up advanced nuclear technology development.
Key points
- The paper treats critical experiment design as an inverse optimization problem.
- It uses a neural surrogate trained on OpenMC sensitivity vectors for grid-based geometries.
- A multigroup attention pooling layer is introduced to capture spatial dependence in the sensitivities.
- Gradient-based optimization changes the material assignment in each grid position to maximize $c_k$.
- The method is demonstrated on TN-Americas TN-LC validation with HALEU fuel.
If the method holds up beyond the paper, it could help engineers design better validation experiments much faster than manual trial and error. The attention layer’s interpretability may also make it easier to understand why certain layouts score better.
The results are uneven across the three configurations, and one score falls below the 0.9 similarity threshold the abstract says is usually needed. The approach also depends on the quality of the surrogate model and the OpenMC training data, so it may not transfer cleanly to other experiment families.



