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

Xianghong Hu, Tianning Xu, Ying Chen, Shuai Wang
1/27/2026
cs.LG

No confident code implementation found for this paper.

Abstract

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.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{hu2026predicting,
  title  = {Predicting Mortgage Default with Machine Learning: AutoML, Class Imbalance, and Leakage Control},
  author = {Xianghong Hu and Tianning Xu and Ying Chen and Shuai Wang},
  year   = {2026},
  eprint = {2602.00120},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.00120v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.00120.svg)](https://codeofpaper.com/paper/2602.00120)

Discussion