Wasserstein Moment Nudging for Vlasov-Poisson Data Assimilation

Abstract

We introduce a continuous data assimilation method for particle-in-cell simulations of the Vlasov-Poisson equation when only hydrodynamic moments are observed. The forecast state is an empirical measure on phase space, whereas the observed fields (density, bulk velocity, and temperature) constrain only a few velocity moments and leave the velocity-space shape of the distribution undetermined. We construct the moment feedback as a Wasserstein gradient flow of a moment-mismatch functional over phase-space measures. The resulting drift acts directly on particle positions and velocities, couples the density, momentum, and energy residuals through a single variational structure, and vanishes on the entire moment-compatible set. Under the standard Wasserstein metric, the energy residual produces a position correction that grows quadratically with the particle speed, and the particle system falls outside standard well-posedness theory. Our primary formulation pairs the quadratic moment mismatch with a velocity-weighted Wasserstein metric that penalizes spatial transport at large peculiar velocity relative to the observed bulk flow, which removes this growth. A direction-split variant retains the plain metric instead. Under the same weighted metric, an alternative moment-relative-entropy functional yields an affine, shape-preserving velocity correction and explicit global moment balances for the space-inhomogeneous system. We prove that the finite-particle scheme with linear Lenard-Bernstein collisions is globally well posed. In 1D1V and 2D2V experiments with several collision models, the nudged formulations reduce bulk-velocity and temperature errors by up to two orders of magnitude relative to an unassimilated run.