Learning on the Border: Active Learning in Imbalanced Data Classification

Abstract: This paper is concerned with the class imbalance problem which has been known to hinder the learning performance of classification algorithms. The problem occurs when there are significantly less number of observations of the target concept. Various real-world classification tasks, such as medical diagnosis, text categorization and fraud detection suffer from this phenomenon. The standard machine learning algorithms yield better prediction performance with balanced datasets. In this paper, we demonstrate that active learning is capable of solving the class imbalance problem by providing the learner more balanced classes. We also propose an efficient way of selecting informative instances from a smaller pool of samples for active learning which does not necessitate a search through the entire dataset. The proposed method yields an efficient querying system and allows active learning to be applied to very large datasets. Our experimental results show that with an early stopping criteria, active learning achieves a fast solution with competitive prediction performance in imbalanced data classification.

Seyda Ertekin, Jian Huang, Léon Bottou and C. Lee Giles: Learning on the Border: Active Learning in Imbalanced Data Classification, Proceedings of the 16th Conference on Information and Knowledge Management, CIKM2007, ACM Press, Lisboa, November 2007.

cikm-2007.djvu cikm-2007.pdf cikm-2007.ps.gz

@inproceedings{ertekin-2007,
  author = {Ertekin, Seyda and Huang, Jian and Bottou, L\'{e}on and Giles, C. Lee},
  title = {Learning on the Border: Active Learning in Imbalanced Data Classification},
  booktitle = {Proceedings of the 16th Conference on Information and KnowledgeManagement, CIKM2007},
  year = {2007},
  address = {Lisboa},
  month = {November},
  publisher = {ACM Press},
  url = {http://leon.bottou.org/papers/ertekin-2007},
}