Minority Report: A Graph Network Oracle for In Situ Visualization

TL;DR

This paper introduces a machine learning-based oracle using a graph network to identify critical regions in large-scale simulations, enhancing in situ visualization without increasing data I/O costs.

Contribution

It presents a novel approach combining a graph network simulator with in situ visualization to improve detection of critical simulation phenomena.

Findings

Graph network oracle accurately predicts granular flow dynamics.

Enhanced in situ visualization with better data fidelity.

Maintains traditional I/O budgets while improving analysis quality.

Abstract

In situ visualization techniques are hampered by a lack of foresight: crucial simulation phenomena can be missed due to a poor sampling rate or insufficient detail at critical timesteps. Keeping a human in the loop is impractical, and defining statistical triggers can be difficult. This paper demonstrates the potential for using a machine-learning-based simulation surrogate as an oracle to identify expected critical regions of a large-scale simulation. These critical regions are used to drive the in situ analysis, providing greater data fidelity and analysis resolution with an equivalent I/O budget to a traditional in situ framework. We develop a distributed asynchronous in situ visualization by integrating TACC Galaxy with CB-Geo MPM for material point simulation of granular flows. We employ a PyTorch-based 3D Graph Network Simulator (GNS) trained on granular flow problems as an oracle to predict the dynamics of granular flows. Critical regions of interests are manually tagged in GNS for in situ rendering in MPM.