journal article·2023-03-21Open access

Effective Class-Imbalance Learning Based on SMOTE and Convolutional Neural Networks

Javad Hassannataj Joloudari, Abdolreza Marefat, Mohammad Ali Nematollahi, Solomon Sunday Oyelere, Sadiq Hussain

Applied Sciences· Volume 13· Issue 6

Abstract

Imbalanced Data (ID) is a problem that deters Machine Learning (ML) models from achieving satisfactory results. ID is the occurrence of a situation where the quantity of the samples belonging to one class outnumbers that of the other by a wide margin, making such models’ learning process biased towards the majority class. In recent years, to address this issue, several solutions have been put forward, which opt for either synthetically generating new data for the minority class or reducing the number of majority classes to balance the data. Hence, in this paper, we investigate the effectiveness of methods based on Deep Neural Networks (DNNs) and Convolutional Neural Networks (CNNs) mixed with a variety of well-known imbalanced data solutions meaning oversampling and undersampling. Then, we propose a CNN-based model in combination with SMOTE to effectively handle imbalanced data. To evaluate our methods, we have used KEEL, breast cancer, and Z-Alizadeh Sani datasets. In order to achieve reliable results, we conducted our experiments 100 times with randomly shuffled data distributions. The classification results demonstrate that the mixed Synthetic Minority Oversampling Technique (SMOTE)-Normalization-CNN outperforms different methodologies achieving 99.08% accuracy on the 24 imbalanced datasets. Therefore, the proposed mixed model can be applied to imbalanced binary classification problems on other real datasets.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Imbalanced Data Classification Techniques · Electricity Theft Detection Techniques · Artificial Intelligence in Healthcare

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Department of Computer Engineering, Faculty of Engineering, University of Birjand, Birjand 9717434765, Iran

Department of Artificial Intelligence, Technical and Engineering Faculty, South Tehran Branch, Islamic Azad University, Tehran 1477893780, Iran

Department of Computer Sciences, Fasa University, Fasa 7461686131, Iran

Department of Computer Science, Electrical and Space Engineering, Luleå University of Technology, 93187 Skellefteå, Sweden

Examination Branch, Dibrugarh University, Dibrugarh 786004, Assam, India

References (88)

