Prediction of Mechanical Properties of Cold-Rolled Steel Based on Improved Graph Attention Network

Abstract

The prediction of mechanical properties of cold-rolled steel is very important for the quality control, process optimization, and cost control of cold-rolled steel, but it is still a challenging task to predict accurately. For the existing graph structure of graph attention networks, it is difficult to effectively establish the complex coupling relationship and nonlinear causal relationship between variables. At the same time, it is considered that the process of cold-rolled steel has typical full-flow process characteristics and the graph attention network makes it difficult to extract the path information between the central node and its higher-order neighborhood. The neural Granger causality algorithm is used to extract the latent relationship between variables, and the basic graph structure of mechanical property prediction data is constructed. Secondly, the node embedding layer is added before the graph attention network, which leverages the symmetry nature of Node2vec method by incorporating both breadth-first and depth-first exploration strategies. This ensures a balanced exploration of diverse paths in the graph, capturing not only local structures but also higher-order relationships. The combined graph attention networks are then able to effectively capture the symmetry path information between nodes and dependencies between variables. The accuracy and superiority of this method are verified by experiments in real cold-rolled steel production cases.