Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces population-aware physics-informed neural particle flow (PA-PINPF), which enhances the standard PINPF by incorporating a permutation-invariant Deep Sets representation of the entire particle population to inform transport decisions. By utilizing two distinct population encoders, PA-PINPF-State and PA-PINPF-Feature, the method captures both the particle states and local physics-informed features, leading to improved Bayesian transport performance. Experimental results demonstrate that PA-PINPF outperforms traditional particle-wise approaches, particularly with the feature-population encoding, highlighting the significance of population-level information in Bayesian updates.
Incorporating global population context into particle transport can drastically enhance Bayesian inference accuracy in complex systems.
Physics-informed neural particle flow (PINPF) learns a deterministic transport field that moves particles from a prior distribution toward a Bayesian posterior while enforcing the governing probability-evolution equation. However, the standard PINPF velocity model processes particles independently and therefore does not explicitly condition its transport decisions on the empirical particle population. This paper introduces population-aware PINPF (PA-PINPF), which augments each particle update with a permutation-invariant Deep Sets representation of the full particle set. We investigate two population encoders. PA-PINPF-State summarizes the particle states, whereas PA-PINPF-Feature summarizes the complete local physics-informed feature vectors, including particle position, pseudo-time, measurement information, likelihood values, and score information. The latter allows the population context to represent not only particle-cloud geometry, but also the population-level Bayesian transport geometry. The methods retain the original unsupervised physics-informed residual objective and require no ground-truth posterior samples during training. Experiments on range-measurement tasks and nonlinear time-difference-of-arrival posterior transport demonstrate that both population-aware variants improve over particle-wise PINPF, while feature-population encoding provides the strongest performance. These results show that population-level physics features provide useful global information for learned Bayesian particle transport.