Imbalanced bearing fault diagnosis based on adaptive cost-sensitive neural network

Chenhan Yang,Funa Zhou
DOI: https://doi.org/10.1109/cac53003.2021.9727984
2021-10-22
Abstract:When the traditional machine learning model deals with class imbalance problems, there will be a problem of insufficient classification accuracy of rare class samples. The traditional cost sensitive strategy uses weighting on the misclassification cost of rare class samples as a means to solve this problem. There is a problem that the class weight may not be the optimal parameter of the network model. In order to overcome this problem, this paper designs an adaptive cost sensitive strategy, so that the network model can adaptively find the optimal class weight suitable for the current class imbalance problem in the process of training, and balance the proportion of the misclassification cost of large class samples and rare samples by optimizing the class weight, so as to make the network model converge to a set of better parameters, and further improve the classification accuracy of rare samples.
What problem does this paper attempt to address?