UPDF AI

A Lightweight Neural Network with Transformer to Predict Credit Default

Zongqi Hu,Chai Kiat Yeo

2024 · DOI: 10.1109/CAI59869.2024.00015
Conference on Algebraic Informatics · 0 Citations

TLDR

A simple encoder-only transformer which extracts latent features directly from credit card transaction data, followed by a simple neural network for the credit default prediction achieves comparable performance on the American Express Default Prediction Kaggle Competition dataset.

Abstract

Financial institutions desperately need accurate credit default prediction tools to minimize losses, optimize lending and foster responsible borrowing. Existing methods often rely on complex ensemble approaches which are computationally intensive. This paper proposes a relatively lightweight solution: a simple encoder-only transformer which extracts latent features directly from credit card transaction data, followed by a simple neural network for the credit default prediction. The proposed approach achieves comparable performance on the American Express (AMEX) Default Prediction Kaggle Competition dataset against state-of-the-art (SOTA) ensemble techniques such as 3-Ensemble Modules with Gradient Boosting Decision Tree (GBDT) and 3-LightGBM with Dropouts meet Multiple Additive Regression Trees (DART) and pseudo-labelling ensemble methods. It achieves better performance than LightGBM based method.

Cited Papers
Citing Papers