Skip to main navigation Skip to search Skip to main content

Dual heterogeneous attention-based deep learning model for multi-output prediction of TBM operations

Ankang Ji*, Limao Zhang, Yudan Dou*, Yuexiong Ding*, Minggong Zhang, Luqi Wang

*Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

Abstract

Predicting tunnel boring machine (TBM) performance in real-time is challenging due to the complex, dynamic, and multi-output nature of TBM operations. To address the challenges, this paper proposes a deep-learning method to provide an effective and efficient solution for predicting multi-output TBM performance in real-time, while also guiding TBM operations. This method integrates various essential components, including two parallel bi-directional long short-term memory (BiLSTM), a dual heterogeneous attention module (DHAM), a loss function, and evaluation metrics to ensure precise predictions while maintaining computational efficiency for real-time deployment. Experiments on real-world TBM operation data showcase the model's enhanced capabilities, achieved through the model featuring the learning rate of 0.00001, the batch size of 4, the full training set, the 2-step time window, the utilizations of the Nadam optimizer and the DHAM, and the ensemble of multiple modules. A comparative analysis reveals that the proposed method outperforms existing state-of-the-art models. This paper not only demonstrates the capabilities of the proposed method but also opens up opportunities for further advancements in utilizing deep learning to enhance decision-making processes and operational efficiency within the infrastructure construction fields. © 2025 Elsevier B.V. All rights are reserved, including those for text and data mining, AI training, and similar technologies.
Original languageEnglish
Article number106605
JournalAutomation in Construction
Volume181
Issue numberPart A
Online published18 Oct 2025
DOIs
Publication statusPublished - Jan 2026

Funding

This paper was supported in part by the Guangdong Basic and Applied Basic Research Foundation (No. 2025A1515012989). It was also supported in part by the National Natural Science Foundation of China (No. 72301233, 72101044, and 72204058), the Natural Science Foundation of Liaoning Province (No. 2023-BSBA-026), Guangdong Basic and Applied Basic Research Foundation (No. 2022A1515010106), and Guangdong Provincial Philosophy and Social Science Planning Project (No. GD23YGL13).

Research Keywords

  • Attention mechanism
  • BiLSTM
  • Deep learning
  • Multi-output prediction
  • TBM performance

Fingerprint

Dive into the research topics of 'Dual heterogeneous attention-based deep learning model for multi-output prediction of TBM operations'. Together they form a unique fingerprint.

Cite this