Machine Learning: Science and Technology (Jan 2024)

Analysis and Benchmarking of feature reduction for classification under computational constraints

  • Omer Subasi,
  • Sayan Ghosh,
  • Joseph Manzano,
  • Bruce Palmer,
  • Andrés Marquez

DOI
https://doi.org/10.1088/2632-2153/ad3726
Journal volume & issue
Vol. 5, no. 2
p. 020501

Abstract

Read online

Machine learning is most often expensive in terms of computational and memory costs due to training with large volumes of data. Current computational limitations of many computing systems motivate us to investigate practical approaches, such as feature selection and reduction, to reduce the time and memory costs while not sacrificing the accuracy of classification algorithms. In this work, we carefully review, analyze, and identify the feature reduction methods that have low costs/overheads in terms of time and memory. Then, we evaluate the identified reduction methods in terms of their impact on the accuracy, precision, time, and memory costs of traditional classification algorithms. Specifically, we focus on the least resource intensive feature reduction methods that are available in Scikit-Learn library. Since our goal is to identify the best performing low-cost reduction methods, we do not consider complex expensive reduction algorithms in this study. In our evaluation, we find that at quadratic-scale feature reduction, the classification algorithms achieve the best trade-off among competitive performance metrics. Results show that the overall training times are reduced 61%, the model sizes are reduced 6×, and accuracy scores increase 25% compared to the baselines on average with quadratic scale reduction.

Keywords