We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Oversampling for Imbalanced Learning Based on K-Means and SMOTE

Abstract: Learning from class-imbalanced data continues to be a common and challenging problem in supervised learning as standard classification algorithms are designed to handle balanced class distributions. While different strategies exist to tackle this problem, methods which generate artificial data to achieve a balanced class distribution are more versatile than modifications to the classification algorithm. Such techniques, called oversamplers, modify the training data, allowing any classifier to be used with class-imbalanced datasets. Many algorithms have been proposed for this task, but most are complex and tend to generate unnecessary noise. This work presents a simple and effective oversampling method based on k-means clustering and SMOTE oversampling, which avoids the generation of noise and effectively overcomes imbalances between and within classes. Empirical results of extensive experiments with 71 datasets show that training data oversampled with the proposed method improves classification results. Moreover, k-means SMOTE consistently outperforms other popular oversampling methods. An implementation is made available in the python programming language.
Comments: 19 pages, 8 figures
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Journal reference: Information Sciences 465 (2018) 1-20
DOI: 10.1016/j.ins.2018.06.056
Cite as: arXiv:1711.00837 [cs.LG]
  (or arXiv:1711.00837v2 [cs.LG] for this version)

Submission history

From: Felix Last [view email]
[v1] Thu, 2 Nov 2017 17:43:03 GMT (2061kb,D)
[v2] Tue, 12 Dec 2017 18:33:14 GMT (2061kb,D)

Link back to: arXiv, form interface, contact.