计算机集成制造系统 ›› 2023, Vol. 29 ›› Issue (9): 2929-2936.DOI: 10.13196/j.cims.2023.09.006

• • 上一篇    下一篇

基于随机森林和量子粒子群优化的SVM算法

崔兆亿,耿秀丽   

  1. 上海理工大学管理学院
  • 出版日期:2023-09-30 发布日期:2023-10-09
  • 基金资助:
    国家自然科学基金资助项目(72271164);教育部人文社会科学研究规划基金资助项目(19YJA630021)。

Support vector machine algorithm based on random forest and quantum particle swarm optimization

CUI Zhaoyi,GENG Xiuli   

  1. School of Business,University of Shanghai for Science and Technology
  • Online:2023-09-30 Published:2023-10-09
  • Supported by:
    Project supported by the National Natural Science Foundation,China(No.72271164),and the Humanities and Social Sciences Research Planning Fund of the Ministry of Education,China(No.19YJA630021).

摘要: 特征属性过多及内部参数的优选是影响支持向量机(SVM)模型泛化能力的重要因素,针对以上两个问题,为了提高SVM模型的泛化能力和分类精度,将随机森林(RF)算法和量子粒子群优化(QPSO)算法结合优化SVM模型的核函数参数和惩罚因子。首先利用RF算法计算出每个特征的重要性,通过特征选择筛选出重要性高的特征作为特征子集;再利用QPSO算法的全局搜索能力寻找出SVM模型的最优核函数参数和惩罚因子,最后将最优参数应用于SVM模型中进行分类预测。实验仿真结果表明,与其他机器学习算法相比,所提模型具有较高的训练精度和预测精度,也证实了该模型具有较好的鲁棒性和泛化性能。

关键词: 随机森林, 量子粒子群优化, 支持向量机, 特征选择, 鲁棒性

Abstract: Too many feature attributes and the optimization of internal parameters are important factors that affect the generalization ability of Support Vector Machine(SVM)models.For the above two problems,to improve the generalization ability and classification accuracy of the SVM model,Random Forest(RF)algorithm and Quantum Particle Swarm Optimization algorithm(QPSO)were combined to optimize the SVM model.The importance of each feature was calculated with RF algorithm,and the features with high importance were selected as a feature subset through feature selection.Using the global search capability of QPSO algorithm,the optimal kernel function parameter  σ and penalty factor C of SVM model were found.The optimal parameters were applied to SVM model for classification prediction.Experimental simulation results showed that compared with other machine learning algorithms,the proposed model had higher training accuracy and prediction accuracy,which also confirmed that the model had better robustness and generalization performance.

Key words: random forest, quantum particle swarm optimization, support vector machine, feature selection, robustness

中图分类号: