An improved light gradient boosting machine algorithm based on swarm algorithms for predicting loan default of peer-to-peer lending

Open

Much Aziz Muslim, Yosza Dasril, Muhammad Sam’an, Yahya Nur Ifriza

2022 Indonesian Journal of Electrical Engineering and Computer Science Vol. 28 Issue 2 Article Cited by 11 SDG 9SDG 17 Quartile

Abstract

Internet finance and big data technology are booming in the world. The launch of peer to peer (P2P) lending platforms is a sign and a great opportunity for entrepreneurs to easily increase their capital injection. However, this great opportunity has a high risk of impacting the sustainability and security development of the platform. One way to minimize loan risk is to predict the possibility of loan default. Hence, this study aims to find the best predictive model for predicting loan default of P2P Lending Club dataset. An improved light gradient boosting machine (LightGBM) via features selection by using swarm algorithms i.e. Ant colony optimization (ACO) and bee colony optimization (BCO) to the prediction analysis process. The best feature selection process is selected 6 out of 18 features. The synthetic minority oversampling technique (SMOTE) method is also provided to solve the unbalance class problem in the dataset, then a series of operations such as data cleaning and dimension reduction are performed. The experimental results prove that the LightGBM algorithm has been successfully improved. This success is shown by the prediction accuracy of LightGBM+ACO is 95.64%, LighGBM+BCO is 94.70% and LightGBM is 94.38%. This success also demonstrates outstanding performance in predicting loan default and strong generalizations. © 2022 Institute of Advanced Engineering and Science. All rights reserved.

Affiliations

Department of Technology Management, Faculty of Technology Management and Business, Universiti Tun Hussein Onn Malaysia, Batu Pahat, Malaysia; Department of Computer Science, Universitas Negeri Semarang, Semarang, Indonesia

Research at a Glance

Premium content — register to unlock

Research at a Glance

Register to unlock

Topics & SDG Alignment

Premium content — register to unlock

Topics & SDG Alignment

Register to unlock

Collaboration

Premium content — register to unlock

Collaboration

Register to unlock

Author Profile (Selected)

Premium content — register to unlock

Author Profile (Selected)

Register to unlock

References Overview

Premium content — register to unlock

References Overview

Register to unlock

Journal & Source

Premium content — register to unlock

Journal & Source

Register to unlock

Metadata & Integrity

Premium content — register to unlock

Metadata & Integrity

Register to unlock