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

中国管理科学 ›› 2026, Vol. 34 ›› Issue (8): 64-75.doi: 10.16381/j.cnki.issn1003-207x.2024.0800cstr: 32146.14.j.cnki.issn1003-207x.2024.0800

• • 上一篇    下一篇

基于AutoGluon框架的个人违约判别方法研究

李刚1,2, 曹博雄1, 秦思萌1,2, 成静怡1, 赵芳1,2, 张亚京3()   

  1. 1.东北大学工商管理学院,辽宁 沈阳 110819
    2.河北省数据科学与知识管理重点实验室,河北 秦皇岛 066004
    3.上海对外经贸大学金融管理学院,上海 201620
  • 收稿日期:2024-05-20 修回日期:2025-06-14 出版日期:2026-08-25 发布日期:2026-07-14
  • 通讯作者: 张亚京 E-mail:yajing1990.08@163.com
  • 基金资助:
    国家自然科学基金面上项目(71971051);国家自然科学基金面上项目(72371067);教育部人文社会科学研究青年项目(24YJCZH426);中央高校基本科研业务费项目(N2523013);上海市科委启明星计划扬帆专项(23YF1415000);东北大学博士生科研创新项目(N2323002);河北省教育厅在读研究生创新能力培养项目(CXZZSS2025168)

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

摘要:

准确评估借款人违约状态,是个人信用风险评价的关键,也是金融机构信贷决策的重要依据。AutoGluon(AGT)框架作为一种新兴的自动机器学习模型,在表格数据处理、图像识别领域表现出优良的性能。本文开发一种基于AutoGluon框架的个人信用违约判别模型,以提高违约判别的准确性。此外,为解决金融数据类别不平衡问题,引入Class Balance Loss函数修正交叉熵,以加强模型对违约样本的识别。使用Lending club 信贷数据集和拍拍贷信用数据以及UCI公布的德国、日本、澳大利亚信贷数据集,以第二类错误率、AUC和准确率为判别标准验证AutoGluon个人信用违约判别模型效果。在传统机器学习模型、Super Learner集成学习模型以及其他学者在公开数据集上的评价结果进行对比。结果表明,AutoGluon模型具有优秀的违约判别性能,并在不同的数据集上表现出稳定性。

关键词: 违约判别, 网络借贷, AutoGluon, Stacking集成

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

中图分类号: