CELGJun 15, 2021

CatBoost model with synthetic features in application to loan risk assessment of small businesses

arXiv:2106.07954v3
Originality Incremental advance
AI Analysis

This work addresses loan risk prediction for small businesses, which can support entrepreneurship and job creation, but it is incremental as it builds on existing CatBoost methods.

The paper tackled loan risk assessment for small businesses by applying CatBoost with synthetic features, achieving 95.84% accuracy and 98.80% AUC on a large SBA dataset.

Loan risk for small businesses has long been a complex problem worthy of exploring. Predicting the loan risk can benefit entrepreneurship by developing more jobs for the society. CatBoost (Categorical Boosting) is a powerful machine learning algorithm suitable for dataset with many categorical variables like the dataset for forecasting loan risk. In this paper, we identify the important risk factors that contribute to loan status classification problem. Then we compare the performance between boosting-type algorithms(especially CatBoost) with other traditional yet popular ones. The dataset we adopt in the research comes from the U.S. Small Business Administration (SBA) and holds a very large sample size (899,164 observations and 27 features). In order to make the best use of the important features in the dataset, we propose a technique named "synthetic generation" to develop more combined features based on arithmetic operation, which ends up improving the accuracy and AUC of the original CatBoost model. We obtain a high accuracy of 95.84% and well-performed AUC of 98.80% compared with the existent literature of related research.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes