A multi-stage semi-supervised learning approach for intelligent fault diagnosis of rolling bearing using data augmentation and metric learning

•A SSL method for intelligent fault diagnosis of rolling bearing is proposed.•Data augmentation and metric learning are the main elements of the proposed method.•A multi-stage strategy is formulated to improve the identification ability.•The proposed method achieves excellent performance on two case studies.