Multi-task learning of classification and denoising (MLCD) for noise-robust rotor system diagnosis

Abstract

Deep learning-based research has drawn much attention in the field of fault diagnosis of various mechanical systems due to its powerful performance. In deep learning-based methods, signals become the input for a deep learning algorithm. However, the performance of an algorithm can be diminished if there is significant noise in the data. To address the noise issue, this paper proposes a fault diagnosis method called multi-task learning of classification and denoising (MLCD). The proposed method is designed to make a fault diagnosis algorithm robust against the noise in vibration signals by learning the denoising task simultaneously with the classification. Given a noisy input, MLCD can improve test accuracy by implementing denoising as an auxiliary task, using hyperparameters chosen by Bayesian optimization. To validate the proposed MLCD method, it is integrated with the two most commonly used deep learning algorithms: long short-term memory and one-dimensional convolutional neural network. For a case study, these algorithms are tested to classify the states of a rotor testbed data. The results show that the proposed MLCD method extracts noise-robust and meaningful features; ultimately, this improves the fault diagnosis performance.