Exploiting probabilistic topic models to improve text categorization under class imbalance

Enhong Chen, Yanggang Lin, Hui Xiong, Qiming Luo, Haiping Ma

Research output: Contribution to journalArticlepeer-review

36 Scopus citations

Abstract

In text categorization, it is quite often that the numbers of documents in different categories are different, i.e.; the class distribution is imbalanced. We propose a unique approach to improve text categorization under class imbalance by exploiting the semantic context in text documents. Specifically, we generate new samples of rare classes (categories with relatively small amount of training data) by using global semantic information of classes represented by probabilistic topic models. In this way, the numbers of samples in different categories can become more balanced and the performance of text categorization can be improved using this transformed data set. Indeed, the proposed method is different from traditional re-sampling methods, which try to balance the number of documents in different classes by re-sampling the documents in rare classes. Such re-sampling methods can cause overfitting. Another benefit of our approach is the effective handling of noisy samples. Since all the new samples are generated by topic models, the impact of noisy samples is dramatically reduced. Finally, as demonstrated by the experimental results, the proposed methods can achieve better performance under class imbalance and is more tolerant to noisy samples.

Original languageEnglish (US)
Pages (from-to)202-214
Number of pages13
JournalInformation Processing and Management
Volume47
Issue number2
DOIs
StatePublished - Mar 2011

All Science Journal Classification (ASJC) codes

  • Information Systems
  • Media Technology
  • Computer Science Applications
  • Management Science and Operations Research
  • Library and Information Sciences

Keywords

  • Class imbalance
  • Noisy data
  • Probabilistic topic model
  • Rare class analysis
  • Text categorization

Fingerprint

Dive into the research topics of 'Exploiting probabilistic topic models to improve text categorization under class imbalance'. Together they form a unique fingerprint.

Cite this