A Robust Weight-Shared Capsule Network for Intelligent Machinery Fault Diagnosis
Ruyi Huang, Jipu Li, Shuhua Wang, Guanghui Li, Weihua Li
Abstract
In practical industrial applications, the working conditions of machinery are changing with long-term operation, and the health status is declining with the degradation of crucial components. When the working condition changes, prior diagnosis models cannot be generalized from one condition to another. To solve this challenging issue, in this article a robust weight-shared capsule network (WSCN) is introduced for intelligent fault diagnosis of machinery under varying working conditions. First, taking raw accelerometer signals as inputs, one-dimensional convolutional neural network is constructed to extract discriminative characteristics. Second, various capsule layers based on multistacked weight-shared capsules are developed to enhance the generalization performance for further fault classification. Finally, margin loss function as well as agreement-based dynamic routing algorithm are employed to optimize the WSCN. In this article, two diagnosis cases are carried out to demonstrate the generalization performance of the WSCN which obtains higher accuracy under varying working conditions than that of other state-of-the-art methods.