Affiliation:
1. Department of Computer and Data Sciences, Case Western Reserve University, Cleveland, OH 44106, USA
Abstract
Feature selection metrics are commonly used in the machine learning pipeline to rank and select features before creating a predictive model. While many different metrics have been proposed for feature selection, final models are often evaluated by accuracy. In this paper, we consider the relationship between common feature selection metrics and accuracy. In particular, we focus on misorderings: cases where a feature selection metric may rank features differently than accuracy would. We analytically investigate the frequency of misordering for a variety of feature selection metrics as a function of parameters that represent how a feature partitions the data. Our analysis reveals that different metrics have systematic differences in how likely they are to misorder features which can happen over a wide range of partition parameters. We then perform an empirical evaluation with different feature selection metrics on several real-world datasets to measure misordering. Our empirical results generally match our analytical results, illustrating that misordering features happens in practice and can provide some insight into the performance of feature selection metrics.
Funder
National Science Foundation
Subject
General Physics and Astronomy
Reference55 articles.
1. Hira, Z.M., and Gillies, D.F. (2015). A Review of Feature Selection and Feature Extraction Methods Applied on Microarray Data. Adv. Bioinform., 2015.
2. Dinga, R., Penninx, B.W., Veltman, D.J., Schmaal, L., and Marquand, A.F. (2019). Beyond accuracy: Measures for assessing machine learning models, pitfalls and guidelines. bioRxiv.
3. An Introduction to Variable and Feature Selection;Guyon;J. Mach. Learn. Res.,2003
4. Nguyen, T., and Sanner, S. (2013, January 17–19). Algorithms for direct 0–1 loss optimization in binary classification. Proceedings of the International Conference on Machine Learning, PMLR, Atlanta, GA, USA.
5. Quadratic convergence of smoothing Newton’s method for 0/1 loss optimization;Zhou;SIAM J. Optim.,2021