mlsanityData to 14 Oct 2026

Predicting Mortgage Default with Machine Learning: AutoML, Class Imbalance, and Leakage Control

arXiv 2602.00120 ↗First posted 27 Jan 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Mortgage default prediction is a core task in financial risk management, and machine learning models are increasingly used to estimate default probabilities and provide interpretable signals for downstream decisions. In real-world mortgage datasets, however, three factors frequently undermine evaluation validity and deployment reliability: ambiguity in default labeling, severe class imbalance, and information leakage arising from temporal structure and post-event variables. We compare multiple machine learning approaches for mortgage default prediction using a real-world loan-level dataset, with emphasis on leakage control and imbalance handling. We employ leakage-aware feature selection, a strict temporal split that constrains both origination and reporting periods, and controlled downsampling of the majority class. Across multiple positive-to-negative ratios, performance remains stable, and an AutoML approach (AutoGluon) achieves the strongest AUROC among the models evaluated. An extended and pedagogical version of this work will appear as a book chapter.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 15 references that are in the map. First posted on arXiv 27 Jan 2026.

Builds on

6 of 15 references.

References · 6

  1. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  3. 2019AutoML: A Survey of the State-of-the-Artcited by 390
  4. 2020AutoGluon-Tabular: Robust and Accurate AutoML for Structured Datacited by 383
  5. 2025Enhancing Customer Contact Efficiency with Graph Neural Networks in Credit Card Fraud Detection Workflowcited by 8
  6. 2022On Variance Estimation of Random Forests with Infinite-Order U-statisticscited by 6

Built on by

Not cited yet.