Bridging simulation and deep learning - convolutional neural networks on unstructured grids

ORAL

Abstract

We develop a novel method for efficiently deploying Convolutional Neural Networks (CNNs) on arbitrary unstructured grids. Unstructured grids have been the major workhorse for PDE solvers, while CNNs have been the predominant neural network architecture in deep learning for processing spatial data. Recent works have shown successes in utilizing CNN-based deep neural networks for better modeling of physical systems (e.g., turbulence modeling), and acceleration of solutions to PDEs. However regular CNN framework operates under regular grids and cannot be easily incorporated into PDE solvers that operate on unstructured grids (e.g., FEM, DG, FV). Natively performing CNN-based deep learning in the unstructured grid domain for simulation allows for smooth integration between physical simulations and deep learning based physical models.

*‘Max’ Chiyu Jiang is supported by the National Energy Research Scientific Computer (NERSC) Center summer internship program at Lawrence Berkeley National Laboratory. Prabhat and Karthik Kashinath are partly supported by the Intel Big Data Center. The authors used resources of NERSC, a DOE Office of Science User Facility supported by the Office of Science of the U.S. Department of Energy under Contract No. DE-AC02-05CH11231.

–

Presenters

  • Chiyu Max Jiang

    • Univ of California - Berkeley, UC Berkeley
    • Univ of California - Berkeley, Lawrence Berkeley National Laboratory
    • Univ of California - Berkeley, Lawrence Berkeley National Labratory

Authors

  • Chiyu Max Jiang

    • Univ of California - Berkeley, UC Berkeley
    • Univ of California - Berkeley, Lawrence Berkeley National Laboratory
    • Univ of California - Berkeley, Lawrence Berkeley National Labratory
  • Karthik Kashinath

    • Lawrence Berkeley National Laboratory
  • Philip S Marcus

    • Univ of California - Berkeley
    • University of California, Berkeley
  • Mr Prabhat

    • Lawrence Berkeley National Laboratory