Show source-provided references
1. Multi-objective optimization-based adaptive class-specific cost extreme learning machine for imbalanced classification10.1016/j.neucom.2022.05.008 ↗
2. Learning from imbalanced data10.1109/TKDE.2008.239 ↗
3. Mining with rarity: A unifying framework10.1145/1007730.1007734 ↗
4. A study of the behavior of several methods for balancing machine learning training data10.1145/1007730.1007735 ↗
5. Mani, I., and Zhang, I. (2003, January 21). kNN approach to unbalanced data distributions: A case study involving information extraction. Proceedings of the Workshop on Learning from Imbalanced Datasets (ICML 2003), Washington, DC, USA.
6. Liu, W., and Chawla, S. (2011, January 24–27). Class confidence weighted knn algorithms for imbalanced data sets. Proceedings of the Pacific-Asia Conference on Knowledge Discovery and Data Mining, Shenzhen, China.10.1007/978-3-642-20847-8_29 ↗
7. Chawla, N.V., Lazarevic, A., Hall, L.O., and Bowyer, K.W. (2003, January 22–26). SMOTEBoost: Improving prediction of the minority class in boosting. Proceedings of the European Conference on Principles of Data Mining and Knowledge Discovery, Cavtat-Dubrovnik, Croatia.10.1007/978-3-540-39804-2_12 ↗
8. RUSBoost: A Hybrid Approach to Alleviating Class Imbalance10.1109/TSMCA.2009.2029559 ↗
9. Provost, F. (2000, January 31). Machine learning from imbalanced data sets 101. Proceedings of the AAAI’2000 Workshop on Imbalanced Data Sets, Austin, TX, USA.
10. Cost-sensitive boosting for classification of imbalanced data10.1016/j.patcog.2007.04.009 ↗
11. Exploratory Undersampling for Class-Imbalance Learning
12. Rangel, Strategies for learning in class imbalance problems10.1016/S0031-3203(02)00257-1 ↗
13. Inverse random under sampling for class imbalance problem and its application to multi-label classification10.1016/j.patcog.2012.03.014 ↗
14. SMOTE: Synthetic Minority Over-sampling Technique10.1613/jair.953 ↗
15. Evolutionary Undersampling for Classification with Imbalanced Datasets: Proposals and Taxonomy10.1162/evco.2009.17.3.275 ↗
16. Cost-sensitive learning vs. sampling: Which is best for handling unbalanced classes with unequal error costs?
17. Training cost-sensitive neural networks with methods addressing the class imbalance problem10.1109/TKDE.2006.17 ↗
18. Seiffert, C., Khoshgoftaar, T.M., Van Hulse, J., and Napolitano, A. (2008, January 5–19). A Comparative Study of Data Sampling and Cost Sensitive Learning. Proceedings of the IEEE International Conference on Data Mining Workshops, Pisa, Italy.10.1109/ICDMW.2008.119 ↗
19. Bagging predictors10.1007/BF00058655 ↗
20. Experiments with a new boosting algorithm
21. Learning from imbalanced data sets with boosting and data generation: The databoost-im approach10.1145/1007730.1007736 ↗
22. Roughly balanced bagging for imbalanced data10.1002/sam.10061 ↗
23. Comparison of sampling techniques for imbalanced learning
24. Two Modifications of CNN10.1109/TSMC.1976.4309452 ↗
25. Addressing the curse of imbalanced training sets: One-sided selection
26. Synergy of sampling techniques and ensemble classifiers for classification of urban environments using full-waveform LiDAR data
27. Weighted Ensemble with one-class Classification and Over-sampling and Instance selection (WECOI): An approach for learning from imbalanced data streams10.1016/j.jocs.2022.101614 ↗
28. PF-SMOTE: A novel parameter-free SMOTE for imbalanced datasets10.1016/j.neucom.2022.05.017 ↗
29. Two density-based sampling approaches for imbalanced and overlapping data10.1016/j.knosys.2022.108217 ↗
30. A systematic study of the class imbalance problem in convolutional neural networks10.1016/j.neunet.2018.07.011 ↗
31. Li, K., Zhou, G., Zhai, J., Li, F., and Shao, M. (2019). Improved PSO_AdaBoost Ensemble Algorithm for Imbalanced Data. Sensors, 19.10.3390/s19061476 ↗
32. On the class overlap problem in imbalanced data classification10.1016/j.knosys.2020.106631 ↗
33. Uncertainty Based Under-Sampling for Learning Naive Bayes Classifiers Under Imbalanced Data Sets10.1109/ACCESS.2019.2961784 ↗
34. Dablain, D., Krawczyk, B., and Chawla, N.V. (2022). DeepSMOTE: Fusing Deep Learning and SMOTE for Imbalanced Data. IEEE Trans. Neural Networks Learn. Syst., 1–15.10.1109/TNNLS.2021.3136503 ↗
35. Resampling imbalanced data for network intrusion detection datasets10.1186/s40537-020-00390-x ↗
36. Imbalanced Data Classification via Cooperative Interaction Between Classifier and Generator10.1109/TNNLS.2021.3052243 ↗
37. A novel progressively undersampling method based on the density peaks sequence for imbalanced data10.1016/j.knosys.2020.106689 ↗
38. An automatic sampling ratio detection method based on genetic algorithm for imbalanced data classification10.1016/j.knosys.2021.106800 ↗
39. CDSMOTE: Class decomposition and synthetic minority class oversampling technique for imbalanced-data classification10.1007/s00521-020-05130-z ↗
40. SMOTE-LOF for noise identification in imbalanced data classification
41. Bagging Supervised Autoencoder Classifier for credit scoring10.1016/j.eswa.2022.118991 ↗
42. El Bakrawy, L.M., Cifci, M.A., Kausar, S., Hussain, S., Islam, A., Alatas, B., and Desuky, A.S. (2022). A Modified Ant Lion Optimization Method and Its Application for Instance Reduction Problem in Balanced and Imbalanced Data. Axioms, 11.10.3390/axioms11030095 ↗
43. Gravitation balanced multiple kernel learning for imbalanced classification10.1007/s00521-022-07187-4 ↗
44. Improving imbalanced classification using near-miss instances10.1016/j.eswa.2022.117130 ↗
45. An extension of Synthetic Minority Oversampling Technique based on Kalman filter for imbalanced datasets
46. An improved and random synthetic minority oversampling technique for imbalanced data10.1016/j.knosys.2022.108839 ↗
47. A Hierarchical Training-Convolutional Neural Network for Imbalanced Fault Diagnosis in Complex Equipment10.1109/TII.2022.3177662 ↗
48. Mohammed, R., Rawashdeh, J., and Abdullah, M. (2020, January 7–9). Machine Learning with Oversampling and Undersampling Techniques: Overview Study and Experimental Results. Proceedings of the 11th International Conference on Information and Communication Systems (ICICS), Irbid, Jordan.10.1109/ICICS49469.2020.239556 ↗
49. EID-GAN: Generative Adversarial Nets for Extremely Imbalanced Data Augmentation10.1109/TII.2022.3182781 ↗
50. Boosted SVM with active learning strategy for imbalanced data10.1007/s00500-014-1407-5 ↗
51. A novel ensemble method for credit scoring: Adaption of different imbalance ratios10.1016/j.eswa.2018.01.012 ↗
52. Learning class-imbalanced data with region-impurity synthetic minority oversampling technique10.1016/j.ins.2022.06.067 ↗
53. SMOTE for Learning from Imbalanced Data: Progress and Challenges, Marking the 15-year Anniversary10.1613/jair.1.11192 ↗
54. MLTL: A multi-label approach for the Tomek Link undersampling algorithm10.1016/j.neucom.2019.11.076 ↗
55. Hernandez, J., Carrasco-Ochoa, J.A., and Martínez-Trinidad, J.F. (2013). Progress in Pattern Recognition, Image Analysis, Computer Vision, and Applications, Proceedings of the 18th Iberoamerican Congress, CIARP 2013, Havana, Cuba, 20–23 November 2013, Springer.
56. Kamei, Y., Monden, A., Matsumoto, S., Kakimoto, T., and Matsumoto, K.-I. (2007, January 20–21). The effects of over and under sampling on fault-prone module detection. Proceedings of the First International Symposium on Empirical Software Engineering and Measurement (ESEM 2007), Madrid, Spain.10.1109/ESEM.2007.28 ↗
57. More, A. (2016). Survey of resampling techniques for improving classification performance in unbalanced datasets. arXiv.
58. A survey of deep neural network architectures and their applications10.1016/j.neucom.2016.12.038 ↗
59. Caterini, A.L., and Chang, D.E. (2018). Deep Neural Networks in a Mathematical Framework, Springer International Publishing.10.1007/978-3-319-75304-1 ↗
60. Multilayer Perceptron, Fuzzy Sets, Classification10.1109/72.159058 ↗
61. Automatic myocardial infarction detection in contrast echocardiography based on polar residual network10.1016/j.cmpb.2020.105791 ↗
62. Albawi, S., Mohammed, T.A., and Al-Zawi, S. (2017, January 21–23). Understanding of a convolutional neural network. Proceedings of the International Conference on Engineering and Technology (ICET), Antalya, Turkey.10.1109/ICEngTechnol.2017.8308186 ↗
63. O’Shea, K., and Nash, R. (2015). An introduction to convolutional neural networks. arXiv.
64. Lin, T.-Y., Goyal, P., Girshick, R., He, K., and Dollár, P. (2017, January 22–29). Focal loss for dense object detection. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.10.1109/ICCV.2017.324 ↗
65. Mulyanto, M., Faisal, M., Prakosa, S.W., and Leu, J.-S. (2020). Effectiveness of Focal Loss for Minority Classification in Network Intrusion Detection Systems. Symmetry, 13.10.3390/sym13010004 ↗
66. Keel data-mining software tool: Data set repository, integration of algorithms and experimental analysis framework
67. GSVMA: A Genetic Support Vector Machine ANOVA Method for CAD Diagnosis
68. Li, J., Fong, S., and Zhuang, Y. (2015, January 7–9). Optimizing SMOTE by metaheuristics with neural network and decision tree. Proceedings of the 3rd International Symposium on Computational and Business Intelligence (ISCBI), Bali, Indonesia.10.1109/ISCBI.2015.12 ↗
69. Chowdary, M.K., Nguyen, T.N., and Hemanth, D.J. (2021). Deep learning-based facial emotion recognition for human–computer interaction applications. Neural Comput. Appl., 1–18.10.1007/s00521-021-06012-8 ↗
70. Understanding auc roc curve
71. Zhang, S., Yuan, Y., Yao, Z., Wang, X., and Lei, Z. (2022). Improvement of the Performance of Models for Predicting Coronary Artery Disease Based on XGBoost Algorithm and Feature Processing Technology. Electronics, 11.10.3390/electronics11030315 ↗
72. Alizadehsani, R., Hosseini, M.J., Sani, Z.A., Ghandeharioun, A., and Boghrati, R. (2012, January 10). Diagnosis of coronary artery disease using cost-sensitive algorithms. Proceedings of the 12th International Conference on Data Mining Workshops, Brussels, Belgium.10.1109/ICDMW.2012.29 ↗
73. Diagnosis of coronary artery disease using data mining techniques based on symptoms and ecg features
74. A data mining approach for diagnosis of coronary artery disease10.1016/j.cmpb.2013.03.004 ↗
75. Babič, F., Olejár, J., Vantová, Z., and Paralič, J. (2017, January 3–6). Predictive and descriptive analysis for heart disease diagnosis. Proceedings of the 2017 Federated Conference on Computer Science and Information Systems (FedCSIS), Prague, Czech Republic.10.15439/2017F219 ↗
76. Computer aided decision making for heart disease detection using hybrid neural network-Genetic algorithm10.1016/j.cmpb.2017.01.004 ↗
77. Li, H., Wang, X., Li, Y., Qin, C., and Liu, C. (2018, January 6–8). Comparison between medical knowledge based and computer automated feature selection for detection of coronary artery disease using imbalanced data. Proceedings of the BIBE 2018, International Conference on Biological Information and Biomedical Engineering, Shanghai, China.
78. NE-nu-SVC: A New Nested Ensemble Clinical Decision Support System for Effective Diagnosis of Coronary Artery Disease10.1109/ACCESS.2019.2953920 ↗
79. A new machine learning technique for an accurate diagnosis of coronary artery disease10.1016/j.cmpb.2019.104992 ↗
80. Khan, Y., Qamar, U., Asad, M., and Zeb, B. (2019). Intelligent Systems and Applications, Proceedings of the 2019 Intelligent Systems Conference (IntelliSys), London, UK, 5–6 September 2019, Springer.
81. Diagnosis of coronary heart disease via classification algorithms and a new feature selection methodology
82. Association between work-related features and coronary artery disease: A heterogeneous hybrid feature selection integrated with balancing approach10.1016/j.patrec.2020.02.010 ↗
83. A Novel Approach for Coronary Artery Disease Diagnosis using Hybrid Particle Swarm Optimization based Emotional Neural Network10.1016/j.bbe.2020.09.005 ↗
84. Decision tree-based diagnosis of coronary artery disease: CART model10.1016/j.cmpb.2020.105400 ↗
85. Joloudari, J.H., Joloudari, E.H., Saadatfar, H., Ghasemigol, M., Razavi, S.M., Mosavi, A., Nabipour, N., Shamshirband, S., and Nadai, L. (2020). Coronary Artery Disease Diagnosis; Ranking the Significant Features Using a Random Trees Model. Int. J. Environ. Res. Public Health, 17.10.20944/preprints202001.0220.v1 ↗
86. Hybrid particle swarm optimization for rule discovery in the diagnosis of coronary artery disease
87. Ashish, L., Kumar, S., and Yeligeti, S. (2021). Ischemic heart disease detection using support vector Machine and extreme gradient boosting method. Mater. Today Proc.10.1016/j.matpr.2021.01.715 ↗
88. C-CADZ: Computational intelligence system for coronary artery disease detection using Z-Alizadeh Sani dataset10.1007/s10489-021-02467-3 ↗