arXiv:2409.02154astro-ph.IMastro-ph.CO2024-09中稿 · publication in A&A被引 6

用机器学习加速宇宙模拟,自动修正误差,更准更快。

COmoving Computer Acceleration (COCA): $N$-body simulations in an emulated frame of reference

  • 在机器学习预测的参考系中解物理方程,误差可自动修正。
  • 减少近90%力计算次数,仍保持高精度密度与速度场结果。
  • 适合需要高可靠性的宇宙学模拟,尤其超越训练数据场景。

N体模拟计算成本高昂,基于机器学习的代理模型虽快但存在可信度问题,因当前方法无法纠正潜在的模拟误差。为此,我们提出共动计算机加速(COCA),一种将机器学习与真实N体模拟结合的混合框架。该方法在机器学习预测的参考系中求解正确的运动方程,使任何模拟误差均能被设计上自动纠正。其本质是求解粒子轨迹相对于机器学习解的微扰,计算成本远低于完整求解,且随着力评估次数增加,结果可保证收敛至真实值。该方法适用于任意机器学习算法和N体模拟器,在粒子网格宇宙学模拟中,以卷积神经网络预测参考系,并将时间依赖性作为额外输入参数。实验表明,COCA显著降低粒子轨迹的模拟误差,所需力评估次数远少于无机器学习的完整模拟,同时获得准确的最终密度场和速度场,计算开销更低。该方法对训练数据外样本也表现出强鲁棒性。相较于使用相同资源直接模拟拉格朗日位移场,COCA通过误差纠正实现更高精度。该方法通过跳过不必要的力计算,既降低成本又确保物理方程正确性并修复机器学习误差。

原文摘要 · Abstract (English)

$N$-body simulations are computationally expensive, so machine-learning (ML)-based emulation techniques have emerged as a way to increase their speed. Although fast, surrogate models have limited trustworthiness due to potentially substantial emulation errors that current approaches cannot correct for. To alleviate this problem, we introduce COmoving Computer Acceleration (COCA), a hybrid framework interfacing ML with an $N$-body simulator. The correct physical equations of motion are solved in an emulated frame of reference, so that any emulation error is corrected by design. This approach corresponds to solving for the perturbation of particle trajectories around the machine-learnt solution, which is computationally cheaper than obtaining the full solution, yet is guaranteed to converge to the truth as one increases the number of force evaluations. Although applicable to any ML algorithm and $N$-body simulator, this approach is assessed in the particular case of particle-mesh cosmological simulations in a frame of reference predicted by a convolutional neural network, where the time dependence is encoded as an additional input parameter to the network. COCA efficiently reduces emulation errors in particle trajectories, requiring far fewer force evaluations than running the corresponding simulation without ML. We obtain accurate final density and velocity fields for a reduced computational budget. We demonstrate that this method shows robustness when applied to examples outside the range of the training data. When compared to the direct emulation of the Lagrangian displacement field using the same training resources, COCA's ability to correct emulation errors results in more accurate predictions. COCA makes $N$-body simulations cheaper by skipping unnecessary force evaluations, while still solving the correct equations of motion and correcting for emulation errors made by ML.

N体模拟机器学习宇宙学误差修正

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