Deterministic and probabilistic neural surrogates of global hybrid-Vlasov simulations
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.
Code (0)
등록된 구현이 없습니다.
Tasks
Graph Neural NetworkSimilar Papers 제목 키워드 기반
P3D: Scalable Neural Surrogates for High-Resolution 3D Physics Simulations with Global Context
We present a scalable framework for learning deterministic and probabilistic neural surrogates for high-resolution 3D physics simulations. We introduce a hybrid CNN-Transformer backbone architecture targeted for 3D physi…
Graph-based Neural Space Weather Forecasting
Accurate space weather forecasting is crucial for protecting our increasingly digital infrastructure. Hybrid-Vlasov models, like Vlasiator, offer physical realism beyond that of current operational systems, but are too c…
Weather ForecastingIntegrating Physics-Based and Data-Driven Approaches for Probabilistic Building Energy Modeling
Building energy modeling is a key tool for optimizing the performance of building energy systems. Historically, a wide spectrum of methods has been explored -- ranging from conventional physics-based models to purely dat…
Micro-Macro Tensor Neural Surrogates for Uncertainty Quantification in Collisional Plasma
Plasma kinetic equations exhibit pronounced sensitivity to microscopic perturbations in model parameters and data, making reliable and efficient uncertainty quantification (UQ) essential for predictive simulations. Howev…
Computational EfficiencyEfficient Hyperparameter Optimization of Deep Learning Algorithms Using Deterministic RBF Surrogates
Automatically searching for optimal hyperparameter configurations is of crucial importance for applying deep learning algorithms in practice. Recently, Bayesian optimization has been proposed for optimizing hyperparamete…
Bayesian OptimizationGaussian ProcessesHyperparameter OptimizationSMAC+1