Journal of Applied Sciences ›› 2022, Vol. 40 ›› Issue (1): 69-79.doi: 10.3969/j.issn.0255-8297.2022.01.007

• Special Issue on Computer Applications • Previous Articles     Next Articles

Ensemble Classification Algorithm Based on Cost Sensitive Convolutional Neural Networks

ZHOU Chuanhua1,2,3, XU Wenqian1, ZHU Junjie1   

  1. 1. School of Management Science & Engineering, Anhui University of Technology, Ma'anshan 243002, Anhui, China;
    2. Key Laboratory of Multidisciplinary Management & Control of Complex Systems of Anhui Higher Education Institutes, Anhui University of Technology, Ma'anshan 243002, Anhui, China;
    3. School of Computer Science & Technology, University of Science & Technology of China, Hefei 230026, Anhui, China
  • Received:2021-07-21 Online:2022-01-28 Published:2022-01-28

Abstract: Aiming at the problem of low recognition rate of a few types of samples in unbalanced data sets, a classification algorithm based on cost sensitive convolutional neural network and AdaBoost (AdaBoost-CSCNN) was proposed. The cost sensitive convolutional neural network (CSCNN) is constructed by coordinating the cross entropy loss function of convolutional neural network (CNN) with a specific cost sensitive index. In training process, cost weighting mechanism is used to reduce the loss degree of key feature attributes of a few samples and realize the classification effect of a single CSCNN as a base classifier in AdaBoost. To verify the effectiveness of the algorithm, we carried out experiments on 9 data sets with different imbalance rates. Experimental performances, including Accuracy, Recall, F1-score and AUC, show that the AdaBoost-CSCNN algorithm has a good display for unbalanced data set classification.

Key words: cost sensitivity, convolutional neural network (CNN), AdaBoost, cost weighting mechanism, unbalanced data set

CLC Number: