Physics-Embedded Gaussian Process for Traffic State Estimation

TL;DR

This paper introduces a Physics-Embedded Gaussian Process (PEGP) that integrates traffic flow physics into Gaussian process models, improving traffic state estimation especially under sparse data conditions with better uncertainty calibration.

Contribution

The novel PEGP framework explicitly incorporates traffic physics into Gaussian processes, addressing limitations of previous methods by enhancing uncertainty calibration and model reliability.

Findings

PEGP outperforms non-physics baselines on real traffic datasets.

PEGP-ARZ provides more reliable estimates under sparse observations.

PEGP-LWR achieves lower errors with denser data.

Abstract

Traffic state estimation (TSE) becomes challenging when probe-vehicle penetration is low and observations are spatially sparse. Pure data-driven methods lack physical explanations and have poor generalization when observed data is sparse. In contrast, physical models have difficulty integrating uncertainties and capturing the real complexity of traffic. To bridge this gap, recent studies have explored combining them by embedding physical structure into Gaussian process. These approaches typically introduce the governing equations as soft constraints through pseudo-observations, enabling the integration of model structure within a variational framework. However, these methods rely heavily on penalty tuning and lack principled uncertainty calibration, which makes them sensitive to model mis-specification. In this work, we address these limitations by presenting a novel Physics-Embedded Gaussian Process (PEGP), designed to integrate domain knowledge with data-driven methods in traffic state estimation. Specifically, we design two multi-output kernels informed by classic traffic flow models, constructed via the explicit application of the linearized differential operator. Experiments on HighD, NGSIM show consistent improvements over non-physics baselines. PEGP-ARZ proves more reliable under sparse observation, while PEGP-LWR achieves lower errors with denser observation. Ablation study further reveals that PEGP-ARZ residuals align closely with physics and yield calibrated, interpretable uncertainty, whereas PEGP-LWR residuals are more orthogonal and produce nearly constant variance fields. This PEGP framework combines physical priors, uncertainty quantification, which can provide reliable support for TSE.