Topology-Agnostic Graph U-Nets for Scalar Field Prediction on Unstructured Meshes

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ferguson, Kevin, Chen, Yu-hsuan, Chen, Yiming, Gillman, Andrew, Hardin, James, Kara, Levent Burak
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912064943947776
author Ferguson, Kevin
Chen, Yu-hsuan
Chen, Yiming
Gillman, Andrew
Hardin, James
Kara, Levent Burak
author_facet Ferguson, Kevin
Chen, Yu-hsuan
Chen, Yiming
Gillman, Andrew
Hardin, James
Kara, Levent Burak
contents Machine-learned surrogate models to accelerate lengthy computer simulations are becoming increasingly important as engineers look to streamline the product design cycle. In many cases, these approaches offer the ability to predict relevant quantities throughout a geometry, but place constraints on the form of the input data. In a world of diverse data types, a preferred approach would not restrict the input to a particular structure. In this paper, we propose Topology-Agnostic Graph U-Net (TAG U-Net), a graph convolutional network that can be trained to input any mesh or graph structure and output a prediction of a target scalar field at each node. The model constructs coarsened versions of each input graph and performs a set of convolution and pooling operations to predict the node-wise outputs on the original graph. By training on a diverse set of shapes, the model can make strong predictions, even for shapes unlike those seen during training. A 3-D additive manufacturing dataset is presented, containing Laser Powder Bed Fusion simulation results for thousands of parts. The model is demonstrated on this dataset, and it performs well, predicting both 2-D and 3-D scalar fields with a median R-squared > 0.85 on test geometries. Code and datasets are available online.
format Preprint
id arxiv_https___arxiv_org_abs_2410_06406
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Topology-Agnostic Graph U-Nets for Scalar Field Prediction on Unstructured Meshes
Ferguson, Kevin
Chen, Yu-hsuan
Chen, Yiming
Gillman, Andrew
Hardin, James
Kara, Levent Burak
Machine Learning
Machine-learned surrogate models to accelerate lengthy computer simulations are becoming increasingly important as engineers look to streamline the product design cycle. In many cases, these approaches offer the ability to predict relevant quantities throughout a geometry, but place constraints on the form of the input data. In a world of diverse data types, a preferred approach would not restrict the input to a particular structure. In this paper, we propose Topology-Agnostic Graph U-Net (TAG U-Net), a graph convolutional network that can be trained to input any mesh or graph structure and output a prediction of a target scalar field at each node. The model constructs coarsened versions of each input graph and performs a set of convolution and pooling operations to predict the node-wise outputs on the original graph. By training on a diverse set of shapes, the model can make strong predictions, even for shapes unlike those seen during training. A 3-D additive manufacturing dataset is presented, containing Laser Powder Bed Fusion simulation results for thousands of parts. The model is demonstrated on this dataset, and it performs well, predicting both 2-D and 3-D scalar fields with a median R-squared > 0.85 on test geometries. Code and datasets are available online.
title Topology-Agnostic Graph U-Nets for Scalar Field Prediction on Unstructured Meshes
topic Machine Learning
url https://arxiv.org/abs/2410.06406