• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (06): 1133-1140.

• Artificial Intelligence and Data Mining • Previous Articles    

IGWOSMOTE: An over sampling method based on improved gray wolf algorithm  for SVM imbalanced data classification

MA Han-da,ZHU Min#br#   

  1. (School of Computer Science and Communication Engineering,Jiangsu University,Jiangsu 212013,China)
  • Received:2020-10-15 Revised:2021-02-25 Accepted:2022-06-25 Online:2022-06-25 Published:2022-06-17

Abstract: In order to improve the classification effect of traditional Support Vector Machine (SVM) for minority classes in unbalanced datasets, an over sampling method based on Improved Gray Wolf algorithm (IGWO),  called SMOTE,  is proposed. Firstly, the generation form of the initial gray wolf population is improved, and the individual gray wolf is composed of the penalty factor, kernel parameter, eigenvector and sampling rate of minority classes of SVM. Then, the optimal correlation parameters and the optimal sampling rate combination are obtained by intelligent search of gray wolf optimization process, and resampling is performed for the prediction of the classifier learning machine. Through the empirical test of six UCI datasets, it is concluded that IGWOSMOTE+SVM can improve the classification accuracy of minority classes by 6.3% and the overall classification accuracy by 2.1% compared with the traditional SMOTE+SVM model. IGWOSMOTE can be used as a new over sampling classification method.

Key words: support vector machine;unbalanced data;over sampling;gray wolf optimization algorithm ,