用图神经网络加速太阳风-磁层模拟,实现秒级高精度预测。
Deterministic and probabilistic neural surrogates of global hybrid-Vlasov simulations
- 基于图神经网络构建确定性与概率性预测模型,学习电磁场和离子分布演化。
- 模型预测相关系数超0.95,单卡运行速度比原模拟快100倍以上。
- 适合需要快速模拟和不确定性评估的太空天气研究者使用。
混合-Vlasov模拟能解析太阳风-磁层相互作用中的离子动力学效应,但即使在5维(2空间+3速度)配置下也极为耗时。本文利用图神经网络(GNN),基于四个由稳定太阳风驱动的5维Vlasiator模拟,学习近地空间电磁场及离子速度分布低阶矩的时空演化。各模拟中上游离子数密度系统性变化,而网格间距保持不变,以扫描离子惯性长度与网格尺寸之比。采用在67万节点二维空间网格上运行的GNN,构建了确定性预测模型(Graph-FM)与基于潜在变量的概率性集合预测模型(Graph-EFM)。通过引入散度惩罚项确保磁场无散,并以连续秩概率评分优化集合校准。相较于在100个CPU上的原始模拟,训练后模型在单张GPU上的每步耗时降低超过两个数量级。50秒预报提前量下,多数场量相关系数高于0.95;但在5维设置中呈退化(近零)分布的场量预测难度较高。集合预测仍存在欠分散问题,展布-技巧比约为0.2–0.3,提供结构化的相对不确定性估计。结果表明,GNN为混合-Vlasov建模中的快速集合生成提供了可行框架,并指明未来研究方向。
原文摘要 · Abstract (English)
Hybrid-Vlasov simulations resolve ion-kinetic effects in the solar wind-magnetosphere interaction, but even 5D (2D + 3V) configurations are computationally expensive. We show that graph-based machine learning emulators can learn the spatiotemporal evolution of electromagnetic fields and lower-order moments of the ion velocity distribution function in near-Earth space from four 5D Vlasiator runs, each driven by steady solar wind conditions. The upstream ion number density is systematically varied between the runs, while the grid spacing is held constant, to scan the ratio of ion inertial length to grid size. Using a graph neural network (GNN) operating on the 2D spatial simulation grid comprising 670k cells, we demonstrate that both a deterministic forecasting model (Graph-FM) and a probabilistic ensemble forecasting model (Graph-EFM) based on a latent variable formulation produce accurate predictions of future plasma states. A divergence penalty is incorporated to encourage divergence-freeness in the magnetic fields. For the probabilistic model, a continuous ranked probability score objective is added to improve the calibration of the ensemble forecasts. In terms of wall time per output step, the trained emulators run over two orders of magnitude faster on a single GPU than the Vlasiator simulations on 100 CPUs. Most forecasted fields have Pearson correlations above 0.95 at 50 seconds lead time. Fields that exhibit degenerate (near-zero) distributions in the 5D setting are more challenging for the emulator to keep well correlated. The ensemble forecasts remain underdispersive, with spread-skill ratios of approximately 0.2-0.3, and thus provide spatially structured relative uncertainty estimates. Overall, these results demonstrate that GNNs provide a viable framework for rapid ensemble generation in hybrid-Vlasov modeling and highlight promising directions for future work.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。