StressGAN: A Generative Deep Learning Model for 2D Stress Distribution Prediction

Haoliang Jiang,Zhenguo Nie,Roselyn Yeo,Amir Barati Farimani,Levent Burak Kara
DOI: https://doi.org/10.1115/1.4049805
2020-05-30
Abstract:Using deep learning to analyze mechanical stress distributions has been gaining interest with the demand for fast stress analysis methods. Deep learning approaches have achieved excellent outcomes when utilized to speed up stress computation and learn the physics without prior knowledge of underlying equations. However, most studies restrict the variation of geometry or boundary conditions, making these methods difficult to be generalized to unseen configurations. We propose a conditional generative adversarial network (cGAN) model for predicting 2D von Mises stress distributions in solid structures. The cGAN learns to generate stress distributions conditioned by geometries, load, and boundary conditions through a two-player minimax game between two neural networks with no prior knowledge. By evaluating the generative network on two stress distribution datasets under multiple metrics, we demonstrate that our model can predict more accurate high-resolution stress distributions than a baseline convolutional neural network model, given various and complex cases of geometry, load and boundary conditions.
Computer Vision and Pattern Recognition,Machine Learning,Image and Video Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the prediction of 2D von Mises stress distribution in solid - mechanical structures. Specifically, although existing deep - learning methods have achieved very good results in accelerating stress calculations, most studies have restricted the changes in geometric shapes or boundary conditions, which makes these methods difficult to generalize to unseen configurations. Therefore, this paper proposes a model based on the conditional generative adversarial network (cGAN) - StressGAN, which aims to predict high - resolution stress distributions by inputting geometric shapes, loads, and boundary conditions, thereby overcoming the limitations of existing methods. StressGAN can learn to generate stress distributions through a two - player min - max game between two neural networks without prior knowledge, thereby achieving accurate predictions in various complex situations.