EAAI Journal 2025 Journal Article
Eviformer: An uncertainty fault diagnosis framework guided by evidential deep learning
- Jingjie Luo
- Fucai Li
- Xiaolei Xu
- Wenqiang Zhao
- Dongqing Zhang
Unpredictable signals are commonly encountered during equipment operation, and existing deep learning-based fault diagnosis methods often fail to accurately evaluate the uncertainty of diagnostic results, limiting the model's capacity to respond to unexpected signals. To address this challenge, this study introduces a modified Transformer model based on deep evidential learning—Eviformer. The proposed model first employs a Swin-Transformer (ST)-based distribution projector, which preserves the advantages of ST in extracting features from vibration signals and simultaneously projects these signals directly into a Dirichlet distribution with second-order probabilities. Furthermore, by incorporating novel evidence correction terms and a constraint factor to reconstruct the evidence constraint loss, more precise uncertainty quantification in diagnostic predictions is achieved. Using a gear-bearing vibration dataset, comparative experiments were conducted across various scenarios, including out-of-distribution gear faults, faults in unmonitored components, noise interference, and variable speed conditions. The results demonstrate that the proposed method can promptly issue uncertainty-based warnings when encountering vibration signals that significantly differ from the training set distribution, thereby offering essential support for maintenance decisions.