SDA: Regularization with Cut-Flip and Mix-Normal for machinery fault diagnosis under small dataset

Abstract

Data-driven intelligent diagnosis model plays a key role in the monitoring and maintenance of mechanical equipment. However, due to practical limitations, the fault data is difficult to obtain, which makes model training unsatisfactory and results in poor testing performance. Based on the characteristics of 1-D mechanical vibration signal, this paper proposes Supervised Data Augmentation (SDA) as a regularization method to provide more effective training samples, which includes Cut-Flip and Mix-Normal. Cut-Flip is used directly on the raw sample without parameter selection. Mix-Normal mixes the data and labels of a random sample with a random normal sample at a certain ratio. The proposed SDA is verified on two bearing datasets with some popular intelligent diagnosis networks. Besides, we also design a Batch Normalization CNN (BNCNN) to learn the small dataset. Results show that SDA can significantly improve the classification accuracy of BNCNN by 10%–30% under 1-8 samples of each class. The proposed method also shows a competitive performance with existing advanced methods. Finally, we further discuss each data augmentation method through a series of ablation experiments and summarize the advantages and disadvantages of the proposed SDA.