A vehicle body structural performance prediction method based on heterogeneous graph neural network

Abstract

The present invention belongs to the technical field of engineering structure analysis and optimization, and discloses a method for predicting vehicle body structural performance based on heterogeneous graph neural networks. The technical points include: sampling the structural samples required by the data set in the design domain to be predicted, using heterogeneous graph representation methods and numerical analysis methods to establish a heterogeneous graph data set; defining the HGNN design space with the HGT layer as the core; using part of the heterogeneous graph data set, controlled random search method and grid search method to construct the optimal HGNN model; using the entire heterogeneous graph data set and the optimal training configuration parameters to complete the training of the optimal HGNN model; using the trained HGNN model to predict the performance of any structural sample in the design domain. The present invention is not only applicable to homogeneous structures corresponding to homogeneous graphs, but also to more widely distributed structures corresponding to heterogeneous graphs, which greatly improves the versatility of the method and enhances the scalability of the method without losing the scope of application.

Description

Technical Field

[0001] The present invention relates to the field of engineering structure analysis and optimization, and in particular to a vehicle body structure performance prediction method based on heterogeneous graph neural network. Background Art

[0002] Over the past decade, deep learning technology has been widely used in the task of predicting vehicle body structure performance. In this field, deep learning models are usually used with a certain amount of vehicle body structure analysis data to build a proxy model to predict the target performance of the vehicle body structure. Currently, the types of neural networks used for vehicle body structure performance prediction are mainly multi-layer perceptrons (MLP) and convolutional neural networks (CNN). However, a key drawback of these networks is the lack of scalability. Specifically, the proxy models based on MLP or CNN are limited to structures of a specific topology type and will immediately fail when applied to new topologies.

[0003] While graph neural networks (GNNs) can significantly improve the scalability of surrogate models by leveraging neighborhood aggregation mechanisms, this comes at the expense of their applicability. This is because GNNs can only be applied to homogeneous structures corresponding to homogeneous graphs, rather than heterogeneous structures corresponding to heterogeneous graphs. However, the real world contains a wide variety of materials, each typically possessing unique properties. Therefore, heterogeneous graphs are far more prevalent than homogeneous graphs, making it crucial and necessary to develop a vehicle body structural performance prediction method that is applicable to both homogeneous and heterogeneous graphs. Summary of the Invention

[0004] In response to the shortcomings of the existing technology, the present invention proposes a vehicle body structure performance prediction method that is both scalable and universal by representing the vehicle body structure as heterogeneous graph data and introducing a heterogeneous graph neural network.

[0005] The technical solution adopted in the present invention is:

[0006] A vehicle body structural performance prediction method based on heterogeneous graph neural network includes the following steps:

[0007] Step 1: Create a heterogeneous graph dataset corresponding to vehicle body structures of different topological types

[0008] (1) Using sampling methods to collect the required structural topology type of the data set within the design domain to be predicted;

[0009] (2) Using the sampling method to change the design parameters of the connection, the same number of structural samples are generated for each of the above structural topology types;

[0010] (3) Decompose the structural sample into vertex components and edge components;

[0011] a. Decompose the structure sample into vertex components of the same or different types;

[0012] b. Determine edge components (of the same or different types) based on the topological relationships between different vertex components in the structure sample;

[0013] c. Assigning feature vectors of the same dimension but different values ​​to each type of vertex component to distinguish different vertex components in each type of vertex component;

[0014] d. Assign a feature vector with the same dimension but different values ​​to each type of edge component to distinguish different edge components in each type of edge component. This step is optional.

[0015] (4) According to the vertex components and edge components obtained after decomposition of the structural sample, they are converted into heterogeneous graph data:

[0016] a. Determine vertices in the heterogeneous graph: Map each type of vertex component to a type of vertex in the heterogeneous graph; Map each vertex component to a vertex in the heterogeneous graph, and use the feature vector of the vertex component as the feature of its corresponding vertex;

[0017] b. Determine the edges in the heterogeneous graph: Map each type of edge component to a type of edge in the heterogeneous graph; Map each edge component to an edge in the heterogeneous graph, and use the feature vector of the edge component as the feature of its corresponding edge.

[0018] (5) Constructing a heterogeneous graph dataset:

[0019] a. Taking the heterogeneous graph data corresponding to the structural sample as the heterogeneous graph sample in the dataset;

[0020] b. Use numerical analysis methods to solve the true performance value of each structural sample as the label of the heterogeneous graph sample.

[0021] Step 2: Define the HGNN design space

[0022] (1) HGT layer: Its computation process includes three main components: heterogeneous mutual attention, heterogeneous message passing, and target-specific heterogeneous message aggregation.

[0023] a. Heterogeneous mutual attention:

[0024]

[0025] Among them, s, t and e represent the source vertex, target vertex and the edge between the two vertices respectively; H (l-1)[s] and H (l-1)[t] represents the features of the source and target vertices before the lth layer, respectively; h represents the number of attention heads, d represents the common feature dimension of all vertices in the heterogeneous graph after being projected into the same space, and μ represents the prior tensor encoding the general meaning of each meta-relation; and Weight matrices corresponding to edge type, source vertex type, and target vertex type respectively.

[0026] b. Heterogeneous message passing:

[0027] in, and Represents the weight matrices associated with the source vertex type and edge type respectively;

[0028] c. Aggregation of heterogeneous messages for specific targets:

[0029]

[0030] in, Represents the vertex features after HGT layer update.

[0031] (2) Define the intra-layer, inter-layer, and training configuration design dimensions: Define the intra-layer design dimensions based on the variable operations and their optional values ​​of the above HGT layers; define the inter-layer design dimensions based on the variable operations and their optional values ​​between HGT layers; and define the training configuration design dimensions based on the training hyperparameters of the network model and their optional values.

[0032] Step 3: HGNN design space search

[0033] (1) Use controlled random search method to search for design dimensions within and between layers:

[0034] a. Select the i-th design dimension dim (i) (i=1, 2, ..., I), I is the total number of intra-layer and inter-layer design dimensions;

[0035] i) Select the first optional value of the design dimension Randomly sample N satisfying The design combination constitutes a collection N is the number of sampled design combinations; use these N design combinations to build HGNN models, then use some graph datasets and common values ​​of the training configuration design dimensions to complete the training, and record the highest prediction accuracy of each model on the validation set As the final score of this design portfolio;

[0036] ii) Select the next optional value for the design dimension Let \(J_i\) be the number of optional values for the current design dimension; make the following modifications to the values of the \(i\)-th design dimension for all \(N\) design combinations in i): Meanwhile, keep the values of other design dimensions unchanged, thus generating \(N\) new design combinations to form a set Use these \(N\) new design combinations to construct HGNN models respectively, then use part of the graph dataset and the common values of the training configuration design dimensions to complete the training, and record the highest prediction accuracy of each model on the validation set As the final score of this design combination;

[0037] iii) If there are still optional values for this design dimension that have not been taken, return to ii); otherwise, proceed to the next step;

[0038] iv) Take out all the \(n\) ( \(n = 1, 2, \ldots, N\))-th design combinations from the set and sort them according to their final scores The design combination with the highest score is marked as 1, the second-highest design combination is marked as 2, and so on.

[0039] v) If \(n < N\), then let \(n=n + 1\) and return to iv); otherwise, proceed to the next step;

[0040] vi) Aggregate the above \(N\) groups of rankings for all optional values of the current design dimension and determine the optimal design choice \(optimal\_val\) of the design dimension \(dim\) by analyzing the mean and distribution of the rankings (i) The smaller the average value of the ranking and the more concentrated the distribution, the better the design choice; (i)

[0041] b) If there are still design dimensions in the HGNN design space that have not been taken, then let \(i = i+1\) and return to a); otherwise, proceed to the next step;

[0042]

[0043] c) Aggregate the optimal design choices \(optimal\_val\) of all the intra-layer and inter-layer design dimensions above (i) ( \(i = 1, 2, \ldots, I\)) as the input for the next step.

[0044] (2) Construct an optimal HGNN model based on the optimal design choices of the intra-layer and inter-layer design dimensions, use the grid search method to search for the training configuration design dimensions, and output the optimal training configuration parameters.<\(0000118\)>

[0044] Step 4: Use the entire graph dataset and the above optimal training configuration parameters to complete the training of the optimal HGNN model, and output the model checkpoint with the highest prediction accuracy on the validation set as the input for the next step.

[0045] Step 5: Randomly sample new structural samples from the design domain to be predicted, and use the above model checkpoints to directly give the predicted value of its performance.

[0046] By introducing heterogeneous graph data and heterogeneous graph neural networks, the present invention overcomes the problem of poor scalability of previous engineering structure performance prediction methods. A single model can be used to complete performance prediction tasks for structures of various topological types without sacrificing the application scope of the method. By defining and searching the HGNN design space, the optimal HGNN model and training configuration parameters for the current performance prediction task are found, overcoming the suboptimality of the model in the existing methods and improving the prediction accuracy of the model. Compared with GNN, the versatility of the method is greatly improved. It can be applied to both homogeneous graph data and heterogeneous graph data, and has broader application prospects. BRIEF DESCRIPTION OF THE DRAWINGS

[0047] Figure 1 It is a schematic diagram of the process of the present invention;

[0048] Figure 2 It provides descriptions of three types of body frames with plates and analysis conditions;

[0049] Figure 3 The process of updating vertex features using the HGT layer;

[0050] Figure 4 It is the prediction accuracy iteration curve of the optimal HGNN model on the validation set;

[0051] Figure 5 Comparison of the prediction results of the three methods, radial basis function, MLP and HGNN, on the test dataset samples. DETAILED DESCRIPTION

[0052] The embodiments of the present invention are described in detail below in conjunction with the technical solutions and drawings.

[0053] The structural performance prediction method of the present invention has the following process: Figure 1 As shown in the embodiment. Figure 2 Taking the three types of vehicle body frames with plates as examples, the specific implementation process of this method is explained in conjunction with the accompanying drawings.

[0054] 1. Establish a heterogeneous graph dataset corresponding to three topological types of vehicle body frames.

[0055] The three body frames with plates are subjected to the same torsional stiffness analysis condition. Specifically, a concentrated force F = 1980 N of equal magnitude and opposite direction is applied at the front suspension support point, while all degrees of freedom are constrained at the rear suspension support point. The body frame consists of a thin-walled rectangular beam and a plate of constant thickness. The beam is characterized by four parameters: a, b, t1, and t2, while the plate is characterized by only one thickness parameter, T. Here, the z-direction displacement at point V1 is used as a performance indicator of the body's torsional stiffness. A larger displacement indicates a lower torsional stiffness.

[0056] To represent a body frame with panels as a heterogeneous graph, we first decompose it into two types of vertex components: beam components and plate components. Each beam component corresponds to a beam in the original structure, while each plate component corresponds to a plate in the original structure. Beam components are characterized by the coordinates of the beam's two endpoints (x1, y1, z1) and (x2, y2, z2) and cross-sectional parameters (a, b, t1, t2), while plate components are characterized by the coordinates of the plate's four corner points (x1, y1, z1), (x2, y2, z2), (x3, y3, z3), and (x4, y4, z4), as well as the plate thickness T.

[0057] Since a body frame with panels contains two types of vertex components after decomposition, namely beam components and plate components, the body frame with panels will contain four types of edge components after decomposition. These are edge components between vertex components of the same type (beam-beam and plate-plate), and edge components between vertex components of different types (beam-plate, with the beam component as the source; plate-beam, with the plate component as the source). When two beams share an endpoint, a beam-beam edge component exists. Similarly, when two plates share an edge, a plate-plate edge component exists. For edge components between vertex components of different types, when the endpoint of the beam is located on one edge of the plate, there are beam-plate and plate-beam edge components. In this embodiment, the edge components do not contain any features.

[0058] After decomposing the original structure, the vertex components and edge components are used as input to the next step to generate a heterogeneous graph representation corresponding to the original structure. We map each type of vertex component to a type of vertex in the heterogeneous graph, and use the feature vector of the vertex component as the feature of the corresponding vertex. Specifically, a beam component will be mapped to a beam vertex in the heterogeneous graph, and the feature vector of the beam component [x1, y1, z1, x2, y2, z2, a, b, t1, t2] will be used as the feature vector of the beam component. T As the feature of the vertex; a board component will be mapped to a board vertex in the heterogeneous graph, and the feature vector of the board component [x1,y1,z1,x2,y2,z2,x3,y3,z3,x4,y4,z4,T] TAs the feature of the vertex of the board. At the same time, each type of edge component is mapped to a type of edge in the heterogeneous graph to represent the topological relationship between different vertices in the heterogeneous graph.

