用5维神经网络模拟等离子体湍流,大幅降低计算成本且保持物理准确性。
GyroSwin: 5D Surrogates for Gyrokinetic Plasma Turbulence Simulations
- 将视觉变换器扩展至5维,融合电势与分布函数的跨维度交互
- 预测热通量精度超越传统简化模型,计算成本降低1000倍
- 适合等离子体物理、核聚变仿真与高性能计算领域研究者
核聚变是实现可靠可持续能源的关键,但等离子体湍流严重影响约束性能,制约下一代反应堆设计。该过程由非线性陀螺动力学方程描述,需演化5维分布函数,计算代价极高。现有降阶模型虽降低成本,却忽略全5维动力学中的非线性效应。本文提出首个可扩展的5维神经代理模型GyroSwin,首次实现对5维非线性陀螺动力学仿真的高效建模,准确捕捉湍流动能级联,并显著提升热输运预测能力。GyroSwin(i)将层级视觉变换器扩展至5维,(ii)引入交叉注意力与融合模块,实现3维电势场与5维分布函数间的潜在交互,(iii)借鉴非线性物理思想进行通道级模式分离。实验表明,其在热通量预测上优于广泛使用的降阶数值方法,捕获湍流动能级联,同时将完整非线性陀螺动力学的计算成本降低三个数量级,且结果具备物理可验证性。模型在最高达十亿参数规模下展现良好可扩展性,为构建可扩展的陀螺动力学神经代理模型开辟道路。
原文摘要 · Abstract (English)
Nuclear fusion plays a pivotal role in the quest for reliable and sustainable energy production. A major roadblock to viable fusion power is understanding plasma turbulence, which significantly impairs plasma confinement, and is vital for next-generation reactor design. Plasma turbulence is governed by the nonlinear gyrokinetic equation, which evolves a 5D distribution function over time. Due to its high computational cost, reduced-order models are often employed in practice to approximate turbulent transport of energy. However, they omit nonlinear effects unique to the full 5D dynamics. To tackle this, we introduce GyroSwin, the first scalable 5D neural surrogate that can model 5D nonlinear gyrokinetic simulations, thereby capturing the physical phenomena neglected by reduced models, while providing accurate estimates of turbulent heat transport. GyroSwin (i) extends hierarchical Vision Transformers to 5D, (ii) introduces cross-attention and integration modules for latent 3D$\leftrightarrow$5D interactions between electrostatic potential fields and the distribution function, and (iii) performs channelwise mode separation inspired by nonlinear physics. We demonstrate that GyroSwin outperforms widely used reduced numerics on heat flux prediction, captures the turbulent energy cascade, and reduces the cost of fully resolved nonlinear gyrokinetics by three orders of magnitude while remaining physically verifiable. GyroSwin shows promising scaling laws, tested up to one billion parameters, paving the way for scalable neural surrogates for gyrokinetic simulations of plasma turbulence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。