Cost-sensitive learning strategies for high-dimensional and imbalanced data: a comparative study

PeerJ Comput Sci. 2021 Dec 24:7:e832. doi: 10.7717/peerj-cs.832. eCollection 2021.

Abstract

High dimensionality and class imbalance have been largely recognized as important issues in machine learning. A vast amount of literature has indeed investigated suitable approaches to address the multiple challenges that arise when dealing with high-dimensional feature spaces (where each problem instance is described by a large number of features). As well, several learning strategies have been devised to cope with the adverse effects of imbalanced class distributions, which may severely impact on the generalization ability of the induced models. Nevertheless, although both the issues have been largely studied for several years, they have mostly been addressed separately, and their combined effects are yet to be fully understood. Indeed, little research has been so far conducted to investigate which approaches might be best suited to deal with datasets that are, at the same time, high-dimensional and class-imbalanced. To make a contribution in this direction, our work presents a comparative study among different learning strategies that leverage both feature selection, to cope with high dimensionality, as well as cost-sensitive learning methods, to cope with class imbalance. Specifically, different ways of incorporating misclassification costs into the learning process have been explored. Also different feature selection heuristics have been considered, both univariate and multivariate, to comparatively evaluate their effectiveness on imbalanced data. The experiments have been conducted on three challenging benchmarks from the genomic domain, gaining interesting insight into the beneficial impact of combining feature selection and cost-sensitive learning, especially in the presence of highly skewed data distributions.

Keywords: Class imbalance; Cost-sensitive learning; Feature selection; High-dimensional data analysis; Random forest.

Grants and funding

This work was supported by the Fondazione di Sardegna, within the projects “ADAM” (L.R. 7 agosto 2007, no 7, annualità 2018, CUP F74I19000900007) and “ASTRID” (L.R. 7 agosto 2007, no 7, annualità 2020, CUP F75F21001220007). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.