Revisiting Tabular Deep Regression: Objectives and the Limits of Global Early Stopping
cs.LG, cs.AI
Submitted: 2026-10-09
Updated: 2026-10-09
Code: https://github.com/yandex-research/tabular-regression
License: http://creativecommons.org/licenses/by-sa/4.0/
The gist: 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.
Abstract
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.
Sources
- Building Bridges between Regression, Clustering, and Classification
- Unveiling the Role of Data Uncertainty in Tabular Deep Learning
- Benchmarking Optimizers for MLPs in Tabular Deep Learning
- Early Stopping in Deep Networks: Double Descent and How to Eliminate it
- Rethinking Early Stopping: Refine, Then Calibrate
- Understanding Pathologies of Deep Heteroskedastic Regression
- Beyond IID: How General Are Tabular Foundation Models, Really?
- Adaptive Label Smoothing
- Scikit-learn: Machine Learning in Python
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks