Skip to main navigation Skip to search Skip to main content

Joint optimisation of feature selection and SVM parameters based on an improved fireworks algorithm

  • Xiaoning Shen*
  • , Jiyong Xu
  • , Mingjian Mao
  • , Jiaqi Lu
  • , Liyan Song
  • , Qian Wang
  • *Corresponding author for this work
  • Nanjing University of Information Science & Technology
  • Southern University of Science and Technology

Research output: Contribution to journalArticlepeer-review

Abstract

In order to reduce the redundant features and improve the accuracy in classification, an improved fireworks algorithm for joint optimisation of feature selection and SVM parameters is proposed. A new fitness evaluation method is designed, which can adjust the punishment degree adaptively with the increase of the number of selected features. A differential mutation operator is introduced to enhance the information interaction among fireworks and improve the local search ability of the fireworks algorithm. A fitness-based roulette wheel selection strategy is proposed to reduce the computational complexity of the selection operator. Three groups of comparisons on 14 UCI classification datasets with increasing scales validate the effectiveness of our strategies and the significance of joint optimisation. Experimental results show that the proposed algorithm can obtain a higher accuracy in classification with fewer features.

Original languageEnglish
Pages (from-to)702-714
Number of pages13
JournalInternational Journal of Computational Science and Engineering
Volume26
Issue number6
DOIs
StatePublished - 2023
Externally publishedYes

Keywords

  • feature selection
  • fireworks algorithm
  • joint optimisation
  • parameter optimisation
  • support vector machines

Fingerprint

Dive into the research topics of 'Joint optimisation of feature selection and SVM parameters based on an improved fireworks algorithm'. Together they form a unique fingerprint.

Cite this