Joint discriminative and shared dictionary learning with dictionary extension strategy for bearing fault classification

Abstract

Sparse representation is one of the effective approaches for bearing fault diagnosis. Conventional sparse representation only focuses on fault feature extraction from bearing vibration signals while requiring experts to empirically make a diagnosis. This paper investigates bearing intelligent fault classification based on sparse representation based classification (SRC) method. A novel SRC algorithm, named joint discriminative and shared dictionary learning (JDSDL), is proposed for simultaneously characterize the class-specific and common features in vibration signals of bearings with different faults. The JDSDL seamlessly integrates discriminative dictionary learning, shared dictionary learning, and classifier training into a joint model to automatically recognize bearing faults. Moreover, a dictionary extension strategy is embedded within the proposed JDSDL model to make learned discriminative dictionary shift-invariant to improve its classification performance. The main advantages of the proposed method are that the JDSDL not only works well even if the training set is small, but also is robust to corruption of environment noises. Experimental results on three bearing datasets and comparison with competing methods are presented to demonstrate the effectiveness and superiority of the proposed JDSDL in bearing intelligent fault classification.