Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (06): 1133-1140.
• Artificial Intelligence and Data Mining • Previous Articles
MA Han-da,ZHU Min#br#
Received:
Revised:
Accepted:
Online:
Published:
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 ,
MA Han-da, ZHU Min. IGWOSMOTE: An over sampling method based on improved gray wolf algorithm for SVM imbalanced data classification[J]. Computer Engineering & Science, 2022, 44(06): 1133-1140.
0 / / Recommend
Add to citation manager EndNote|Ris|BibTeX
URL: http://joces.nudt.edu.cn/EN/
http://joces.nudt.edu.cn/EN/Y2022/V44/I06/1133