Scalar Field Prediction on Meshes Using Interpolated Multi-Resolution Convolutional Neural Networks

Abstract

Scalar fields, such as stress or temperature fields, are often calculated in shape optimization and design problems in engineering. For complex problems where shapes have varying topology and cannot be parametrized, data-driven scalar field prediction can be faster than traditional finite element methods. However, current data-driven techniques to predict scalar fields are limited to a fixed grid domain, instead of arbitrary mesh structures. In this work, we propose a method to predict scalar fields on arbitrary meshes. It uses a convolutional neural network whose feature maps at multiple resolutions are interpolated to node positions before being fed into a multilayer perceptron to predict solutions to partial differential equations at mesh nodes. The model is trained on finite element von Mises stress fields, and once trained it can estimate stress values at each node on any input mesh. Two shape datasets are investigated, and the model has strong performance on both, with a median R-squared value of 0.91. We also demonstrate the model on a temperature field in a heat conduction problem, where its predictions have a median R-squared value of 0.99. Our method provides a potential flexible alternative to finite element analysis in engineering design contexts. Code and datasets are available online.

15 pages, 9 figures

References in corpus (9)

• Adam: A Method for Stochastic Optimization
• A Comprehensive Survey on Graph Neural Networks
• Instant Neural Graphics Primitives with a Multiresolution Hash Encoding
• Learning to Simulate Complex Physics with Graph Networks
• A Point-Cloud Deep Learning Framework for Prediction of Fluid Flow Fields on Irregular Geometries
• PairNorm: Tackling Oversmoothing in GNNs
• Stress Field Prediction in Cantilevered Structures Using Convolutional Neural Networks
• StressGAN: A Generative Deep Learning Model for 2D Stress Distribution Prediction
• Towards Reusable Surrogate Models: Graph-Based Transfer Learning on Trusses