Journal of Northeastern University(Natural Science) ›› 2023, Vol. 44 ›› Issue (11): 1631-1638.DOI: 10.12068/j.issn.1005-3026.2023.11.016

• Resources & Civil Engineering • Previous Articles     Next Articles

Shield Load Prediction Method Based on Deep Learning with Multiattention Mechanism

CHEN Cheng1, SHI Pei-xin1, WANG Zhan-sheng2, JIA Peng-jiao1   

  1. 1. School of Rail Transportation, Soochow University, Suzhou 215000, China; 2. Suzhou Rail Transit Group Co., Ltd., Suzhou 215004, China.
  • Published:2023-12-05
  • Contact: SHI Pei-xin
  • About author:-
  • Supported by:
    -

Abstract: Shield load is the main performance indicator of the shield, accurate load prediction is significant to ensure the safety and efficiency of the shield and the stability of the surrounding environment. Recognizing the limitations of the traditional prediction methods, this paper proposes a hybrid model(CBM), combining convolutional neural network (CNN), bi-directional long short-term memory (BiLSTM) and attention mechanism, to predict the shield load accurately based on the high-dimensional feature and time series characteristic of the data. The proposed model not only can extract the high-dimensional features and time series characteristics of the data, but also can highlight the importance of high-dimensional features and important time node information. The experiment results show that compared with the existing models, the proposed model achieves a higher prediction performance, the prediction accuracy of the thrust and torque is 94.2% and 96.2%.

Key words: deep learning; attention mechanism; time series feature; high-dimension; load prediction

CLC Number: