Search papers, labs, and topics across Lattice.
This paper introduces the stretch transformation framework for preprocessing numeric features in tabular data, addressing the challenges posed by their heterogeneous distributions and scales. By formulating the transformation as an optimization problem aimed at enhancing target function smoothness, the framework offers two variants: unsupervised stretch, which redistributes feature density, and supervised stretch, which optimizes transformations based on target function characteristics. Experimental results across 38 datasets reveal that supervised stretch significantly outperforms existing methods, highlighting the importance of optimizing for smoothness in tabular deep learning.
Explicitly optimizing for target function smoothness can dramatically enhance the performance of models on tabular data.
Tabular data presents unique challenges for deep learning due to its heterogeneous nature, where numeric features exhibit diverse distributions, scales, and statistical properties. Although recent advances have improved how models learn from tabular data, how numeric data are transformed into model-friendly representations remains comparatively underexplored. We introduce the stretch transformation framework, which formulates numeric feature preprocessing as an optimization problem to make the target function smoother and thus more learnable. Our framework has two variants: (1) unsupervised stretch, which uniformly redistributes feature density via minimax optimization, and (2) supervised stretch, which optimizes target-aware numeric feature transformations from the perspective of target-function smoothness by minimizing the target function's Dirichlet energy in the transformed space. Our theoretical analysis further connects this framework to several popular transformations: unsupervised stretch is closely related to Piecewise Linear Encoding through a shared piecewise-linear geometry and approaches the empirical CDF transformation as the number of bins grows, while supervised stretch becomes closely related to target encoding in the fine-binning limit. Comprehensive experiments on 38 datasets from the TALENT benchmark demonstrate that supervised stretch consistently outperforms all baselines. These results show that explicitly optimizing for target function smoothness is a powerful and underexplored strategy for tabular deep learning.
CMU ML