Our research on ML and X-AI focuses on developing interpretable techniques for modeling liquefaction and full-wave form inversions. We also develop Graph-Network Simulator (GNS) to model large-deformation flows.


  • Graph Network Simulator (GNS)GitHub


GNS simulation of granular flow


Krishna Kumar

Assistant Professor, UT Austin

Yongjin Choi

PhD candidate, UT Austin

Jodie Crocker

PhD candidate, UT Austin

Cheng-Hsi Hsiao

PhD candidate, UT Austin

Joseph Vantassel

PhD, UT Austin


Minority Report: A graph network oracle for in situ visualization

Kumar, K., Navratil, P., Solis, A., Vantassel, J;

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.

A machine learning approach to predicting pore pressure response in liquefiable sands under cyclic loading

Choi, Y., Kumar, K.;

Shear stress history controls the pore pressure response in liquefiable soils. The excess pore pressure does not increase under cyclic loading when shear stress amplitude is lower than the peak prior amplitude – the shielding effect. Many sophisticated constitutive models fail to capture the shielding effect observed in the cyclic liquefaction experiments. We develop a data-driven machine learning model based on the LSTM neural network to capture the liquefaction response of soils under cyclic loading. The LSTM model is trained on 12 laboratory cyclic simple shear tests on Nevada sand in loose and dense conditions subjected to different cyclic simple shear loading conditions. The LSTM model features include the relative density of soil and the previous stress history to predict the pore water pressure response. The LSTM model successfully replicates the pore pressure response for three cyclic simple test results considering the shielding and density effects.

Using Convolutional Neural Networks (CNN) to develop starting models for 2D full waveform inversion

Vantassel, J. P.; Kumar, K.; Cox, B. R.;

Non-invasive subsurface imaging using full waveform inversion (FWI) has the potential to fundamen-tally change engineering site characterization by enabling the recovery of high resolution 2D/3D maps ofsubsurface stiffness. Yet, the accuracy of FWI remains quite sensitive to the choice of the initial start-ing model due to the complexity and non-uniqueness of the inverse problem. In response, we presentthe novel application of convolutional neural networks (CNNs) to transform an experimental seismicwavefield acquired using a linear array of surface sensors directly into a robust starting model for 2DFWI. We begin by describing three key steps used for developing the CNN, which include: selectionof a network architecture, development of a suitable training set, and performance of network training.The ability of the trained CNN to predict a suitable starting model for 2D FWI was compared againstother commonly used starting models for a classic near-surface imaging problem; the identification of anundulating, two-layer, soil-bedrock interface. The CNN developed during this study was able to predictcomplex 2D subsurface images of the testing set directly from their seismic wavefields with an averagemean absolute percent error of 6%. When compared to other common approaches, the CNN approachwas able to produce starting models with smaller seismic image and waveform misfits, both before andafter FWI. The ability of the CNN to generalize to subsurface models which were dissimilar to the onesupon which it was trained was assessed using a more complex, three-layered model. While the predictiveability of the CNN was slightly reduced, it was still able to achieve seismic image and waveform misfitscomparable to the other commonly used starting models. This study demonstrates that CNNs have greatpotential as a tool for developing good starting models for FWI, which are critical for producing accurateFWI results.