HFDF-EffNetV2: A Lightweight, Noise-Robust Model for Fault Diagnosis in Rolling Bearings
In rolling bearing intelligent fault diagnosis (FD), lightweight models are constrained by issues such as noise interference and the scarcity of fault data, making it challenging to achieve real-time, high-accuracy diagnosis on resource-limited devices. To address these challenges, this study propos...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2025-04-01
|
| Series: | Applied Sciences |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2076-3417/15/9/4902 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | In rolling bearing intelligent fault diagnosis (FD), lightweight models are constrained by issues such as noise interference and the scarcity of fault data, making it challenging to achieve real-time, high-accuracy diagnosis on resource-limited devices. To address these challenges, this study proposes a lightweight model that combines the hierarchical fine-grained decision fusion (HFDF) strategy with an improved EfficientNetV2 architecture (HFDF-EffNetV2). The model optimizes depth and width multiplicity factors to enhance parameter utilization efficiency. It uses pyramidal convolution (PyConv) combined with Fused-MBConv (Fused-MBPyConv) to obtain multi-scale time-frequency information. Additionally, an enhanced MBConv, termed BSMB-Conv-MLCA, integrates subspace blueprint separable convolution (BSConv-S) with mixed local channel attention (MLCA) extract deep-grained fault features. The HFDF strategy outputs confidence in stages and updates weights to prevent the model from falling into local overfitting when handling confusable samples. Experimental results on Case 1 and Case 2 show that HFDF-EffNetV2 achieved 100% accuracy with diagnostic times of 18.67 millisecond (ms) and 17.56 ms, respectively, and 1.85 million (M) parameters. Under noisy conditions, average accuracies reached 98.19% and 85.68%, respectively. Additionally, the model performed well with small samples, yielding accuracies of 98.69% and 97.51%. These results highlight its superior robustness to noise and lightweight performance compared with other advanced models. |
|---|---|
| ISSN: | 2076-3417 |