Interpretable A-posteriori Error Indication for Graph Neural Network Surrogate Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Barwey, Shivam, Kim, Hojin, Maulik, Romit
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910663566163968
author Barwey, Shivam
Kim, Hojin
Maulik, Romit
author_facet Barwey, Shivam
Kim, Hojin
Maulik, Romit
contents Data-driven surrogate modeling has surged in capability in recent years with the emergence of graph neural networks (GNNs), which can operate directly on mesh-based representations of data. The goal of this work is to introduce an interpretability enhancement procedure for GNNs, with application to unstructured mesh-based fluid dynamics modeling. Given a black-box baseline GNN model, the end result is an interpretable GNN model that isolates regions in physical space, corresponding to sub-graphs, that are intrinsically linked to the forecasting task while retaining the predictive capability of the baseline. These structures identified by the interpretable GNNs are adaptively produced in the forward pass and serve as explainable links between the baseline model architecture, the optimization goal, and known problem-specific physics. Additionally, through a regularization procedure, the interpretable GNNs can also be used to identify, during inference, graph nodes that correspond to a majority of the anticipated forecasting error, adding a novel interpretable error-tagging capability to baseline models. Demonstrations are performed using unstructured flow field data sourced from flow over a backward-facing step at high Reynolds numbers, with geometry extrapolations demonstrated for ramp and wall-mounted cube configurations.
format Preprint
id arxiv_https___arxiv_org_abs_2311_07548
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Interpretable A-posteriori Error Indication for Graph Neural Network Surrogate Models
Barwey, Shivam
Kim, Hojin
Maulik, Romit
Machine Learning
Computational Physics
Fluid Dynamics
Data-driven surrogate modeling has surged in capability in recent years with the emergence of graph neural networks (GNNs), which can operate directly on mesh-based representations of data. The goal of this work is to introduce an interpretability enhancement procedure for GNNs, with application to unstructured mesh-based fluid dynamics modeling. Given a black-box baseline GNN model, the end result is an interpretable GNN model that isolates regions in physical space, corresponding to sub-graphs, that are intrinsically linked to the forecasting task while retaining the predictive capability of the baseline. These structures identified by the interpretable GNNs are adaptively produced in the forward pass and serve as explainable links between the baseline model architecture, the optimization goal, and known problem-specific physics. Additionally, through a regularization procedure, the interpretable GNNs can also be used to identify, during inference, graph nodes that correspond to a majority of the anticipated forecasting error, adding a novel interpretable error-tagging capability to baseline models. Demonstrations are performed using unstructured flow field data sourced from flow over a backward-facing step at high Reynolds numbers, with geometry extrapolations demonstrated for ramp and wall-mounted cube configurations.
title Interpretable A-posteriori Error Indication for Graph Neural Network Surrogate Models
topic Machine Learning
Computational Physics
Fluid Dynamics
url https://arxiv.org/abs/2311.07548