arXiv:2411.03389v3 通告类型: 替换-交叉 摘要: 域分解是一种用于减少大型中子传输问题中内存开销的技术。目前，这些域的最优负载均衡处理器分配通常是通过小规模问题仿真来确定的，这可能会占用研究人员大量时间，并且每次更改问题输入时都需要重新进行。我们提出了一种具有独特3D输入嵌入的Transformer模型，以及专为域分解中子传输问题设计的输入表示，该模型可以预测由小规模仿真生成的子域计算负载。我们展示了在域分解的小模块反应堆(SMR)仿真上训练的模型，可以实现98.2%的准确率，并完全省去了小规模仿真步骤。还讨论了该模型在不同燃料组件、其他问题几何形状以及仿真参数变化时的稳健性测试。