A Novel Dual-Branch Transformer with Gated Cross Attention for Remaining Useful Life Prediction of Bearings

Research output: Contribution to journalArticlepeer-review

Abstract

Features from different domains in vibration signals offer valuable insights for remaining useful life (RUL) prediction of bearings. While fusing these features can improve the prediction performance, traditional fusion methods lack effective information exchange across domains, limiting adaptive feature fusion. This limitation can lead to the information redundancy and hinder the accurate identification of bearing degradation states. To address these challenges, this study introduces a dual-branch Transformer with gated cross attention (DTGCA), designed to handle and integrate features from different domains for precise RUL prediction. Specifically, one branch processes 1-D time-series feature from the time and frequency domains, while the other branch uses a residual convolutional gated recurrent unit (res-ConvGRU) to handle 2-D time-frequency image features. The proposed gated cross-attention (GCA) mechanism enables adaptive information exchange between the branches, effectively fusing their information to provide a clearer representation of bearing degradation states. The proposed method is validated on the two real run-to-failure datasets. Comprehensive ablation experiments confirm the method's underlying rationality, while the detailed comparative experiments with other approaches clearly demonstrate its superiority.

Original languageEnglish
Pages (from-to)41410-41423
Number of pages14
JournalIEEE Sensors Journal
Volume24
Issue number24
DOIs
StatePublished - 2024

Keywords

  • Convolutional gated recurrent units (ConvGRUs)
  • cross-attention
  • feature fusion
  • remaining useful life (RUL) prediction

Fingerprint

Dive into the research topics of 'A Novel Dual-Branch Transformer with Gated Cross Attention for Remaining Useful Life Prediction of Bearings'. Together they form a unique fingerprint.

Cite this