Saved in:
Bibliographic Details
Main Authors: Whewell, Ben, Gibson, Nathan, Khatiwada, Ajeeta
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2605.27895
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911723547525120
author Whewell, Ben
Gibson, Nathan
Khatiwada, Ajeeta
author_facet Whewell, Ben
Gibson, Nathan
Khatiwada, Ajeeta
contents The optimization of energy group structures is integral to ensure the accuracy of multigroup neutron transport calculations. This works introduces the use of reinforcement learning (RL) with surrogate modeling to optimize the group structure for one-dimensional spherical k-criticality problems. The proximal policy optimization (PPO) RL algorithm is modified to be used with energy grid structures, rewarding accurate group structures while favoring fewer energy groups. This method starts from a high-fidelity energy grid and remove energy bounds until reaching a target energy structure. The RL agent identify which bounds are important for the final group structure, which prevent it being stuck in local minima without limiting the initial group structure. Neural network surrogate models that incorporate energy, material, and spatial information are used for evaluating energy grid structures without requiring full transport simulations. This alleviates the computational constraint commonly used in other group structure optimization problems in addition to accelerating the RL training process. Applied to Godiva and BeRP ball problems, the RL constructed group structures outperform commonly used group structures. The RL group structure optimization method is also shown to perform similar to the hierarchical agglomeration approach but offers more flexibility.
format Preprint
id arxiv_https___arxiv_org_abs_2605_27895
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Application of Reinforcement Learning for Multigroup Energy Grid Optimization for Neutron Transport Criticality Problems
Whewell, Ben
Gibson, Nathan
Khatiwada, Ajeeta
Computational Physics
The optimization of energy group structures is integral to ensure the accuracy of multigroup neutron transport calculations. This works introduces the use of reinforcement learning (RL) with surrogate modeling to optimize the group structure for one-dimensional spherical k-criticality problems. The proximal policy optimization (PPO) RL algorithm is modified to be used with energy grid structures, rewarding accurate group structures while favoring fewer energy groups. This method starts from a high-fidelity energy grid and remove energy bounds until reaching a target energy structure. The RL agent identify which bounds are important for the final group structure, which prevent it being stuck in local minima without limiting the initial group structure. Neural network surrogate models that incorporate energy, material, and spatial information are used for evaluating energy grid structures without requiring full transport simulations. This alleviates the computational constraint commonly used in other group structure optimization problems in addition to accelerating the RL training process. Applied to Godiva and BeRP ball problems, the RL constructed group structures outperform commonly used group structures. The RL group structure optimization method is also shown to perform similar to the hierarchical agglomeration approach but offers more flexibility.
title Application of Reinforcement Learning for Multigroup Energy Grid Optimization for Neutron Transport Criticality Problems
topic Computational Physics
url https://arxiv.org/abs/2605.27895