Fourier neural operator for learning solutions to macroscopic traffic flow models: Application to the forward and inverse problems

Abstract

Deep learning methods are emerging as popular computational tools for solving forward and inverse problems in traffic flow. In this paper, we study a neural operator framework for learning solutions to nonlinear hyperbolic partial differential equations with applications in macroscopic traffic flow models. In this framework, an operator is trained to map heterogeneous and sparse traffic input data to the complete macroscopic traffic state in a supervised learning setting. We chose a physics-informed Fourier neural operator ($\pi$-FNO) as the operator, where an additional physics loss based on a discrete conservation law regularizes the problem during training to improve the shock predictions. We also propose to use training data generated from random piecewise constant input data to systematically capture the shock and rarefied solutions. From experiments using the LWR traffic flow model, we found superior accuracy in predicting the density dynamics of a ring-road network and urban signalized road. We also found that the operator can be trained using simple traffic density dynamics, e.g., consisting of $2-3$ vehicle queues and $1-2$ traffic signal cycles, and it can predict density dynamics for heterogeneous vehicle queue distributions and multiple traffic signal cycles $(\ge 2)$ with an acceptable error. The extrapolation error grew sub-linearly with input complexity for a proper choice of the model architecture and training data. Adding a physics regularizer aided in learning long-term traffic density dynamics, especially for problems with periodic boundary data.

References in corpus (12)

• DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
• A comprehensive study of non-adaptive and residual-based adaptive sampling for physics-informed neural networks
• A Physics-Informed Deep Learning Paradigm for Car-Following Models
• Multipole Graph Neural Operator for Parametric Partial Differential Equations
• Macroscopic Traffic Flow Modeling with Physics Regularized Gaussian Process: A New Insight into Machine Learning Applications
• A Generative Car-following Model Conditioned On Driving Styles
• Physics-Informed Deep Learning For Traffic State Estimation: A Survey and the Outlook
• Short-term traffic prediction using physics-aware neural networks
• Generic bounds on the approximation error for physics-informed (and) operator learning
• Learning Traffic Speed Dynamics from Visualizations
• Deep Operator Learning Lessens the Curse of Dimensionality for PDEs
• Learning-based solutions to nonlinear hyperbolic PDEs: Empirical insights on generalization errors

Cited by in corpus (3)

• GoRINNs: Godunov-Riemann Informed Neural Networks for Learning Hyperbolic Conservation Laws
• Spatiotemporal Implicit Neural Representation as a Generalized Traffic Data Learner
• Physics-informed Partitioned Coupled Neural Operator for Complex Networks