Revisiting Tabular Deep Regression: Objectives and the Limits of Global Early Stopping
Not on the map. Why?
Abstract (arXiv)
While deep learning models for tabular data have recently received significant research attention, the choice of objective functions for regression tasks remains an understudied degree of freedom. Several lines of research offer alternatives to the standard Mean Squared Error (MSE), but differences in datasets, backbones, and evaluation protocols make their practical benefits difficult to compare. In this work, we address this gap by performing a systematic re-evaluation of regression objectives across a broad collection of established tabular datasets and multiple neural backbones. Our comparison reveals that many proposed objectives provide limited or inconsistent gains over a properly tuned MSE baseline, while quantile loss yields consistent aggregate improvements over MSE across the tested backbones. We then identify a limitation that remains even with this strong objective: regions with different levels of target noise can reach their best predictive performance at different training epochs. Consequently, a single global early-stopping checkpoint can impose a compromise between uncertainty regions, even when the objective can represent heteroskedastic target distributions. Motivated by this limitation, we introduce a heteroskedasticity-aware extension of quantile loss that uses input-dependent label smoothing as a potential remedy and improves over quantile loss on several backbones. Overall, our results provide a modern empirical picture of regression objectives for tabular deep learning and highlight how differences in optimal stopping times across uncertainty regions can limit the effectiveness of global early stopping. The code is available at https://github.com/yandex-research/tabular-regression.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 30 of its 44 references that are in the map. First posted on arXiv 9 Oct 2026.
Builds on
30 of 44 references.
References · 30
- 2017Neural Discrete Representation Learningcited by 4,712
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2022On Embeddings for Numerical Features in Tabular Deep Learningcited by 186
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 144
- 2026TabICLv2: A better, faster, scalable, and open tabular foundation modelcited by 144
- 2022On the Pitfalls of Heteroscedastic Uncertainty Estimation with Probabilistic Neural Networkscited by 120
- 2024TabM: Advancing Tabular Deep Learning with Parameter-Efficient Ensemblingcited by 113
- 2022CARD: Classification and Regression Diffusion Modelscited by 103
- 2024Better by Default: Strong Pre-Tuned MLPs and Boosted Trees on Tabular Datacited by 92
- 2022Rank-N-Contrast: Learning Continuous Representations for Regressioncited by 76
- 2018Improving Regression Performance with Distributional Lossescited by 67
18 more
- 2023Improving Deep Regression with Ordinal Entropycited by 45
- 2020Early Stopping in Deep Networks: Double Descent and How to Eliminate itcited by 41
- 2022Faithful Heteroscedastic Regression with Neural Networkscited by 37
- 2017Fast calibrated additive quantile regressioncited by 27
- 2026Beyond IID: How General Are Tabular Foundation Models, Really?cited by 20
- 2025Rethinking Early Stopping: Refine, Then Calibratecited by 17
- 2021High-Dimensional Quantile Regression: Convolution Smoothing and Concave Regularizationcited by 15
- 2025Decoding-based Regressioncited by 12
- 2020Adaptive Label Smoothingcited by 10
- 2023Understanding Pathologies of Deep Heteroskedastic Regressioncited by 10
- 2026Benchmarking Optimizers for MLPs in Tabular Deep Learningcited by 5
- 2023Token-Level Fitting Issues of Seq2seq Modelscited by 2
- 2024A Robust Quantile Huber Loss With Interpretable Parameter Adjustment In Distributional Reinforcement Learningcited by 2
- 2024Gradient Aligned Regression via Pairwise Lossescited by 2
- 2025Building Bridges between Regression, Clustering, and Classificationcited by 2
- 2025Improving Deep Regression with Tightnesscited by 2
- 2025Unveiling the Role of Data Uncertainty in Tabular Deep Learningcited by 2
- 2026TabPack: Efficient Hyperparameter Ensembles for Tabular Deep Learningcited by 2
Built on by
Not cited yet.