TY - JOUR
T1 - A self-adaptive deep learning framework for noise-robust fault diagnosis in high-speed rotating machinery with automated bayesian optimization
AU - Cai, Tongxing
AU - Gao, Jinsong
AU - Xu, Tao
AU - Aijun, Li
AU - Gao, Kui
AU - Hu, Jun
N1 - Publisher Copyright:
© 2025 IOP Publishing Ltd. All rights, including for text and data mining, AI training, and similar technologies, are reserved.
PY - 2025/6/30
Y1 - 2025/6/30
N2 - Data-driven fault diagnosis for rotating machinery faces challenges in deep feature extraction, noise interference, and hyperparameter sensitivity. To address these issues, we propose a novel deep learning framework integrating three key innovations: (1) A multi-channel parallel network combining CNN and TCN to extract spatiotemporal features via dense connections, preventing gradient vanishing; (2) An adaptive soft-threshold denoising module driven by self-attention mechanisms, dynamically suppressing noise while preserving critical fault signatures; (3) A Bayesian hyperparameter optimizer automating model configuration, significantly reducing manual tuning efforts. Additionally, a penalty-enhanced loss function is designed to improve classification of hard-to-distinguish faults. Extensive experiments on two bearing datasets (CWRU and WHUT) demonstrate superior performance: 98.84% accuracy on CWRU and 99.89% on WHUT, outperforming ResNet, DenseNet, and other benchmarks by 5%-30%. Especially, it outperforms the benchmark by 3%-50% in noise conditions. The framework shows strong potential for industrial predictive maintenance under noisy environments.
AB - Data-driven fault diagnosis for rotating machinery faces challenges in deep feature extraction, noise interference, and hyperparameter sensitivity. To address these issues, we propose a novel deep learning framework integrating three key innovations: (1) A multi-channel parallel network combining CNN and TCN to extract spatiotemporal features via dense connections, preventing gradient vanishing; (2) An adaptive soft-threshold denoising module driven by self-attention mechanisms, dynamically suppressing noise while preserving critical fault signatures; (3) A Bayesian hyperparameter optimizer automating model configuration, significantly reducing manual tuning efforts. Additionally, a penalty-enhanced loss function is designed to improve classification of hard-to-distinguish faults. Extensive experiments on two bearing datasets (CWRU and WHUT) demonstrate superior performance: 98.84% accuracy on CWRU and 99.89% on WHUT, outperforming ResNet, DenseNet, and other benchmarks by 5%-30%. Especially, it outperforms the benchmark by 3%-50% in noise conditions. The framework shows strong potential for industrial predictive maintenance under noisy environments.
KW - bayesian hyperparameters optimization
KW - fault diagnosis
KW - parallel neural network
KW - rotating machinery
KW - soft-threshold denoising mechanism
UR - http://www.scopus.com/inward/record.url?scp=105009110350&partnerID=8YFLogxK
U2 - 10.1088/2631-8695/ade4f3
DO - 10.1088/2631-8695/ade4f3
M3 - 文章
AN - SCOPUS:105009110350
SN - 2631-8695
VL - 7
JO - Engineering Research Express
JF - Engineering Research Express
IS - 2
M1 - 025574
ER -