Feature Selection in Imbalanced Data

Firuz Kamalov, Fadi Thabtah, Ho Hon Leung

Research output: Contribution to journalArticlepeer-review

9 Citations (Scopus)

Abstract

The traditional feature selection methods are not suitable for imbalanced data as they tend to be biased towards the majority class. This problem is particularly acute in the field of medical diagnostics and fraud detection where the class distribution is highly skewed. In this paper, we propose a novel filter approach using decision tree-based F1-score. The F1-score incorporates the accuracy with respect to the minority class data and hence is a good measure in the case of imbalanced data. In the proposed implementation, the F1-score is calculated based on a 1-dimensional decision tree classifier resulting in a fast and effective feature evaluation method. Numerical experiments confirm that the proposed method achieves robust dimensionality reduction and accuracy results. In addition, the low computational complexity of the algorithm makes it a practical choice for big data applications.

Original languageEnglish
Pages (from-to)1527-1541
Number of pages15
JournalAnnals of Data Science
Volume10
Issue number6
DOIs
Publication statusPublished - Dec 2023

Keywords

  • Big data
  • Data mining
  • F-score
  • Feature selection
  • Filter method
  • Imbalanced data
  • Machine learning

ASJC Scopus subject areas

  • Business, Management and Accounting (miscellaneous)
  • Computer Science Applications
  • Statistics, Probability and Uncertainty
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Feature Selection in Imbalanced Data'. Together they form a unique fingerprint.

Cite this