On improving the accuracy of the classification on imbalanced classes with machine learning

Imbalance of the classes, characterized by a disproportional ratio of observations in each class, is one of the significant problems in machine learning. Class imbalances can be detected in many areas, including medical diagnostics, spam filtering, and fraud detection. Most machine learning algorithms work optimally when the number of samples in each class is approximately the same. This is because most algorithms are designed to maximize accuracy and reduce error. However, under conditions of class imbalance, the model may be overfitted, which leads to incorrect estimates of object classification. Thus, in order to avoid this phenomenon and achieve better results, it is necessary to research methods for working with unbalanced data, as well as develop effective algorithms for classifying them. In this paper, we study machine learning methods to eliminate class imbalance in data in order to improve accuracy in multi-class classification problems. In this paper, to improve the accuracy of classification, it is proposed to use a combination of classification algorithms and feature selection methods RFE, Random Forest and Boruta with pre-balancing classes by random sampling, SMOTE and ADASYN. Using data on skin diseases as an example, computer experiments have shown that the use of sampling algorithms to eliminate the imbalance of classes, as well as the selection of the most informative features, significantly improves the accuracy of classification results. The Random Forest algorithm was the most effective in terms of classification accuracy when sampling data using the ADASYN algorithm.

Shchetinin E.Y.1 , Sevastianov L.A. 2, 3 , Kulyabov D.S. 2, 3 , Ayrjan E.A. 3
Институт проблем управления им. В.А. Трапезникова РАН
  • 1 Financial University, Government of the Russian Federation
  • 2 Peoples' Friendship University of Russia (RUDN University)
  • 3 Joint Institute for Nuclear Research
  • 4 Dubna State University
multiclass classification; imbalanced classes; machine learning; Smote; ADASYN; random forest
Date of creation
Date of change
Short link

Other records

Mamonov A.A., Varlamov R.A., Salpagarov S.I.
Distributed computer and communication networks: control, computation, communications (DCCN-2020). Институт проблем управления им. В.А. Трапезникова РАН. 2020. P. 347-354