Incorporating receiver operating characteristics into naive Bayes for unbalanced data classification

Taeheung Kim, Byung Do Chung, Jong Seok Lee

Research output: Contribution to journalArticle

5 Citations (Scopus)

Abstract

Naive Bayesian classification has been widely used in data mining area because of its simplicity and robustness to missing values and irrelevant attributes. However, naive Bayes classifiers sometimes show poor performance due to their unrealistic assumption that all attributes are equally important and conditionally independent of each other. In this research, we dispense with the former assumption by proposing a new attribute weighting method. The proposed method considers each attribute as a single classifier and measures its discriminating ability using the area under an ROC curve (AUC). Each AUC value is then used to weight the corresponding attribute. In addition, we try to reduce the complexity of classification models by selecting high AUC attributes. Using 20 real datasets from the machine learning repository at UC Irvine (UCI), we conduct a numerical experiment to show that the proposed method is an improvement over standard naive Bayes classification and existing weighting methods.

Original languageEnglish
Pages (from-to)203-218
Number of pages16
JournalComputing
Volume99
Issue number3
DOIs
Publication statusPublished - 2017 Mar 1

All Science Journal Classification (ASJC) codes

  • Software
  • Theoretical Computer Science
  • Numerical Analysis
  • Computer Science Applications
  • Computational Theory and Mathematics
  • Computational Mathematics

Fingerprint Dive into the research topics of 'Incorporating receiver operating characteristics into naive Bayes for unbalanced data classification'. Together they form a unique fingerprint.

  • Cite this