Home > Archive > 2013 > Volume 3 Number 2 (Apr. 2013) >
IJMLC 2013 Vol.3(2): 224-228 ISSN: 2010-3700
DOI: 10.7763/IJMLC.2013.V3.307

Addressing the Class Imbalance Problem in Medical Datasets

M. Mostafizur Rahman and D. N. Davis

Abstract—A well balanced dataset is very important for creating a good prediction model. Medical datasets are often not balanced in their class labels. Most existing classification methods tend to perform poorly on minority class examples when the dataset is extremely imbalanced. This is because they aim to optimize the overall accuracy without considering the relative distribution of each class. In this paper we examine the performance of over-sampling and under-sampling techniques to balance cardiovascular data. Well known over-sampling technique SMOTE is used and some under-sampling techniques are also explored. An improved under sampling technique is proposed. Experimental results show that the proposed method displays significant better performance than the existing methods.

Index Terms—Class imbalance, under-sampling, oversampling, clustering, SMOTE.

The authors are with Department of Computer Science, University of Hull, UK (e-mail: M.M.Rahman@2009.hull.ac.uk, D.N.Davis@hull.ac.uk).


Cite:M. Mostafizur Rahman and D. N. Davis, "Addressing the Class Imbalance Problem in Medical Datasets," International Journal of Machine Learning and Computing vol. 3, no. 2, pp. 224-228, 2013.

General Information

  • ISSN: 2010-3700 (Online)
  • Abbreviated Title: Int. J. Mach. Learn. Comput.
  • Frequency: Bimonthly
  • DOI: 10.18178/IJMLC
  • Editor-in-Chief: Dr. Lin Huang
  • Executive Editor:  Ms. Cherry L. Chen
  • Abstracing/Indexing: Inspec (IET), Google Scholar, Crossref, ProQuest, Electronic Journals LibraryCNKI.
  • E-mail: ijmlc@ejournal.net

Article Metrics in Dimensions