[0059] After obtaining a heterogeneous graph representation of the vehicle body frame with panels, the finite element method was used to obtain the true values ​​of the torsional stiffness of different vehicle body frames, thereby generating labels for the heterogeneous graph samples. Subsequently, a sampling method was used to vary the cross-sectional dimensions of the beams and the thickness of the panels to generate different heterogeneous graph samples, completing the collection of a heterogeneous graph dataset. The final heterogeneous graph dataset consists of three main components: a feature matrix for all vertex types, an adjacency matrix for all edge types, and a label matrix for all heterogeneous graphs. We collected 5,000 samples for each topology type, resulting in a vehicle body frame heterogeneous graph dataset containing 15,000 samples. During training, the dataset was shuffled and split into training, validation, and test sets with a ratio of 80%, 10%, and 10%.

[0060] 2. Define the HGNN design space.

[0061] The process of the HGT layer using the neighborhood aggregation mechanism and the multi-head attention mechanism to update the vertex features in the heterogeneous graph is as follows: Figure 3 According to the HGT layer, the intra-layer design dimension, inter-layer design dimension and training configuration design dimension of the HGNN design space are defined respectively, as shown in Table 1.

[0062] Table 1 HGNN design space

[0063]

[0064] 3. Construct the optimal HGNN model for the current vehicle body frame torsional stiffness prediction task.

[0065] A controlled random search method was used to search for the intra-layer and inter-layer design dimensions, and a grid search method was used to search for the training configuration design dimensions. The optimal parameters obtained were shown in Table 2. The optimal HGNN model was constructed based on the optimal values ​​of the intra-layer and inter-layer design dimensions.

[0066] Table 2 Optimal parameters of HGNN selected

[0067]

[0068] 4. The optimal HGNN model is trained using the entire heterogeneous graph dataset of the vehicle body frame and the above optimal training configuration parameters. The prediction accuracy iteration curve of the model on the validation set is shown as follows: Figure 4 As shown, the final prediction accuracy is 96.04%.

[0069] 5. The trained HGNN model is used to give a fast and accurate prediction of torsional stiffness for the heterogeneous graph samples in the test set. The comparison of the prediction results of the radial basis function (RBF), MLP and HGNN methods on the test dataset samples is shown in the figure. Figure 5 As shown, the closer the distribution of the blocks is to the dotted line, the higher the prediction accuracy of the model and the better the performance.

Claims

1. A vehicle body structure performance prediction method based on heterogeneous graph neural network, characterized in that: The following steps are involved: Step 1: Create a heterogeneous graph dataset corresponding to vehicle body structures of different topological types: Step 2: Define the HGNN design space; Step 3: HGNN design space search: Step 4: Use the entire graph dataset and the above optimal training configuration parameters to complete the training of the optimal HGNN model, and output the model checkpoint with the highest prediction accuracy on the validation set as the input for the next step; Step 5: Randomly sample new structural samples from the design domain to be predicted, and use the above model checkpoints to directly give the predicted value of its performance; Step 2 includes: (1) HGT layer: Its computation process includes three main components: heterogeneous mutual attention, heterogeneous message passing, and target-specific heterogeneous message aggregation; a. Heterogeneous mutual attention: Among them, s, t and e represent the source vertex, target vertex and the edge between the two vertices respectively; H (l-1) [s] and H (l-1) [t] represents the features of the source and target vertices before the lth layer, respectively; h represents the number of attention heads, d represents the common feature dimension of all vertices in the heterogeneous graph after being projected into the same space, and μ represents the prior tensor encoding the general meaning of each meta-relation; and Weight matrices corresponding to edge type, source vertex type, and target vertex type respectively; b. Heterogeneous message passing: in, and Represents the weight matrices associated with the source vertex type and edge type respectively; c. Aggregation of heterogeneous messages for specific targets: in, Represents the vertex features after HGT layer update; (2) Define the intra-layer, inter-layer, and training configuration design dimensions: Define the intra-layer design dimensions based on the variable operations and their optional values ​​of the above HGT layers; define the inter-layer design dimensions based on the variable operations and their optional values ​​between HGT layers; and define the training configuration design dimensions based on the training hyperparameters of the network model and their optional values. Step 3: HGNN design space search, including: (1) Use controlled random search method to search for design dimensions within and between layers: (2) Based on the optimal design selection of intra-layer and inter-layer design dimensions, the optimal HGNN model is constructed, the grid search method is used to search the training configuration design dimensions, and the optimal training configuration parameters are output.

2. The method for predicting vehicle body structure performance based on heterogeneous graph neural network according to claim 1 is characterized in that: Step 1: Create a heterogeneous graph dataset corresponding to vehicle body structures of different topological types, including: (1) Using sampling methods to collect the required structural topology type of the data set within the design domain to be predicted; (2) Using the sampling method to change the design parameters of the connection, the same number of structural samples are generated for each of the above structural topology types; (3) Decompose the structural sample into vertex components and edge components; a. Decompose the structure sample into vertex components of the same or different types; b. Determine edge components (of the same or different types) based on the topological relationships between different vertex components in the structure sample; c. Assigning feature vectors of the same dimension but different values ​​to each type of vertex component to distinguish different vertex components in each type of vertex component; d. Assign a feature vector with the same dimension but different values ​​to each type of edge component to distinguish different edge components within each type. This step is optional. (4) According to the vertex components and edge components obtained after decomposition of the structural sample, they are converted into heterogeneous graph data: a. Determine vertices in the heterogeneous graph: Map each type of vertex component to a type of vertex in the heterogeneous graph; Map each vertex component to a vertex in the heterogeneous graph, and use the feature vector of the vertex component as the feature of its corresponding vertex; b. Determine the edges in the heterogeneous graph: Map each type of edge component to an edge type in the heterogeneous graph; Map each edge component to an edge in the heterogeneous graph, and use the feature vector of the edge component as the feature of its corresponding edge; (5) Constructing a heterogeneous graph dataset: a. Use the heterogeneous graph data corresponding to the structural samples as heterogeneous graph samples in the dataset; b. Use numerical analysis methods to solve the true performance values of each structural sample as the labels of the heterogeneous graph samples.

3. The method for predicting vehicle body structure performance based on heterogeneous graph neural network according to claim 1 is characterized in that: In step 3, use the controlled random search method to search for the design dimensions within and between layers, specifically as follows: a. Select the i-th design dimension dim (i) (i=1, 2, ..., I), I is the total number of intra-layer and inter-layer design dimensions; i) Select the first optional value of the design dimension Randomly sample N satisfying The design combination constitutes a collection N is the number of sampled design combinations; use these N design combinations to build HGNN models, then use some graph datasets and common values ​​of the training configuration design dimensions to complete the training, and record the highest prediction accuracy of each model on the validation set As the final score of this design portfolio; ii) Select the next optional value for the design dimension J i is the number of optional values ​​for the current design dimension; the following modifications are made to the values ​​of the i-th design dimension of all N design combinations in i). At the same time, keep the values ​​of other design dimensions unchanged, thus generating N new design combinations to form a set Use these N new design combinations to build HGNN models respectively, and then use some graph datasets and common values ​​of training configuration design dimensions to complete training, and record the highest prediction accuracy of each model on the validation set. As the final score of this design portfolio; iii) If there are still optional values that have not been taken for this design dimension, return to ii); otherwise, proceed to the next step; iv) From the collection Take out all nth (n=1,2,..,N) design combinations And based on their final scores Sort by design; the design combination with the highest score is labeled 1, the design combination with the second highest score is labeled 2, and so on; v) If n < N method, let n = n + 1 and return to iv); otherwise, proceed to the next step; vi) Summarize all the optional values ​​of the above N groups for the current design dimension The sorting of the design dimension dim is determined by analyzing the mean and distribution of the sorting. (i) The optimal design choice optimal_val (i) ; The smaller the mean of the ranking and the more concentrated the distribution, the better the design choice; b. If there are still design dimensions in the HGNN design space that have not been taken, let i = i + 1 and return to a); otherwise, proceed to the next step; c. Summarize the optimal design choices for all the above intra-layer and inter-layer design dimensions optimal_val (i) (i=1,2,...,I), as the input for the next step.

Images