Survey on deep learning with class imbalance

被引:1438
作者
Johnson, Justin M. [1 ]
Khoshgoftaar, Taghi M. [1 ]
机构
[1] Florida Atlantic Univ, 777 Glades Rd, Boca Raton, FL 33431 USA
关键词
Deep learning; Deep neural networks; Class imbalance; Big data; CLASSIFICATION; MACHINE; ALGORITHM; NETWORKS; FRAUD; SMOTE;
D O I
10.1186/s40537-019-0192-5
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
The purpose of this study is to examine existing deep learning techniques for addressing class imbalanced data. Effective classification with imbalanced data is an important area of research, as high class imbalance is naturally inherent in many real-world applications, e.g., fraud detection and cancer detection. Moreover, highly imbalanced data poses added difficulty, as most learners will exhibit bias towards the majority class, and in extreme cases, may ignore the minority class altogether. Class imbalance has been studied thoroughly over the last two decades using traditional machine learning models, i.e. non-deep learning. Despite recent advances in deep learning, along with its increasing popularity, very little empirical work in the area of deep learning with class imbalance exists. Having achieved record-breaking performance results in several complex domains, investigating the use of deep neural networks for problems containing high levels of class imbalance is of great interest. Available studies regarding class imbalance and deep learning are surveyed in order to better understand the efficacy of deep learning when applied to class imbalanced data. This survey discusses the implementation details and experimental results for each study, and offers additional insight into their strengths and weaknesses. Several areas of focus include: data complexity, architectures tested, performance interpretation, ease of use, big data application, and generalization to other domains. We have found that research in this area is very limited, that most existing work focuses on computer vision tasks with convolutional neural networks, and that the effects of big data are rarely considered. Several traditional methods for class imbalance, e.g. data sampling and cost-sensitive learning, prove to be applicable in deep learning, while more advanced methods that exploit neural network feature learning abilities show promising results. The survey concludes with a discussion that highlights various gaps in deep learning from class imbalanced data for the purpose of guiding future research.
引用
收藏
页数:54
相关论文
共 132 条
[91]  
Orenstein ECC, 2015, ARXIV151000745
[92]  
Paszke A., 2017, NIPS 2017 WORKSH AUT
[93]   Dynamic Sampling in Convolutional Neural Networks for Imbalanced Data Classification [J].
Pouyanfar, Samira ;
Tao, Yudong ;
Mohan, Anup ;
Tian, Haiman ;
Kaseb, Ahmed S. ;
Gauen, Kent ;
Dailey, Ryan ;
Aghajanzadeh, Sarah ;
Lu, Yung-Hsiang ;
Chen, Shu-Ching ;
Shyu, Mei-Ling .
IEEE 1ST CONFERENCE ON MULTIMEDIA INFORMATION PROCESSING AND RETRIEVAL (MIPR 2018), 2018, :112-117
[94]  
Provost F., 1997, Proceedings of the Third International Conference on Knowledge Discovery and Data Mining, P43
[95]   Adaptive Cross-Generation Differential Evolution Operators for Multiobjective Optimization [J].
Qiu, Xin ;
Xu, Jian-Xin ;
Tan, Kay Chen ;
Abbass, Hussein A. .
IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2016, 20 (02) :232-244
[96]  
Quattoni A, 2009, PROC CVPR IEEE, P413, DOI 10.1109/CVPRW.2009.5206537
[97]   SMOTE-RSB*: a hybrid preprocessing approach based on oversampling and undersampling for high imbalanced data-sets using SMOTE and rough sets theory [J].
Ramentol, Enislay ;
Caballero, Yaile ;
Bello, Rafael ;
Herrera, Francisco .
KNOWLEDGE AND INFORMATION SYSTEMS, 2012, 33 (02) :245-265
[98]  
Rao R.B., 2006, ACM SIGKDD Explor. Newsl., V8, P3, DOI DOI 10.1145/1147234.1147236
[99]   You Only Look Once: Unified, Real-Time Object Detection [J].
Redmon, Joseph ;
Divvala, Santosh ;
Girshick, Ross ;
Farhadi, Ali .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :779-788
[100]  
Riedmiller, 2015, STRIVING SIMPLICITY