Discriminative Sparse Autoencoder for Gearbox Fault Diagnosis Toward Complex Vibration Signals

Zhiqiang Zhang,Qingyu Yang,Yanyang Zi,Zongze Wu
DOI: https://doi.org/10.1109/tim.2022.3203440
IF: 5.6
2022-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:Single-layer representation learning (SLRL) is very promising in automatically learning features for gearbox fault diagnosis. Most of the existing autoencoder (AE)-based SLRL methods perform feature learning by mining intrinsic correlations of the data, whereas the learned features may be insufficiently discriminative when the vibration signals are very complex. To cope with this problem, we propose a new SLRL method named discriminative sparse AE (DSAE), which incorporates label information of the data through a discriminator. By exploring intrinsic correlations and exploiting label information, DSAE is able to learn sufficient discriminative features from complex vibration signals. The proposed DSAE is evaluated by experiments on a gearbox dataset. The results show that DSAE exhibits superior and more reliable performance with acceptable training efficiency in terms of diagnosis accuracy, stability, and anti-noise capability, by comparing with the current state-of-the-art SLRL methods. More significantly, a meaningful discover is that the discriminator in DSAE can also play the role of feature selection with better ability over traditional feature selectors.
What problem does this paper attempt to address?