Proceedings

CFDONEval: A Comprehensive Evaluation of Operator-Learning Neural Network Models for Computational Fluid Dynamics

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 5752-5760.

https://doi.org/10.24963/ijcai.2025/640

Abstract

In this paper, we introduce CFDONEval, a comprehensive evaluation of 12 operator-learning-based neural network (ON) models to simulate 7 benchmark fluid dynamics problems. These problems cover a range of 2D scenarios, including Darcy flow, two-phase flow, Taylor-Green vortex, lid-driven cavity flow, tube flow, circular cylinder flow, and 3D periodic hill flow. For a rigorous evaluation, we establish 22 fluid dynamics datasets for these benchmark problems, 18 of which are newly generated using traditional numerical methods, such as the finite element method. Our evaluation tackles 5 key challenges: multiscale phenomena, convection dominance, long-term predictions, multiphase flows, and unstructured meshes over complex geometries. We assess computational accuracy, efficiency, and flow field visualization, offering valuable insights into the application of ON models in fluid dynamics research. Our findings show that attention-based models perform well in handling almost all challenges; models with a U-shaped structure excel in handling multiscale problems; and the NU-FNO model demonstrates the smallest relative error in L2 norm when processing nonuniform grid data. The related code, dataset, and appendix are publicly available at: https://github.com/Sysuzqs/CFDNNEval.

  1. [Alleborn et al., 1999] N. Alleborn, H. Raszillier, and F. Durst. Lid-driven cavity with heat and mass trans- port. International Journal of Heat and Mass Transfer, 42(5):833-853, 1999.
  2. [Almeida et al., 1993] G.P. Almeida, D.F.G. Durão, and M.V. Heitor. Wake flows behind two-dimensional model hills. Experimental Thermal and Fluid Science, 7(1):87- 101, 1993.
  3. [Anonymous, 2024] Anonymous. MPFBench: A large scale dataset for sciML of multi-phase-flows: Droplet and bub- ble dynamics. In Submitted to The Thirteenth Interna- tional Conference on Learning Representations, 2024. un- der review. [Brandstetter et al., 2022]
  4. J. Brandstetter, D. E. Worrall, and M. Welling. Message passing neural PDE solvers. In Inter- national Conference on Learning Representations, 2022.
  5. [Cao, 2021] S. Cao. Choose a transformer: Fourier or Galerkin. In Advances in Neural Information Processing Systems (NeurIPS 2021), volume 34, 2021.
  6. [COM, 2023] COMSOL Multiphysics ® v. 6.2, Stockholm, Sweden. AC/DC Module User's Guide, 2023.
  7. [Feng, 2006] X. Feng. Fully discrete finite element approxi- mations of the Navier-Stokes-Cahn-Hilliard diffuse inter- face model for two-phase fluid flows. SIAM Journal on Numerical Analysis, 44(3):1049-1072, 2006.
  8. [Gal and Grasselli, 2010] C. G. Gal and M. Grasselli. Asymptotic behavior of a Cahn-Hilliard-Navier-Stokes system in 2D. Annales de l'I.H.P. Analyse non linéaire, 27(1):401-436, 2010.
  9. [Guo et al., 2016] X. Guo, W. Li, and F. Iorio. Convolutional neural networks for steady flow approximation. In Pro- ceedings of the 22nd ACM SIGKDD International Con- ference on Knowledge Discovery and Data Mining, KDD '16, page 481-490, New York, NY, USA, 2016. Associa- tion for Computing Machinery.
  10. [Gupta and Brandstetter, 2022] J. K. Gupta and J. Brandstet- ter. Towards multi-spatiotemporal-scale generalized pde modeling. arXiv preprint arXiv:2209.15616, 2022.
  11. [Hao et al., 2023] Z. Hao, Z. Wang, H. Su, C. Ying, Y. Dong, S. Liu, Z. Cheng, J. Song, and J. Zhu. GNOT: A general neural operator transformer for operator learning. In An- dreas Krause, Emma Brunskill, Kyunghyun Cho, Barbara Engelhardt, Sivan Sabato, and Jonathan Scarlett, editors, Proceedings of the 40th International Conference on Ma- chine Learning, volume 202 of Proceedings of Machine Learning Research, pages 12556-12569. PMLR, 23-29 Jul 2023.
  12. [Hassan et al., 2023] S. M. S. Hassan, A. Feeney, A. Dhruv, J. Kim, Y. Suh, J. Ryu, Y. Won, and A. Chandramowlish- waran. BubbleML: A multiphase multiphysics dataset and benchmarks for machine learning. In Thirty-seventh Conference on Neural Information Processing Systems Datasets and Benchmarks Track, 2023.
  13. [Kochkov et al., 2021] D. Kochkov, J. A. Smith, A. Alieva, Q. Wang, M. P. Brenner, and S. Hoyer. Ma- chine learning-accelerated computational fluid dynam- ics. Proceedings of the National Academy of Sciences, 118(21):e2101784118, 2021.
  14. [Lesieur, 2008] M. Lesieur. Turbulence in Fluids. Fluid Me- chanics and Its Applications. Springer Netherlands, 2008.
  15. [Li et al., 2020] Z. Li, N. Kovachki, K. Azizzadenesheli, B. Liu, K. Bhattacharya, A. Stuart, and A. Anandkumar. Fourier neural operator for parametric partial differential equations. arXiv preprint arXiv:2010.08895, 2020.
  16. [Li et al., 2023] Z. Li, K. Meidani, and A. Farimani. Trans- former for partial differential equations' operator learning. Transactions on Machine Learning Research, 2023.
  17. [Liu et al., 2023] S. Liu, Z. Hao, C. Ying, H. Su, Z. Cheng, and J. Zhu. NUNO: A general framework for learn- ing parametric PDEs with non-uniform data. In Andreas Krause, Emma Brunskill, Kyunghyun Cho, Barbara En- gelhardt, Sivan Sabato, and Jonathan Scarlett, editors, Pro- ceedings of the 40th International Conference on Ma- chine Learning, volume 202 of Proceedings of Machine Learning Research, pages 21658-21671. PMLR, 23-29 Jul 2023.
  18. [Lu et al., 2021] L. Lu, P. Jin, G. Pang, Z. Zhang, and G. E. Karniadakis. Learning nonlinear operators via deeponet based on the universal approximation theorem of opera- tors. Nature machine intelligence, 3(3):218-229, 2021.
  19. [Luo et al., 2023] Y. Luo, Y. Chen, and Z. Zhang. CFD- Bench: A large-scale benchmark for machine learn- ing methods in fluid dynamics. arXiv preprint arXiv:2310.05963, 2023.
  20. [Motamedi et al., 2012] A. Motamedi, A. Pacheco-Vega, and J. R. Pacheco. Numerical analysis of a multi-row multi-column compact heat exchanger. Journal of Physics: Conference Series, 395(1):012047, nov 2012.
  21. [Navah et al., 2020] F. Navah, M. de la Llave Plata, and V. Couaillier. A high-order multiscale approach to turbu- lence for compact nodal schemes. Computer Methods in Applied Mechanics and Engineering, 363:112885, 2020.
  22. [Navah, 2024] F. Navah. Energy spectrum computation script. https://github.com/fanav/Energy Spectrum, 2024. GitHub repository. Link
  23. [Patankar, 1980] S.V. Patankar. Numerical Heat Transfer and Fluid Flow. Electro Skills Series. Hemisphere Pub- lishing Corporation, 1980.
  24. [Qin et al., 2022] Y. Qin, H. Huang, Y. Zhu, C. Liu, and S. Xu. A phase field model for mass transport with semi- permeable interfaces. Journal of Computational Physics, 464:111334, 2022.
  25. [Rahman et al., 2022] M. A. Rahman, Z. E. Ross, and K. Az- izzadenesheli. U-NO: U-shaped neural operators. arXiv preprint arXiv:2204.11127, 2022.
  26. [Rodi et al., 1995] W. Rodi, J.C. Bonnin, and T. Buchal. ER- COFTAC workshop on Data Bases and Testing of Calcu- lation Methods for Turbulent Flows., 1995.
  27. [Ronneberger et al., 2015] O. Ronneberger, P. Fischer, and T. Brox. U-Net: Convolutional networks for biomedical image segmentation. In Medical Image Computing and Computer-Assisted Intervention-MICCAI 2015: 18th In- ternational Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18, pages 234-241. Springer, 2015.
  28. [Shankar and Deshpande, 2000] P. N. Shankar and M. D. Deshpande. Fluid mechanics in the driven cavity. Annual Review of Fluid Mechanics, 32(Volume 32, 2000):93-136, 2000.
  29. [Shen et al., 2024] J. Shen, T. Marwah, and A. Talwalkar. UPS: Efficiently building foundation models for PDE solving via cross-modal adaptation. Submitted to Transac- tions on Machine Learning Research, 2024. Under review.
  30. [Takamoto et al., 2022] M. Takamoto, T. Praditia, R. Lei- teritz, D. MacKinlay, F. Alesiani, D. Pflüger, and M. Niepert. PDEBench Datasets, 2022.
  31. [Tali et al., 2024] R. Tali, A. Rabeh, C. Yang, M. Shadkhah, S. Karki, A. Upadhyaya, S. Dhakshinamoorthy, M. Saa- dati, S. Sarkar, A. Krishnamurthy, C. Hegde, A. Balu, and B. Ganapathysubramanian. Flowbench: A large scale benchmark for flow simulation over complex geometries, 2024.
  32. [Taylor and Green, 1937] G. I. Taylor and A. E. Green. Mechanism of the production of small eddies from large ones. Proceedings of the Royal Society of London. Series A -Mathematical and Physical Sciences, 158(895):499- 521, 1937.
  33. [Todd and Mays, 2004] D.K. Todd and L.W. Mays. Ground- water Hydrology. Wiley, 2004.
  34. [Toshev et al., 2024] A. Toshev, G. Galletti, F. Fritz, S. Adami, and N. Adams. Lagrangebench: A lagrangian fluid mechanics benchmarking suite. Advances in Neural Information Processing Systems, 36, 2024.
  35. [Williamson, 1996] C. H. K. Williamson. Vortex dynamics in the cylinder wake. Annual Review of Fluid Mechanics, 28:477-539, 1996.
  36. [Wu et al., 2023] H. Wu, T. Hu, H. Luo, J. Wang, and M. Long. Solving high-dimensional pdes with latent spec- tral models. In International Conference on Machine Learning, 2023.
  37. [Xiong et al., 2024] W. Xiong, X. Huang, Z. Zhang, R. Deng, P. Sun, and Y. Tian. Koopman neural operator as a mesh-free solver of non-linear partial differential equa- tions. Journal of Computational Physics, page 113194, 2024.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards