Abstract
This paper proposes the TrafficFlowGAN, a physics-informed flow based generative adversarial network (GAN), for uncertainty quantification (UQ) of dynamical systems. TrafficFlowGAN adopts a normalizing flow model as the generator to explicitly estimate the data likelihood. This flow model is trained to maximize the data likelihood and to generate synthetic data that can fool a convolutional discriminator. We further regularize this training process using prior physics information, so-called physics-informed deep learning (PIDL). To the best of our knowledge, we are the first to propose an integration of normalizing flow, GAN and PIDL for the UQ problems. We take the traffic state estimation (TSE), which aims to estimate the traffic variables (e.g. traffic density and velocity) using partially observed data, as an example to demonstrate the performance of our proposed model. We conduct numerical experiments where the proposed model is applied to learn the solutions of stochastic differential equations. The results demonstrate the robustness and accuracy of the proposed model, together with the ability to learn a machine learning surrogate model. We also test it on a real-world dataset, the Next Generation SIMulation (NGSIM), to show that the proposed TrafficFlowGAN can outperform the baselines, including the pure flow model, the physics-informed flow model, and the flow based GAN model. Source code and data are available at https://github.com/ZhaobinMo/TrafficFlowGAN.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Arjovsky, M., Chintala, S., Bottou, L.: Wasserstein generative adversarial networks. In: International Conference on Machine Learning, pp. 214–223. PMLR (2017)
Aw, A., Rascle, M.: Resurrection of “second order" models of traffic flow. SIAM J. Appl. Math. 60(3), 916–938 (2000)
Brehmer, J., Cranmer, K.: Flows for simultaneous manifold learning and density estimation. Adv. Neural Inf. Process. Syst. 33, 442–453 (2020)
Council, N.R., et al.: Assessing the Reliability of Complex Models: Mathematical and Statistical Foundations of Verification, Validation, and Uncertainty Quantification. National Academies Press, Washington, DC (2012)
Daw, A., Maruf, M., Karpatne, A.: Pid-gan: a gan framework based on a physics-informed discriminator for uncertainty quantification with physics. In: Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining, pp. 237–247 (2021)
Dinh, L., Sohl-Dickstein, J., Bengio, S.: Density estimation using real nvp. arXiv preprint arXiv:1605.08803 (2016)
Goodfellow, I., et al.: Generative adversarial nets. Adv. Neural Inf. Process. Syst. 27 (2014)
Grover, A., Chute, C., Shu, R., Cao, Z., Ermon, S.: Alignflow: cycle consistent learning from multiple domains via normalizing flows. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 4028–4035 (2020)
Grover, A., Dhar, M., Ermon, S.: Flow-gan: combining maximum likelihood and adversarial learning in generative models. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 32 (2018)
Guo, L., Wu, H., Zhou, T.: Normalizing field flows: solving forward and inverse stochastic differential equations using physics-informed flow models. arXiv preprint arXiv:2108.12956 (2021)
Lighthill, M.J., Whitham, G.B.: On kinematic waves II: a theory of traffic flow on long crowded roads. Proc. Roy. Soc. Lond. Ser. A. Math. Phys. Sci. 229(1178), 317–345 (1955)
Mo, Z., Di, X.: Uncertainty quantification of car-following behaviors: physics-informed generative adversarial networks. In: The 28th ACM SIGKDD in Conjunction with the 11th International Workshop on Urban Computing (UrbComp2022) (2022)
Mo, Z., Fu, Y., Di, X.: Quantifying uncertainty in traffic state estimation using generative adversarial networks. In: 2022 IEEE 25th International Conference on Intelligent Transportation Systems (ITSC), pp. 2769–2774. IEEE (2022)
Mo, Z., Shi, R., Di, X.: A physics-informed deep learning paradigm for car-following models. Transp. Res. Part C: Emerg. Technol. 130, 103240 (2021)
Raissi, M.: Deep hidden physics models: deep learning of nonlinear partial differential equations. J. Mach. Learn. Res. 19(1), 932–955 (2018)
Seo, T., Bayen, A.M.: Traffic state estimation method with efficient data fusion based on the aw-rascle-zhang model. In: 2017 IEEE 20th International Conference on Intelligent Transportation Systems (ITSC), pp. 1–6. IEEE (2017)
Shi, R., Mo, Z., Di, X.: Physics-informed deep learning for traffic state estimation: a hybrid paradigm informed by second-order traffic models. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 540–547 (2021)
Shi, R., Mo, Z., Huang, K., Di, X., Du, Q.: A physics-informed deep learning paradigm for traffic state and fundamental diagram estimation. IEEE Trans. Intell. Transp. Syst. 23, 11688–11698 (2021)
Siddani, B., Balachandar, S., Moore, W.C., Yang, Y., Fang, R.: Machine learning for physics-informed generation of dispersed multiphase flow using generative adversarial networks. Theor. Comput. Fluid Dyn. 35(6), 807–830 (2021). https://doi.org/10.1007/s00162-021-00593-9
Theis, L., Oord, A.V.D., Bethge, M.: A note on the evaluation of generative models. arXiv preprint arXiv:1511.01844 (2015)
Yang, L., Zhang, D., Karniadakis, G.E.: Physics-informed generative adversarial networks for stochastic differential equations. SIAM J. Sci. Comput. 42(1), A292–A317 (2020)
Yang, Y., Perdikaris, P.: Adversarial uncertainty quantification in physics-informed neural networks. J. Comput. Phys. 394, 136–152 (2019)
Zang, C., Wang, F.: Moflow: an invertible flow model for generating molecular graphs. In: Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (2020)
Acknowledgements
This work is sponsored by NSF under CPS-2038984.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
1 Electronic supplementary material
Below is the link to the electronic supplementary material.
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Mo, Z., Fu, Y., Xu, D., Di, X. (2023). TrafficFlowGAN: Physics-Informed Flow Based Generative Adversarial Network for Uncertainty Quantification. In: Amini, MR., Canu, S., Fischer, A., Guns, T., Kralj Novak, P., Tsoumakas, G. (eds) Machine Learning and Knowledge Discovery in Databases. ECML PKDD 2022. Lecture Notes in Computer Science(), vol 13715. Springer, Cham. https://doi.org/10.1007/978-3-031-26409-2_20
Download citation
DOI: https://doi.org/10.1007/978-3-031-26409-2_20
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-26408-5
Online ISBN: 978-3-031-26409-2
eBook Packages: Computer ScienceComputer Science (R0)
