Int J Performability Eng ›› 2018, Vol. 14 ›› Issue (2): 280-289.doi: 10.23940/ijpe.18.02.p9.280289

• Original articles • Previous Articles     Next Articles

Relief Feature Selection and Parameter Optimization for Support Vector Machine based on Mixed Kernel Function

Wei Zhanga, b and Junjie Chenb   

  1. aInformation Center, Shanxi Medical College for Continuing Education, Taiyuan, 030012, China
    bCollege of Computer Science and Technology, Taiyuan University of Technology, Taiyuan, 030024, China

Abstract:

In order to improve the classification performance of Support Vector Machine (SVM), Relief feature selection algorithm was used to obtain the most relevant feature subset and remove redundant features. The mixed kernel function, which combined the global kernel function with the local kernel function, was proposed to strengthen the learning ability and generalization performance of SVM. In addition, the parameter optimization of SVM, which combined Genetic Algorithm (GA) with grid search, was performed to reduce computation time and find optimal solutions. Finally, the methods presented in this paper were used in the Heart disease data set and the Breast cancer data set in the UCI. Compared with KNN and BP neural network, the classification result of SVM model with Relief algorithm and mixed kernel function significantly outperformed the other comparable classification model and the experimental results demonstrate the validity of the proposed model.