JZ-FMM：基于快速多极方法的GPU原生可微分N体模拟
JZ-FMM: GPU-native differentiable N-body simulations with the Fast Multipole Method
浏览论文内容
中文总结 AI 辅助
JZ-FMM提出GPU原生可微分快速多极N体模拟，性能提升超一个数量级，并成功应用于潮汐剥离卫星的深度非线性重建问题。
中文摘要 AI 辅助
N体模拟是模拟引力系统演化的基本工具。可微分N体模拟通过将演化系统的观测与初始条件的简单信息性先验联系起来，能够解决复杂的重建问题。尽管迄今为止大多数重建工作集中于弱非线性的宇宙大尺度动力学，星系和星系团的内部动力学应蕴含关于其形成的丰富信息。在这一高度非线性区域探索重建可能性，自然需要具有小尺度力的高性能可微分N体模拟。在此，我们提出JZ-FMM，一种用于N体模拟的GPU原生可微分快速多极方法实现。我们表明，树结构、双树遍历和多极平移算子的GPU导向设计，使性能比成熟的CPU原生和混合N体代码显著提升一个数量级以上。此外，我们证明梯度能够被高效且准确地评估，因此JZ-FMM可作为未来场级重建工作的基本构建模块。我们将新代码应用于潮汐剥离卫星的重建问题，以表明深度非线性问题确实可以通过模拟梯度高效求解，尽管必须适当注意以导航复杂的优化景观。
英文摘要
N-body simulations are an essential tool for modelling the evolution of gravitating systems. Differentiable N-body simulations allow solving complicated reconstruction problems by connecting observations of evolved systems to simple informative priors on their initial conditions. While so far most reconstruction efforts focus on mildly non-linear large scale dynamics, the internal dynamics of galaxies and clusters should exhibit rich information about their formation. The exploration of reconstruction possibilities in this highly non-linear regime naturally requires highly performant differentiable N-body simulations with small scale forces. Here, we present JZ-FMM, a GPU-native differentiable implementation of the fast multipole method for N-body simulations. We show that the GPU oriented design of tree structure, dual tree traversal and multipole translation operators improves performance over established CPU-native and hybrid N-body codes by notably more than an order of magnitude. Further, we show that gradients can be evaluated efficiently and accurately so that JZ-FMM can be used as an essential building block in future field-level reconstruction efforts. We apply the new code to the reconstruction problem of a tidally stripped satellite to show that deeply non-linear problems can indeed be solved efficiently with simulation gradients, although appropriate care must be taken to navigate the complicated optimization landscape.
发表机构
- University of Vienna（维也纳大学）
机构由 AI 辅助整理，请以论文原文为准。
