arXiv:2502.06853v1 跨越类型: cross 摘要：在过去十年中，核工程领域对机器学习（ML）的研究有了显著增长。随着许多方法达到成熟阶段，下一阶段的研究将确定ML模型在生产环境中的可行性和实用性。许多用于反应堆设计和评估的代码主要是用Fortran语言编写的，这与TensorFlow训练的ML模型不完全兼容。本研究提供了一个框架，在Fortran中实现深度神经网络（DNNs）和贝叶斯神经网络（BNNs），允许在无需TensorFlow的C API、Python运行时或ONNX转换的情况下进行本地执行。该框架设计用于使用便捷和计算效率高，可以在任何Fortran代码中实现，并支持迭代求解器和通过聚类或BNN进行不确定性量化（UQ）。通过一个包含两个输入和一个输出的噪声正弦波测试案例，进行了验证，将基于Fortran的预测与TensorFlow的预测进行了比较。DNN预测几乎无差异，实现了19.6倍的加速，而BNN预测则有轻微的分歧，这可能是由于随机数生成的不同。BNN推理实现了8.0倍的加速。该方法随后在预测关键热流（CHF）的核相关问题上进行了进一步验证，表现出类似的行为，同时实现了显著的计算效益。还讨论了该框架成功集成到CTF热力代码中的情况，概述了其实际用途。总体而言，该框架证明了在Fortran中有效实现DNN和BNN模型推理的能力，使基于ML的方法在实际核应用中的持续研究成为可能。