Prediction of Tandem Cold-Rolled Strip Flatness Based on the BiGRU-Attention-iTransformer Model
摘要
Cold rolling strip production is a multi-stand continuous rolling process, so flatness prediction is a typical spatiotemporal series data prediction problem, which requires considering various complex factors affecting flatness and paying attention to the correlation of its spatiotemporal dimensions. Based on this, a cold rolled strip flatness prediction model is proposed, integrating a Bidirectional Gated Recurrent Unit (BiGRU), an Attention mechanism, and an Inverted Transformer (iTransformer). The model adopts a parallel structure, where one branch utilizes a BiGRU-Attention module designed to capture the spatiotemporal correlations in strip production data, with the BiGRU’s hidden layer dimension set to 128; the other branch employs an iTransformer module with a feature dimension of 256 and 8 attention heads to effectively extract key features and model the relationships between parameters using the self-attention mechanism. The features extracted from both branches are fused into a 128-dimensional vector, which is then passed through a fully connected layer for flatness prediction. The prediction results show that the error indicators MSE, RMSE and MAE of the proposed model are 0.937, 0.968 and 0.774 respectively, and the fitting performance indicator