主管:中国科学院
主办:中国优选法统筹法与经济数学研究会
   中国科学院科技战略咨询研究院

Chinese Journal of Management Science ›› 2026, Vol. 34 ›› Issue (8): 64-75.doi: 10.16381/j.cnki.issn1003-207x.2024.0800

Previous Articles     Next Articles

Research on Personal Default Prediction Methods Based on the AutoGluon Framework

Gang Li1,2, Boxiong Cao1, Simeng Qin1,2, Jingyi Cheng1, Fang Zhao1,2, Yajing Zhang3()   

  1. 1.School of Business Administration,Northeastern University,Shenyang 110819,China
    2.Hebei Key Laboratory of Data Science and Knowledge Management,Qinhuangdao 066004,China
    3.School of Financial Management,Shanghai University of International Business and Economics,Shanghai 201620,China
  • Received:2024-05-20 Revised:2025-06-14 Online:2026-08-25 Published:2026-07-14
  • Contact: Yajing Zhang E-mail:yajing1990.08@163.com

Abstract:

Accurately assessing borrower default status is a fundamental aspect of personal credit risk evaluation and a critical factor in lending decisions made by financial institutions. However, conventional statistical and machine learning techniques often have limited generalization capability and unstable individual model performance. Furthermore, ensemble learning methods often require labor-intensive hyperparameter configuration, considerable manual intervention and a tendency to overfit. To address these challenges, an automated, highly accurate framework for predicting personal credit default is proposed. The core methodology leverages the Auto Gluon (AGT) framework, an automated machine learning system specifically recognized for its efficacy with tabular data. The model uses a multi-layer stacking ensemble architecture, where each layer incorporates predictions from the previous layer and the original feature set. Predictions from various base models are amalgamated via a weighted aggregation mechanism, thereby automating the processes of model selection and hyperparameter optimization. Furthermore, to mitigate the prevalent issue of class imbalance in financial datasets, the framework incorporates a class-balanced modification of the cross-entropy loss function. This recalibrates the loss function in proportion to the inverse class frequency, enhancing the model’s sensitivity to minority-class instances. To evaluate the proposed AGT-multi-layer-stacking-MCE model, both proprietary and public datasets are utilized. This included real-world lending data from Lending Club and Paipaidai, as well as publicly available credit data from the UCI repository: specifically, the German, Japanese, and Australian datasets. Performance is assessed using key metrics including Type II error, the area under the ROC curve (AUC) and accuracy. Comparative benchmarks include traditional machine learning models, Super Learner ensemble techniques and previously published results on the same datasets. The empirical results show that the proposed model produced Type II errors of 0.0676 and 0.0226 on the two main datasets, representing reductions of 9.2% and 28.7% respectively compared to the most effective baseline model. The model outperformes conventional data-balancing techniques across all evaluated datasets, highlighting its robustness and superior discriminative capacity. The substantial potential of automated machine learning frameworks supplemented by tailored loss functions in advancing credit default prediction is illustrated. The AGT-multi-layer-stacking-MCE model elevates predictive accuracy, particularly in minimizing Type II error, which is critically important in financial risk assessment, and diminishes reliance on expert-driven manual tuning. Consequently, the proposed approach provides a scalable, efficient and highly adaptive solution for credit scoring, facilitating more reliable and automated decision-making in financial practices. It contributes to the broader adoption of advanced machine learning techniques in operational environments, offering meaningful insights into how to handle class imbalance in financial risk modelling.

Key words: default prediction, online lending, AutoGluon, stacking ensemble

CLC Number: