mlsanityData to 14 Oct 2026

A transformer-based model for default prediction in mid-cap corporate markets

arXiv 2111.09902 ↗First posted 18 Nov 2021q-fin.GN cs.CY cs.LGcited by 6

Not on the map. Why?

Abstract (arXiv)

In this paper, we study mid-cap companies, i.e. publicly traded companies with less than US 10 billion in market capitalisation. Using a large dataset of US mid-cap companies observed over 30 years, we look to predict the default probability term structure over the medium term and understand which data sources (i.e. fundamental, market or pricing data) contribute most to the default risk. Whereas existing methods typically require that data from different time periods are first aggregated and turned into cross-sectional features, we frame the problem as a multi-label time-series classification problem. We adapt transformer models, a state-of-the-art deep learning model emanating from the natural language processing domain, to the credit risk modelling setting. We also interpret the predictions of these models using attention heat maps. To optimise the model further, we present a custom loss function for multi-label classification and a novel multi-channel architecture with differential training that gives the model the ability to use all input data efficiently. Our results show the proposed deep learning architecture's superior performance, resulting in a 13% improvement in AUC (Area Under the receiver operating characteristic Curve) over traditional models. We also demonstrate how to produce an importance ranking for the different data sources and the temporal relationships using a Shapley approach specific to these models.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 15 of its 65 references that are in the map. First posted on arXiv 18 Nov 2021.

Builds on

15 of 65 references.

References · 15

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  4. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  5. 2016WaveNet: A Generative Model for Raw Audiocited by 1,722
  6. 2018An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modelingcited by 1,476
  7. 2019Attention is not not Explanationcited by 636
  8. 2019Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecastingcited by 607
  9. 2016Temporal Convolutional Networks for Action Segmentation and Detectioncited by 508
  10. 2019Multi-Task Deep Neural Networks for Natural Language Understandingcited by 350
  11. 2019A Comparative Study on Transformer vs RNN in Speech Applicationscited by 211
  12. 2018Temporal Convolutional Neural Network for the Classification of Satellite Image Time Seriescited by 76
3 more
  1. 2015Human Action Recognition using Factorized Spatio-Temporal Convolutional Networkscited by 40
  2. 2018A Comparison of Transformer and Recurrent Neural Networks on Multilingual Neural Machine Translationcited by 23
  3. 2020The value of text for small business default prediction: A deep learning approachcited by 20

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2023: 2; 2024: 1; 2025: 1; 2026: 2.

6 papers.