ML-PDE: A Framework for a Machine Learning Enhanced PDE Solver

ORAL

Abstract

Simulation of turbulent flows at high Reynolds number is a computationally challenging task relevant to a large number of engineering and scientific applications in diverse fields such as climate science, aerodynamics, and combustion. Turbulent flows are typically modeled by the Navier-Stokes equations. Direct Numerical Simulation (DNS) of the Navier-Stokes equations with sufficient numerical resolution to capture all the relevant scales of the turbulent motions can be prohibitively expensive. Simulation at lower-resolution on a coarse-grid introduces significant errors. We introduce a machine learning (ML) technique based on a deep neural network architecture that corrects the numerical errors induced by a coarse-grid simulation of turbulent flows at high-Reynolds numbers, while simultaneously recovering an estimate of the high-resolution fields. Our proposed simulation strategy is a hybrid ML-PDE solver that is capable of obtaining a meaningful high-resolution solution trajectory while solving the system PDE at a lower resolution.

*This research used resources of the National Energy Research Scientific Computing Center (NERSC), a U.S. Department of Energy Office of Science User Facility operated under Contract No. DE-AC02-05CH11231.

Presenters

  • Jaideep Pathak

    • Lawrence Berkeley National Laboratory

Authors

  • Jaideep Pathak

    • Lawrence Berkeley National Laboratory
  • Mustafa Mustafa

    • Lawrence Berkeley National Laboratory
  • Karthik Kashinath

    • Lawrence Berkeley National Laboratory
  • Emmanuel Motheau

    • Lawrence Berkeley National Laboratory
  • Thorsten Kurth

    • Nvidia Corporation
  • Marcus Day

    • Lawrence Berkeley National Laboratory