This article is cited in 2 papers

System analysis, management and information processing

Comparative analysis of class imbalance reduction methods in building machine learning models in the financial sector

L. P. Dyakonova, L. P. Dyakonova

Plekhanov Russian University of Economics, 115054, Russia, Moscow, 36 Stremyannyy lane

Abstract: The article discusses methods for improving quality metrics of machine learning models used in the financial sector. Due to the fact that the data sets on which the models are trained have class imbalances, it is proposed to use models aimed at reducing the imbalance. The study conducted experiments using 9 methods for accounting for class imbalances with three data sets on retail lending. The CatboostClassifier gradient boosting model, which does not take into account class imbalances, was used as the base model. The experiments showed that the use of the RandomOverSampler method provides a significant increase in classification quality metrics compared to the base model. The results indicate the promise of further research into methods for accounting for class imbalances in the study of financial data, as well as the feasibility of application of the considered methods in practice.

Keywords: financial risks, machine learning, classification, class imbalance

UDC: 004.89

MSC: 90С99

Received: 24.09.2024
Revised: 25.10.2024
Accepted: 09.01.2025

DOI: 10.35330/1991-6639-2025-27-1-143-151

Bibliographic databases:

•