At present, traffic state prediction primarily relies on purely data-driven methods, ignoring the incorporation of physical constraints within the field of traffic flow. Taking this as a starting point, this paper endeavors to embed the physical mechanism of the traffic flow fundamental graph into the deep learning model, and proposes a physics-integrated spatiotemporal graph neural network with fundamental diagram learner (PI-STGnet) for highway traffic flow prediction. The PI-STGnet mainly comprises a semantic enhancement module (SE), a spatiotemporal correlation extraction module (ST-Block), and a fundamental diagram learner module (FD-learner). These modules are strategically designed to sequentially enhance the contextual semantic relationships of the traffic flow and speed inputs, extract intricate spatiotemporal correlations of traffic states, and adaptively learn the dynamic evolution of the physical relationships between traffic flow and speed. The real-world dataset employed to assess the performance of the PI-STGnet is sourced from the monitoring data of highway gantry sensors located in Ningde City, Fujian Province, China. The experimental results demonstrate that the proposed PI-STGnet has the advantages to extract spatiotemporal features and achieve prediction, surpassing the accuracy of cutting-edge baseline models. In summary, as an exploratory work, this paper provides a novel approach that driven by the fusion of deep learning and physical mechanisms for traffic flow prediction to a certain extent.
Read full abstract