arXiv:2606.06866cs.LGnucl-th2026-06中稿 · ICCS 2026

用复数乘积单元改进GRU,提升原子核质量预测精度

Product units in gated recurrent units improve nuclear-mass prediction

论文配图:Product units in gated recurrent units improve nuclear-mass prediction
图 1 · 摘自论文原文
  • 在GRU中引入乘法交互与乘积单元,增强序列建模能力
  • 插值误差仅0.227 MeV,外推误差0.179 MeV,优于现有模型
  • 适合需要高精度核质量预测的物理研究者使用

利用机器学习预测原子核质量可补充理论模型,推动对核素图中未知区域的探索。本文提出基于门控循环单元(GRU)的机器学习方法,通过在循环单元中融合乘法交互与乘积单元变换,显著提升核质量预测性能。计算在复数域进行,以联合捕捉振幅与相位动态。基于原子质量评价(AME2016和AME2020)的插值与时间外推任务中,复数加法-乘法乘积单元门控循环单元(AM-PU-GRU)模型持续取得最低预测误差,插值均方根误差为0.227 ± 0.004 MeV,外推均方根误差为0.179 ± 0.015 MeV。该结果超越其他先进机器学习模型,并优于实数域GRU基线及乘积单元消融变体,且对WS4和SEMF等不同理论先验保持鲁棒性。研究确立了复数域乘积单元循环网络作为序列型核质量预测的新基准。

原文摘要 · Abstract (English)

The prediction of masses of atomic nuclei using machine learning can complement theoretical models and advance the exploration of poorly known domains of the nuclear chart. We propose a machine learning technique based on gated recurrent units (GRU), which have demonstrated competitive performance in nuclear-mass prediction by exploiting long-term dependencies. By integrating multiplicative interactions and product-unit transformations within recurrent units, we report significant improvements in nuclear-mass prediction. Computations are performed in the complex domain to jointly capture amplitude and phase dynamics. For interpolation and temporal-extrapolation tasks based on the atomic mass evaluation (AME2016 and AME2020), the complex additive-multiplicative product-unit gated recurrent unit (AM-PU-GRU) model consistently achieves the lowest prediction errors, with an interpolation RMSE of 0.227 $\pm$ 0.004 MeV and an extrapolation RMSE of 0.179 $\pm$ 0.015 MeV. These results surpass other state-of-the-art machine learning models and also outperform the real-valued GRU baseline and product-unit ablation variants, while remaining robust to different theoretical priors, including WS4 and SEMF. Our findings establish complex-valued product-unit recurrent networks as a new benchmark for sequence-based nuclear-mass prediction.

核物理机器学习序列建模复数网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。