Bayesian mesh optimization for graph neural networks to enhance engineering performance prediction

Abstract

Abstract In engineering design, surrogate models are widely employed to reduce the computational costs of simulations by approximating design variables and geometric parameters from computer-aided design (CAD) models. However, traditional surrogate models often lose critical information when simplified to lower dimensions, and face challenges in handling the complexity of 3D shapes, especially in industrial datasets. To address these limitations, we propose a Bayesian graph neural network (GNN) framework that directly learns geometric features from CAD mesh representations for accurate engineering performance prediction. Our framework leverages Bayesian optimization (BO) to dynamically determine the optimal mesh element size, significantly improving model accuracy while balancing computational efficiency. This approach addresses the challenge of mesh quality by optimizing mesh resolution, ensuring that critical geometric features are preserved in 3D deep-learning-based surrogate models. Unlike existing models that rely on static or predefined mesh resolutions, our method adapts mesh size based on the task, making it highly flexible for a wide range of engineering applications. Experimental results demonstrate that mesh quality directly impacts prediction accuracy, and the proposed BO-EI GNN model outperforms state-of-the-art models such as 3D CNN, SubdivNet, GCN, and GNN in predicting mass, rim stiffness, and disk stiffness. Our method also significantly reduces the computational costs compared to traditional optimization techniques. The proposed framework shows promising potential for application in finite element analysis (FEA) and other mesh-based simulations, enhancing the utility of surrogate models across various engineering fields.