Search for Author, Title, Keyword
Wasserstein Distance- EEMD Enhanced Multi-Head Graph Attention Network for Rolling Bearing Fault Diagnosis Under Different Working Conditions
More details
Hide details
College of Mechanical and Electrical Engineering, Wenzhou University, China
College of mechanical and electrical engineering, Jiaxing Nanhu University, China
School of Electromechanical and Transportation, Jiaxing Nanyang Polytechnic Institute, China
Submission date: 2023-11-05
Final revision date: 2024-01-02
Acceptance date: 2024-02-13
Online publication date: 2024-02-15
Publication date: 2024-02-15
Corresponding author
Chen Gao   

School of Electromechanical and Transportation, Jiaxing Nanyang Polytechnic Institute, China
Eksploatacja i Niezawodność – Maintenance and Reliability 2024;26(2):184037
  • Wasserstein Distance- EEMD is proposed to improve the weights of the node graph.
  • Multi-head AE is used in GAT to enhance the stability of the attention-learning process.
  • The average classification accuracy of 99.55% is obtained in different working conditions.
Traditional fault diagnosis models often overlook the interconnections between segments of vibration data, resulting in the loss of critical feature information. Therefore, an efficient fault diagnosis model tailored for rolling bearings is proposed in this paper. The 1D vibration signals are first preprocessed using ensemble empirical mode decomposition (EEMD) to generate multiple intrinsic mode functions (IMF) as individual nodes. The percentage distance between each node is calculated using the Wasserstein distance (WD) to capture the relationships between nodes and use it as the edge weights to construct a node graph. An improved multi-head graph attention network (MGAT) model is established to extract features and perform classification on the node graph. This MGAT model effectively utilizes the relationships between nodes and enhances the accuracy of fault diagnosis. The experimental results demonstrate that the proposed method achieves higher accuracy compared to similar models while requiring less processing time.
This research was funded by the Science and Technology Plan Project of Jiaxing in China (Grant. 2021AY10072), and in part by the Research Project of Jiaxing Nanhu University (Grant. 62206ZL) in China.
Journals System - logo
Scroll to top