A hybrid feature selection algorithm combining information gain and grouping particle swarm optimization for cancer diagnosis

• Published in: PloS one Vol. 19; no. 3; p. e0290332
• Main Authors: Yang, Fangyuan, Xu, Zhaozhao, Wang, Hong, Sun, Lisha, Zhai, Mengjiao, Zhang, Juan
• Format: Journal Article
• Language: English
• Published: United States Public Library of Science 11.03.2024; Public Library of Science (PLoS)
• Subjects: Algorithms; Analysis; Biology and Life Sciences; Cancer; Care and treatment; Computer and Information Sciences; Data mining; Diagnosis; Evaluation; Gene expression; Genes; Humans; Machine Learning; Medicine and Health Sciences; Neoplasms - diagnosis; Patient outcomes; Physical Sciences; Research and Analysis Methods; Support Vector Machine; Swarm intelligence; China
• ISSN: 1932-6203; 1932-6203
• DOI: 10.1371/journal.pone.0290332

Cancer diagnosis based on machine learning has become a popular application direction. Support vector machine (SVM), as a classical machine learning algorithm, has been widely used in cancer diagnosis because of its advantages in high-dimensional and small sample data. However, due to the high-dimensional feature space and high feature redundancy of gene expression data, SVM faces the problem of poor classification effect when dealing with such data. Based on this, this paper proposes a hybrid feature selection algorithm combining information gain and grouping particle swarm optimization (IG-GPSO). The algorithm firstly calculates the information gain values of the features and ranks them in descending order according to the value. Then, ranked features are grouped according to the information index, so that the features in the group are close, and the features outside the group are sparse. Finally, grouped features are searched using grouping PSO and evaluated according to in-group and out-group. Experimental results show that the average accuracy (ACC) of the SVM on the feature subset selected by the IG-GPSO is 98.50%, which is significantly better than the traditional feature selection algorithm. Compared with KNN, the classification effect of the feature subset selected by the IG-GPSO is still optimal. In addition, the results of multiple comparison tests show that the feature selection effect of the IG-GPSO is significantly better than that of traditional feature selection algorithms. The feature subset selected by IG-GPSO not only has the best classification effect, but also has the least feature scale (FS). More importantly, the IG-GPSO significantly improves the ACC of SVM in cancer diagnostic.