GNS: A generalizable Graph Neural Network-based simulator for particulate and fluid modeling

• URL: http://arxiv.org/abs/2211.10228v1
• Date: Fri, 18 Nov 2022 13:28:03 GMT
• Title: GNS: A generalizable Graph Neural Network-based simulator for particulate and fluid modeling
• Authors: Krishna Kumar, Joseph Vantassel
• Abstract summary: We develop a PyTorch-based Graph Network Simulator (GNS) that learns physics and predicts the flow behavior of particulate and fluid systems. GNS discretizes the domain with nodes representing a collection of material points and the links connecting the nodes representing the local interaction between particles or clusters of particles.
• Score: 2.132096006921048
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: We develop a PyTorch-based Graph Network Simulator (GNS) that learns physics and predicts the flow behavior of particulate and fluid systems. GNS discretizes the domain with nodes representing a collection of material points and the links connecting the nodes representing the local interaction between particles or clusters of particles. The GNS learns the interaction laws through message passing on the graph. GNS has three components: (a) Encoder, which embeds particle information to a latent graph, the edges are learned functions; (b) Processor, which allows data propagation and computes the nodal interactions across steps; and (c) Decoder, which extracts the relevant dynamics (e.g., particle acceleration) from the graph. We introduce physics-inspired simple inductive biases, such as an inertial frame that allows learning algorithms to prioritize one solution (constant gravitational acceleration) over another, reducing learning time. The GNS implementation uses semi-implicit Euler integration to update the next state based on the predicted accelerations. GNS trained on trajectory data is generalizable to predict particle kinematics in complex boundary conditions not seen during training. The trained model accurately predicts within a 5\% error of its associated material point method (MPM) simulation. The predictions are 5,000x faster than traditional MPM simulations (2.5 hours for MPM simulations versus 20 s for GNS simulation of granular flow). GNS surrogates are popular for solving optimization, control, critical-region prediction for in situ viz, and inverse-type problems. The GNS code is available under the open-source MIT license at https://github.com/geoelements/gns.

Related papers

• Cell Graph Transformer for Nuclei Classification [78.47566396839628]
  We develop a cell graph transformer (CGT) that treats nodes and edges as input tokens to enable learnable adjacency and information exchange among all nodes. Poorly features can lead to noisy self-attention scores and inferior convergence. We propose a novel topology-aware pretraining method that leverages a graph convolutional network (GCN) to learn a feature extractor.
  arXiv  Detail & Related papers  (2024-02-20T12:01:30Z)
• Three-dimensional granular flow simulation using graph neural network-based learned simulator [2.153852088624324]
  We use a graph neural network (GNN) to develop a simulator for granular flows. The simulator reproduces the overall behaviors of column collapses with various aspect ratios. The speed of GNS outperforms high-fidelity numerical simulators by 300 times.
  arXiv  Detail & Related papers  (2023-11-13T15:54:09Z)
• Accelerating Particle and Fluid Simulations with Differentiable Graph Networks for Solving Forward and Inverse Problems [2.153852088624324]
  We leverage physics-embedded differentiable graph network simulators to solve particulate and fluid simulations. GNS represents the domain as a graph with particles as nodes and learned interactions as edges. GNS achieves over 165x speedup for granular flow prediction compared to parallel CPU numerical simulations.
  arXiv  Detail & Related papers  (2023-09-23T11:52:43Z)
• Geometry-Informed Neural Operator for Large-Scale 3D PDEs [76.06115572844882]
  We propose the geometry-informed neural operator (GINO) to learn the solution operator of large-scale partial differential equations. We successfully trained GINO to predict the pressure on car surfaces using only five hundred data points.
  arXiv  Detail & Related papers  (2023-09-01T16:59:21Z)
• Graph Neural Network-based surrogate model for granular flows [2.153852088624324]
  Granular flow dynamics is crucial for assessing various geotechnical risks, including landslides and debris flows. Traditional continuum and discrete numerical methods are limited by their computational cost in simulating large-scale systems. We develop a graph neural network-based simulator (GNS) that takes the current state of granular flow and predicts the next state using explicit integration by learning the local interaction laws.
  arXiv  Detail & Related papers  (2023-05-09T07:28:12Z)
• Transformer with Implicit Edges for Particle-based Physics Simulation [135.77656965678196]
  Transformer with Implicit Edges (TIE) captures the rich semantics of particle interactions in an edge-free manner. We evaluate our model on diverse domains of varying complexity and materials.
  arXiv  Detail & Related papers  (2022-07-22T03:45:29Z)
• Fast Simulation of Particulate Suspensions Enabled by Graph Neural Network [21.266813711114153]
  We introduce a new framework, hydrodynamic interaction graph neural network (HIGNN), for inferring and predicting the particles' dynamics in Stokes suspensions. It overcomes the limitations of traditional approaches in computational efficiency, accuracy, and/or transferability. We demonstrate the accuracy, efficiency, and transferability of the proposed HIGNN framework in a variety of systems.
  arXiv  Detail & Related papers  (2022-06-17T14:05:53Z)
• Multiscale Spatio-Temporal Graph Neural Networks for 3D Skeleton-Based Motion Prediction [92.16318571149553]
  We propose a multiscale-temporal graph neural network (MST-GNN) to predict the future 3D-based skeleton human poses. The MST-GNN outperforms state-of-the-art methods in both short and long-term motion prediction.
  arXiv  Detail & Related papers  (2021-08-25T14:05:37Z)
• Dynamic Multiscale Graph Neural Networks for 3D Skeleton-Based Human Motion Prediction [102.9787019197379]
  We propose novel dynamic multiscale graph neural networks (DMGNN) to predict 3D skeleton-based human motions. The model is action-category-agnostic and follows an encoder-decoder framework. The proposed DMGNN outperforms state-of-the-art methods in both short and long-term predictions.
  arXiv  Detail & Related papers  (2020-03-17T02:49:51Z)
• Learning to Simulate Complex Physics with Graph Networks [68.43901833812448]
  We present a machine learning framework and model implementation that can learn to simulate a wide variety of challenging physical domains. Our framework---which we term "Graph Network-based Simulators" (GNS)--represents the state of a physical system with particles, expressed as nodes in a graph, and computes dynamics via learned message-passing. Our results show that our model can generalize from single-timestep predictions with thousands of particles during training, to different initial conditions, thousands of timesteps, and at least an order of magnitude more particles at test time.
  arXiv  Detail & Related papers  (2020-02-21T16:44:28Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.