Feature Selection by Firefly Algorithm with Improved Initialization Strategy

Timea Bezdan,Dusan Cvetnic,Luka Gajic,Miodrag Zivkovic,Ivana Strumberger,Nebojsa Bacanin
DOI: https://doi.org/10.1145/3459960.3459974
2021-05-26
Abstract:In this paper, the application of the swarm intelligence algorithm for feature selection challenges in the field of machine learning is presented. Feature selection, which is part of the dimension reduction process, helps in selecting those features from the data set that have the most significant impact on the performance and accuracy of the machine learning model. Since the feature selection searches for an optimal (sub-optimal) set of features in a large search area and bearing in mind that swarm intelligence algorithms have proven to be good optimizers for tackling these kinds of problems, they can be efficiently employed as a wrapper method for feature selection. This manuscript proposes the improved implementation of a widely used firefly algorithm, adapted for tackling this important and current problem. Observed drawbacks of the original firefly algorithm are overcome by introducing a quasi-reflection-based learning procedure in the initialization phase. The proposed method was validated against 21 datasets, while the k-nearest neighbors were used as a classification model. The enhanced approach is compared with other state-of-the-art algorithms validated on the same data sets and it has been shown that the proposed implementation overcomes others in terms of features’ number, as well as in the terms of classification accuracy.
What problem does this paper attempt to address?