A class-weighted supervised contrastive learning long-tailed bearing fault diagnosis approach using quadratic neural network

1 code implementation • 21 Sep 2023 • Wei-En Yu, Jinwei Sun, Shiping Zhang, Xiaoge Zhang, Jing-Xiao Liao

In this paper, we propose a supervised contrastive learning approach with a class-aware loss function to enhance the feature extraction capability of neural networks for fault diagnosis.

Contrastive Learning Data Augmentation