When Does Self-Supervised Pretraining Help Tabular Models? A Study of Label Scarcity and Missing Data
Summary
This paper evaluates self-supervised pretraining for tabular models under label scarcity and missing data, finding mixed efficacy but reliable improvements under test-time missingness.
View Cached Full Text
Cached at: 08/26/26, 09:36 AM
# When Does Self-Supervised Pretraining Help Tabular Models?A Study of Label Scarcity and Missing Data
Source: [https://arxiv.org/html/2608.24381](https://arxiv.org/html/2608.24381)
Sahand MazroueiAffiliation:Faculty of Mathematics and Computer Science, Kharazmi UniversityEmail:[sahand\.mazrue@khu\.ac\.ir](mailto:)
###### Abstract
Self\-supervised learning \(SSL\) has emerged as a promising approach for tabular data, yet its efficacy under extreme label scarcity and test\-time missingness remains under\-explored\. In this paper, we evaluate a mask\-and\-recover SSL pretraining objective against training from scratch and classical baselines across 14 diverse classification tasks\. First, while SSL outperforms training from scratch on average and remains competitive with state\-of\-the\-art tree ensembles \(achieving∼\\sim0\.8954 AUC vs\. Random Forest’s 0\.9015 at 10% labels\), the SSL\-vs\-scratch gains exhibit high inter\-task variance and lack significance \(p=0\.626p=0\.626at both 5% and 10% labels\)\. Second, contrary to the hypothesis that missing\-value imputation objectives universally benefit datasets with native missingness, SSL yields the most reliable improvements on clean datasets, while frequently degrading performance on datasets with high inherent missingness\. Third, despite this training variance, SSL\-pretrained models achieve a higher average AUC than scratch\-trained models under both test\-time missingness completely at random \(MCAR\) injection \(\+0\.0245 AUC, positive on 11 of 14 tasks\) and structured missingness shifts \(MNAR, \+0\.0418 AUC, positive on 8 of 14 tasks\), though neither difference remains statistically significant after Holm\-Bonferroni correction for multiple comparisons \(adjustedp=0\.118p=0\.118andp=0\.518p=0\.518, respectively\)\. Fourth, comparing our mask\-and\-recover objective against three established tabular SSL baselines \(VIME, SCARF, SubTab\) under an identical encoder architecture, we find no significant difference from any of them \(adjustedp=0\.459p=0\.459,p=1\.000p=1\.000,p=1\.000p=1\.000\), indicating our findings reflect general properties of tabular SSL rather than idiosyncrasies of one particular pretext task\.
Keywords:Tabular Data, Self\-Supervised Learning, Missing Data, Label Scarcity, Robustness
## 1Introduction
Tabular data remains the ubiquitous format for storing heterogeneous information in domains ranging from healthcare to financial forecasting\. Despite the dominance of deep learning in vision and natural language processing, tree\-based ensembles such as Random Forest and Gradient Boosted Trees frequently outperform deep neural networks on tabular tasks\. This performance gap widens in real\-world scenarios characterized by two distinct challenges:*label scarcity*, where annotating examples is costly or slow, and*feature missingness*, where data is lost due to sensor failures, privacy restrictions, or irregular reporting\.
Self\-supervised learning \(SSL\) provides a compelling framework to address these limitations\. By optimizing a reconstruction or contrastive objective on abundant unlabeled features, a deep tabular encoder can learn the underlying data manifold before fine\-tuning on a small labeled subset\. Recent tabular SSL architectures typically employ a “mask, corrupt, and recover” paradigm\. However, the existing literature primarily benchmarks these methods on fully observed, clean datasets\. The interaction between SSL reconstruction objectives and datasets that natively contain missing values—as well as the model’s robustness when missingness distributions shift at test time—remains an open empirical question\.
In this work, we present a comprehensive evaluation of tabular SSL under realistic constraints\. We compare an SSL\-pretrained tabular encoder against the exact same architecture trained from scratch, as well as against classical machine learning baselines, across a curated suite of OpenML classification tasks\. Our experimental design explicitly tests the limits of SSL by enforcing label scarcity fractions of 1%, 5%, 10%, and 20%, and by evaluating resilience to test\-time missingness injection\.
Our analysis moves beyond aggregate leaderboards to uncover the dataset\-dependent dynamics of tabular SSL\. Our key contributions are as follows:
- •Benchmarking under Label Scarcity:We demonstrate that SSL pretraining is highly competitive with classical baselines, achieving average AUC\-ROC scores within roughly 0\.006 of Random Forest at the 10% label fraction despite far less inductive bias\. However, we show that the delta between SSL and scratch training exhibits high inter\-task variance and is not itself statistically significant\.
- •The Missing vs\. Clean Paradox:We uncover a nuanced interaction between SSL and missing data\. While reconstruction objectives inherently learn imputation, our results show that SSL yields the most consistent downstream classification improvements on*clean*datasets\. On datasets with high native missingness, SSL performance is highly dataset\-dependent\.
- •Robustness to Test\-Time Degradation:We show that SSL\-pretrained models retain higher accuracy than scratch\-trained models under input degradation at inference time\. Under Missing Completely At Random \(MCAR\) injection \(\+30% missingness\) and Missing Not At Random \(MNAR\) structured shifts, SSL models degrade more gracefully than models trained from scratch, though this advantage does not reach significance once corrected for multiple comparisons across our four primary hypothesis tests\.
- •Ablation of Tabular SSL Components:We ablate the consistency loss, group masking, and mask\-visibility across four native\-missingness datasets\. An initial single\-seed comparison suggested a large, consistent effect of removing the consistency loss; however, once we re\-ran each variant across three independent pretraining seeds—a stricter protocol than elsewhere in this paper—this effect collapsed to statistical noise \(meanΔ\\DeltaAUC=\+0\.0031=\+0\.0031,p=0\.625p=0\.625\)\. None of the three ablated components show a reliable, generalizable effect within our four\-task sample, underscoring the risk of drawing conclusions from single\-seed ablation comparisons in small\-sample tabular SSL studies\.
- •Comparison Against Prior Tabular SSL Objectives:Using the identical encoder architecture, we compare our mask\-and\-recover objective against three established tabular SSL pretext tasks—VIME, SCARF, and SubTab—at 10% labels across all 14 tasks\. None of the three baselines differ significantly from our method \(Wilcoxonp=0\.153p=0\.153,p=0\.670p=0\.670, andp=0\.502p=0\.502; all remain non\-significant after Holm\-Bonferroni correction\), indicating that our objective is empirically competitive with, rather than distinguishable from, existing tabular SSL formulations under a controlled, architecture\-matched comparison\.
## 2Related Work
### 2\.1Deep Learning for Tabular Data
While deep neural networks have revolutionized unstructured data domains, tree\-based models such as XGBoost\[[1](https://arxiv.org/html/2608.24381#bib.bib1)\], LightGBM\[[2](https://arxiv.org/html/2608.24381#bib.bib2)\], and Random Forest\[[3](https://arxiv.org/html/2608.24381#bib.bib3)\]remain the dominant paradigm for tabular data; Grinsztajn et al\.\[[7](https://arxiv.org/html/2608.24381#bib.bib7)\]attribute this persistent gap in part to deep models’ sensitivity to uninformative features and to inductive biases poorly suited to tabular data\. Recent efforts to adapt deep learning to tabular datasets have introduced specialized architectures\. TabNet\[[4](https://arxiv.org/html/2608.24381#bib.bib4)\]utilizes sequential attention for feature selection, FT\-Transformer\[[5](https://arxiv.org/html/2608.24381#bib.bib5)\]applies multi\-head self\-attention to tabular inputs, and TabTransformer\[[9](https://arxiv.org/html/2608.24381#bib.bib9)\], SAINT\[[6](https://arxiv.org/html/2608.24381#bib.bib6)\], and Non\-Parametric Transformers\[[10](https://arxiv.org/html/2608.24381#bib.bib10)\]extend attention\-based modeling across rows, columns, or both\. Separately, Gorishniy et al\.\[[8](https://arxiv.org/html/2608.24381#bib.bib8)\]show that learned numerical\-feature embeddings can substantially affect downstream performance independent of encoder architecture\. We adopt a deliberately simpler dual\-branch MLP encoder throughout this work \(Section[4](https://arxiv.org/html/2608.24381#S4)\) so that differences in downstream results can be attributed to the pretraining objective rather than architectural capacity\.
### 2\.2Self\-Supervised Learning on Tabular Data
To alleviate the data\-hunger of deep tabular models, Self\-Supervised Learning \(SSL\) has been adapted from natural language processing and computer vision\[[17](https://arxiv.org/html/2608.24381#bib.bib17),[18](https://arxiv.org/html/2608.24381#bib.bib18)\]\. VIME\[[12](https://arxiv.org/html/2608.24381#bib.bib12)\]uses a pretext task of recovering corrupted features and estimating mask vectors\. SCARF\[[13](https://arxiv.org/html/2608.24381#bib.bib13)\]adapts contrastive learning by forming positive pairs through marginal distribution corruption\. SubTab\[[14](https://arxiv.org/html/2608.24381#bib.bib14)\]divides tabular rows into multiple subsets and learns representations by reconstructing the full row from partial views\.
### 2\.3Handling Missing Data in Machine Learning
Missing data is traditionally categorized into three mechanisms: Missing Completely At Random \(MCAR\), Missing At Random \(MAR\), and Missing Not At Random \(MNAR\)\[[19](https://arxiv.org/html/2608.24381#bib.bib19)\]\. Classical approaches rely on statistical imputation \(e\.g\., mean imputation, K\-Nearest Neighbors\[[21](https://arxiv.org/html/2608.24381#bib.bib21)\], or MICE\[[22](https://arxiv.org/html/2608.24381#bib.bib22)\]\) prior to model training, while more recent work learns the imputation mechanism directly from data, such as MissForest’s iterative random\-forest\-based scheme\[[24](https://arxiv.org/html/2608.24381#bib.bib24)\]and GAIN’s adversarial imputation framework\[[23](https://arxiv.org/html/2608.24381#bib.bib23)\]\. Our classical baselines \(Section[5](https://arxiv.org/html/2608.24381#S5)\) use simple statistical imputation rather than these learned alternatives; we return to this choice in Appendix[C](https://arxiv.org/html/2608.24381#A3)\.
## 3Background and Problem Setup
### 3\.1Tabular prediction with missing values
We consider supervised classification on a tabular dataset withNNexamples andddfeatures\. Each example includes numeric featuresxnx\_\{n\}, categorical featuresxcx\_\{c\}, and a labelyy\(available only for a subset\)\.
Missingness is represented via binary masks:mn\(i\)∈\{0,1\}dnm\_\{n\}^\{\(i\)\}\\in\\\{0,1\\\}^\{d\_\{n\}\}for numeric features andmc\(i\)∈\{0,1\}dcm\_\{c\}^\{\(i\)\}\\in\\\{0,1\\\}^\{d\_\{c\}\}for categorical features of instanceii, where 1 denotes a missing entry\. To characterize the degradation of the datasets, we define the numeric missingness rate over allNNtraining instances as:
missing\_pct=∑i=1N∑mn\(i\)Ndn×100%\.\\text\{missing\\\_pct\}=\\frac\{\\sum\_\{i=1\}^\{N\}\\sum m\_\{n\}^\{\(i\)\}\}\{N\\,d\_\{n\}\}\\times 100\\%\.
### 3\.2Label scarcity protocol
We use OpenML’s official train/test split \(fold 0\); preprocessing and pretraining are fit on the training split only, never on test data\. We sample labeled subsets at fractions\{1%,5%,10%,20%\}\\\{1\\%,5\\%,10\\%,20\\%\\\}of the training set\. For each fraction, we run 3 seeds \(0, 1, 2\) controlling the labeled\-subset sample and the fine\-tuning trajectory, and report mean \(and where relevant, standard deviation\); SSL pretraining itself is run once per task and shared across these 3 seeds\. This differs from the ablation study and SSL\-baseline comparison \(Sections[6\.4](https://arxiv.org/html/2608.24381#S6.SS4),[6\.6](https://arxiv.org/html/2608.24381#S6.SS6)\), where 3 independent pretraining seeds are used instead\.
### 3\.3Metric: AUC\-ROC
We evaluate performance using test AUC\-ROC\. AUC\-ROC measures the probability that the model ranks a random positive example above a random negative example, aggregated across all decision thresholds\. AUC\-ROC ranges from 0\.5 \(random\) to 1\.0 \(perfect\)\. For the subset of our 14 tasks that are multiclass, we compute AUC\-ROC using the one\-vs\-rest formulation with macro\-averaging across classes; binary tasks use the standard two\-class AUC\-ROC\.
## 4Method
### 4\.1Model Architecture
Let a tabular dataset be denoted as𝒟=\{\(xi,yi\)\}i=1N\\mathcal\{D\}=\\\{\(x\_\{i\},y\_\{i\}\)\\\}\_\{i=1\}^\{N\}, where each instancexix\_\{i\}consists of a numeric feature vectorxn∈ℝdnx\_\{n\}\\in\\mathbb\{R\}^\{d\_\{n\}\}and a categorical feature vectorxc∈ℤdcx\_\{c\}\\in\\mathbb\{Z\}^\{d\_\{c\}\}\. To explicitly inform the model of missingness, we define binary mask vectorsmn∈\{0,1\}dnm\_\{n\}\\in\\\{0,1\\\}^\{d\_\{n\}\}andmc∈\{0,1\}dcm\_\{c\}\\in\\\{0,1\\\}^\{d\_\{c\}\}, where a value of11indicates a missing entry\.
We process these inputs using a dual\-branch Tabular Encoder\. The numeric branch concatenates the features and their masks, passing them through a two\-layer Multi\-Layer Perceptron \(MLP\):
hn=MLPnum\(\[xn⊕mn\]\)h\_\{n\}=\\text\{MLP\}\_\{\\text\{num\}\}\(\[x\_\{n\}\\oplus m\_\{n\}\]\)where⊕\\oplusdenotes concatenation, and the MLP applies Layer Normalization, ReLU activation, and Dropout \(ratep=0\.1p=0\.1\) to project the input to a hidden dimension of 128\.
Simultaneously, the categorical features are processed via learnable embedding dictionariesEj∈ℝ\|Vj\|×32E\_\{j\}\\in\\mathbb\{R\}^\{\|V\_\{j\}\|\\times 32\}, where\|Vj\|\|V\_\{j\}\|is the vocabulary size of thejj\-th categorical feature\. The embeddings are concatenated and processed:
hc=MLPcat\(⨁j=1dcEj\(xc,j\)\)h\_\{c\}=\\text\{MLP\}\_\{\\text\{cat\}\}\\left\(\\bigoplus\_\{j=1\}^\{d\_\{c\}\}E\_\{j\}\(x\_\{c,j\}\)\\right\)The final representationh∈ℝ128h\\in\\mathbb\{R\}^\{128\}is obtained by fusing the numeric and categorical representations:
h=MLPfusion\(\[hn⊕hc\]\)h=\\text\{MLP\}\_\{\\text\{fusion\}\}\(\[h\_\{n\}\\oplus h\_\{c\}\]\)
### 4\.2Self\-Supervised Pretraining Objective
Our SSL pretraining employs a “mask, corrupt, recover” paradigm\. During pretraining, we sample an artificial corruption maskm~\\tilde\{m\}from a Bernoulli distribution with parameterρ=0\.3\\rho=0\.3\. The corrupted inputx~\\tilde\{x\}replaces masked numeric values with0\.00\.0\(the scaled mean\) and categorical values with a dedicated\[MASK\]token\.
The network produces two augmented views,x~\(1\)\\tilde\{x\}^\{\(1\)\}andx~\(2\)\\tilde\{x\}^\{\(2\)\}\. The total loss is defined as:
ℒSSL\\displaystyle\\mathcal\{L\}\_\{\\mathrm\{SSL\}\}=ℒrecon\(x~\(1\),x\)\+ℒrecon\(x~\(2\),x\)\\displaystyle=\\mathcal\{L\}\_\{\\mathrm\{recon\}\}\(\\tilde\{x\}^\{\(1\)\},x\)\+\\mathcal\{L\}\_\{\\mathrm\{recon\}\}\(\\tilde\{x\}^\{\(2\)\},x\)\+λℒcons\(h\(x~\(1\)\),h\(x~\(2\)\)\)\\displaystyle\\quad\+\\lambda\\mathcal\{L\}\_\{\\mathrm\{cons\}\}\(h\(\\tilde\{x\}^\{\(1\)\}\),h\(\\tilde\{x\}^\{\(2\)\}\)\)The reconstruction lossℒrecon\\mathcal\{L\}\_\{\\mathrm\{recon\}\}is partitioned by feature type\. For numeric features, we compute the Mean Squared Error \(MSE\) strictly over the artificially masked indices:
ℒnum=∑j=1dnm~n,j\(x^n,j−xn,j\)2∑j=1dnm~n,j\+ϵ\\mathcal\{L\}\_\{\\mathrm\{num\}\}=\\frac\{\\sum\_\{j=1\}^\{d\_\{n\}\}\\tilde\{m\}\_\{n,j\}\(\\hat\{x\}\_\{n,j\}\-x\_\{n,j\}\)^\{2\}\}\{\\sum\_\{j=1\}^\{d\_\{n\}\}\\tilde\{m\}\_\{n,j\}\+\\epsilon\}For categorical features, we compute the mean Cross\-Entropy \(CE\) independently for each masked categorical featurejj\(averaged over the instances in the batch wherejjwas masked\), then average these per\-feature losses across alldcd\_\{c\}features that had at least one masked instance in the batch:
ℒcat=1\|𝒥\|∑j∈𝒥∑i=1Bm~c,j\(i\)CE\(x^c,j\(i\),xc,j\(i\)\)∑i=1Bm~c,j\(i\)\\mathcal\{L\}\_\{\\mathrm\{cat\}\}=\\frac\{1\}\{\|\\mathcal\{J\}\|\}\\sum\_\{j\\in\\mathcal\{J\}\}\\frac\{\\sum\_\{i=1\}^\{B\}\\tilde\{m\}\_\{c,j\}^\{\(i\)\}\\,\\text\{CE\}\(\\hat\{x\}\_\{c,j\}^\{\(i\)\},x\_\{c,j\}^\{\(i\)\}\)\}\{\\sum\_\{i=1\}^\{B\}\\tilde\{m\}\_\{c,j\}^\{\(i\)\}\}whereBBis the batch size and𝒥\\mathcal\{J\}is the set of categorical features masked at least once in the batch\. This weights each feature equally, rather than each masked token equally\. When a batch contains both masked numeric and masked categorical entries,ℒrecon=0\.5\(ℒnum\+ℒcat\)\\mathcal\{L\}\_\{\\mathrm\{recon\}\}=0\.5\(\\mathcal\{L\}\_\{\\mathrm\{num\}\}\+\\mathcal\{L\}\_\{\\mathrm\{cat\}\}\); if only one type is present in a given view,ℒrecon\\mathcal\{L\}\_\{\\mathrm\{recon\}\}reduces to that term alone\. Finally,ℒcons\\mathcal\{L\}\_\{\\mathrm\{cons\}\}enforces representation invariance via cosine distance:
ℒcons=1−h\(1\)⋅h\(2\)‖h\(1\)‖2‖h\(2\)‖2\\mathcal\{L\}\_\{\\mathrm\{cons\}\}=1\-\\frac\{h^\{\(1\)\}\\cdot h^\{\(2\)\}\}\{\\\|h^\{\(1\)\}\\\|\_\{2\}\\\|h^\{\(2\)\}\\\|\_\{2\}\}
### 4\.3Group Masking
We optionally mask groups of features \(rather than independent features\) to encourage learning dependencies\. Feature groups are discovered by calculating the absolute Spearman correlation matrix of the numeric features, converting it to a distance matrix \(1−\|corr\|1\-\|\\text\{corr\}\|\), and applying hierarchical clustering with average linkage\. During pretraining, an entire correlated group is masked simultaneously with a probability of 0\.5\.
### 4\.4Fine\-tuning
After SSL pretraining for 50 epochs on the unlabeled features, we fine\-tune the model on the labeled subset for 30 epochs using the AdamW optimizer\[[25](https://arxiv.org/html/2608.24381#bib.bib25)\]\. We employ a learning rate of10−310^\{\-3\}, a weight decay of10−410^\{\-4\}, and a batch size of up to 128 \(capped by the number of labeled examples available\)\.
## 5Experiments
### 5\.1Datasets
We evaluate our approach on 14 diverse OpenML\[[15](https://arxiv.org/html/2608.24381#bib.bib15)\]classification tasks\. The datasets vary significantly in training size and native missingness rates\.
### 5\.2Baselines
We compare our approach against:
- •Scratch \(No Pretrain\):The exact same neural architecture trained only on the labeled data\.
- •Classical baselines \(10% labels\):Logistic Regression, Random Forest, XGBoost, and LightGBM\.
### 5\.3Robustness evaluation
To measure robustness to missing data, we evaluate the models under three conditions:
1. 1\.Normal test performance\(no additional corruption\)\.
2. 2\.MCAR \+30% missingness:We inject additional missing entries uniformly at random \(MCAR\) with a 30% probability into the available test features\.
3. 3\.Missingness shift \(MNAR\):We apply a structured Missing Not At Random \(MNAR\) shift by masking all numeric values that exceed the 70th percentile of their respective column distributions\.
### 5\.4Statistical testing
We compute per\-task performance deltasΔ=AUCSSL−AUCscratch\\Delta=\\mathrm\{AUC\}\_\{\\mathrm\{SSL\}\}\-\\mathrm\{AUC\}\_\{\\mathrm\{scratch\}\}\. To assess statistical significance across datasets, we report the meanΔ\\Deltaand apply the Wilcoxon signed\-rank test, following standard practice for comparing classifiers across multiple independent datasets\[[20](https://arxiv.org/html/2608.24381#bib.bib20)\]\. Our main experiments involve four primary hypothesis tests—label scarcity at 5% and 10%, MCAR robustness, and MNAR shift robustness—which we treat as a single family and correct using the Holm\-Bonferroni procedure atα=0\.05\\alpha=0\.05\. We report both raw and Holm\-adjusted p\-values throughout; our comparison against prior tabular SSL baselines \(Section[6\.6](https://arxiv.org/html/2608.24381#S6.SS6)\) constitutes a separate family of three tests, corrected independently\.
## 6Results
### 6\.1Main results under label scarcity
Figure[1](https://arxiv.org/html/2608.24381#S6.F1)summarizes the average AUC\-ROC across tasks\. At 5% labels, SSL improves the average AUC by \+0\.0074 over scratch training\. At 10% labels, the average improvement is \+0\.0055\. A Wilcoxon signed\-rank test indicates that neither aggregate gain reaches significance \(p=0\.6257p=0\.6257at both 5% and 10%; Holm\-adjustedp=1\.000p=1\.000\)\.111The identicalpp\-value at both fractions is a mathematically valid coincidence resulting from the discrete probability distribution of the exact Wilcoxon signed\-rank test on 14 samples; it does not indicate an error in computation\.This confirms that SSL pretraining is not universally beneficial across all tabular manifolds\. Notably, this shrinking advantage crosses zero by 20% labels: the average SSL AUC \(0\.9145\) falls slightly below the scratch baseline \(0\.9173\) at this fraction, indicating that pretraining’s benefit is concentrated in the label\-scarce regime rather than persisting once more labeled data is available \(full per\-task, per\-fraction results are in Appendix[A](https://arxiv.org/html/2608.24381#A1)\)\.
Figure 1:Average test AUC\-ROC across tasks for SSL\-pretrained vs\. scratch training\.
### 6\.2The Missing vs\. Clean Paradox
A core hypothesis of tabular SSL is that mask\-and\-recover objectives teach the model to natively handle missing values\. However, Figure[2](https://arxiv.org/html/2608.24381#S6.F2)illustrates a counter\-intuitive finding: SSL yields the most reliable performance improvements on*clean*datasets\. On datasets with high inherent missingness, the impact of SSL is highly variable, frequently degrading performance\.
Figure 2:Per\-taskΔ\\DeltaAUC \(SSL−\-Scratch\) at 5% labels\. Datasets with\>1%\>1\\%native missingness are colored “Has Missing”; datasets at or below that threshold are colored “Clean\.”
### 6\.3Robustness to Test\-Time Degradation
Despite the variance in clean test settings, SSL\-pretrained models achieve higher raw AUC than scratch training on average under test\-time data degradation\. Under MCAR \+30% missingness \(injected into both numeric and categorical features\), SSL outperforms scratch by an average of \+0\.0245 AUC \(rawp=0\.0295p=0\.0295; Holm\-adjustedp=0\.118p=0\.118across our family of four primary tests\)\. Under a more severe MNAR missingness shift, SSL achieves a mean gain of \+0\.0418 AUC \(rawp=0\.1726p=0\.1726; Holm\-adjustedp=0\.518p=0\.518\)\. Neither comparison remains significant after correcting for multiple comparisons, though both show the same consistent direction and are the two largest effect sizes among our four primary tests\. Figure[3](https://arxiv.org/html/2608.24381#S6.F3)demonstrates this graceful degradation on Task 3021 across increasing missingness rates\.
Figure 3:AUC\-ROC on Task 3021 as additional missingness is injected at test time\.
### 6\.4Ablation study
We ablate the consistency loss, group masking, and mask\-visibility components at 10% labels on the four tasks with non\-trivial native missingness \(Task 3021, 22\.2%; Task 14954, 5\.3%; Task 2079, 4\.3%; Task 146800, 1\.7%\)\. An initial run trained each variant once per task using a single pretraining seed and appeared to show that removing the consistency loss improved performance on all four tasks\. Given the small number of tasks involved, we treated this as provisional and re\-ran every variant across three independent pretraining seeds \(matching then=3n=3seed protocol used throughout the rest of this paper\), averaging the resulting AUC\-ROC before computing deltas \(Figure[4](https://arxiv.org/html/2608.24381#S6.F4)\)\.
Under this seed\-averaged protocol, the apparent consistency\-loss effect does not replicate\. Removing consistency regularization no longer improves all four tasks: it helps on Task 3021 \(\+0\.0088\+0\.0088\) and Task 146800 \(\+0\.0098\+0\.0098\), but hurts on Task 2079 \(−0\.0037\-0\.0037\) and Task 14954 \(−0\.0025\-0\.0025\), for a meanΔ\\DeltaAUC of\+0\.0031\+0\.0031across tasks \(Wilcoxonp=0\.625p=0\.625,n=4n=4;p=0\.233p=0\.233across all 12 seed\-level pairs\)\. Group masking similarly shows no reliable direction \(meanΔ\\DeltaAUC=\+0\.0004=\+0\.0004,p=1\.0p=1\.0\): Task 3021 \(−0\.0022\-0\.0022\) and Task 14954 \(−0\.0024\-0\.0024\) degrade while Task 2079 \(\+0\.0016\+0\.0016\) and Task 146800 \(\+0\.0045\+0\.0045\) improve, all well within the noise observed across pretraining seeds \(per\-task standard deviations ranged from 0\.004 to 0\.015 AUC\)\. We interpret this as evidence that our original single\-seed ablation reflected uncontrolled variance between independently\-initialized pretraining runs rather than a genuine effect of the consistency loss or group masking on how the encoder handles native missingness\.
We separately tested whether the Missing vs\. Clean Paradox \(Section[6\.2](https://arxiv.org/html/2608.24381#S6.SS2)\) stems from the encoder never observing the\(x=0,m=1\)\(x\{=\}0,m\{=\}1\)signature of true missingness during pretraining, since our default corruption scheme always pairs synthetic masking withm=0m\{=\}0\(Section[4](https://arxiv.org/html/2608.24381#S4)\)\. We re\-ran pretraining with the pretext mask also written intomm\(“Mask\-Aware”\), again averaged over three pretraining seeds\. This effect was also inconsistent across the four tasks \(meanΔ\\DeltaAUC=\+0\.0011=\+0\.0011,p=0\.875p=0\.875: Task 3021\+0\.0105\+0\.0105, Task 2079−0\.0031\-0\.0031, Task 14954\+0\.0074\+0\.0074, Task 146800−0\.0104\-0\.0104\), indicating that mask\-visibility during pretraining is not a reliable driver of the paradox either\.
Overall, none of the three ablated components produced an effect distinguishable from noise once averaged across pretraining seeds\. We view this as an important methodological finding in its own right: with only four native\-missingness tasks and substantial per\-seed variance, single\-seed ablation comparisons in tabular SSL can produce confident\-looking but spurious patterns, and multi\-seed averaging is necessary before attributing an effect to a specific design choice\.
Figure 4:Ablation results at 10% labels across the four native\-missingness tasks \(missingness % in parentheses\), averaged over three independent pretraining seeds; error bars show±1\\pm 1standard deviation across seeds\. “Mask\-Aware” denotes pretraining where the pretext corruption mask is also passed to the encoder viamm\.
### 6\.5Classical baselines at 10% labels
Table[1](https://arxiv.org/html/2608.24381#S6.T1)compares our deep tabular models to classical tree\-based ensembles and linear models\. SSL pretraining elevates the neural network to be highly competitive with state\-of\-the\-art methods: it outperforms Logistic Regression by a wide margin \(\+0\.0688 AUC\), edges out XGBoost by a smaller \+0\.0135 AUC, closely matches LightGBM \(\+0\.0079 AUC\), and narrowly trails Random Forest\.
Table 1:Baseline comparison averaged across tasks at 10% labels\. Classical baselines use a single fixed seed \(42\); SSL Pretrained and Scratch use a single shared pretraining run fine\-tuned across 3 independent seeds \(see Limitations\)\.
### 6\.6Comparison against prior tabular SSL objectives
To assess whether our mask\-and\-recover objective is empirically distinguishable from established tabular SSL pretext tasks, we re\-implemented VIME\[[12](https://arxiv.org/html/2608.24381#bib.bib12)\], SCARF\[[13](https://arxiv.org/html/2608.24381#bib.bib13)\], and SubTab\[[14](https://arxiv.org/html/2608.24381#bib.bib14)\]using the identical dual\-branch encoder architecture described in Section[4](https://arxiv.org/html/2608.24381#S4), differing only in the pretraining objective\. This controls for architectural capacity so that any observed difference can be attributed to the pretraining objective itself\. Each baseline was pretrained and fine\-tuned at 10% labels across all 14 tasks, averaged over 3 independent pretraining seeds, each using its own \(not our\) batch size, LR schedule, and clipping defaults\.
Table[2](https://arxiv.org/html/2608.24381#S6.T2)reports the results\. None of the three baselines differ significantly from our method: VIME \(meanΔ=−0\.0024\\Delta=\-0\.0024, Wilcoxonp=0\.153p=0\.153\), SCARF \(meanΔ=\+0\.0011\\Delta=\+0\.0011,p=0\.670p=0\.670\), and SubTab \(meanΔ=\+0\.0041\\Delta=\+0\.0041,p=0\.502p=0\.502\); all remain non\-significant after Holm\-Bonferroni correction across this family of three tests \(adjustedp=0\.459p=0\.459,p=1\.000p=1\.000, andp=1\.000p=1\.000, respectively\)\. We interpret this as evidence that our objective is empirically competitive with, rather than distinguishable from, existing tabular SSL formulations under a controlled comparison, rather than as evidence that the choice of pretraining objective is inconsequential—a larger and more diverse task sample would be needed to detect smaller effects, if they exist\.
Table 2:Comparison against prior tabular SSL objectives using an identical encoder architecture\.Δ\\Deltais Ours−\-baseline; adjustedpp\-values are Holm\-Bonferroni corrected across this family of 3 tests\.
## 7Discussion
#### When does SSL help most?
Improvements peak at 5% labels and shrink as more labels become available, reversing sign entirely by 20% labels \(Section[6](https://arxiv.org/html/2608.24381#S6)\)\. More interestingly, our findings challenge the assumption that missing\-value imputation objectives universally benefit datasets with heavy native missingness\. Instead, we observed the strongest and most reliable gains on originally clean datasets\.
#### Why does robustness improve under missingness?
A mask\-and\-recover SSL objective forces the encoder to rely on redundant predictive structures across features, reducing sensitivity to missing entries at test time\. This is supported by the positive average gain observed under structured MNAR missingness shifts \(\+0\.0418 AUC, positive on 8 of 14 tasks\), though this effect does not reach significance after Holm\-Bonferroni correction \(adjustedp=0\.518p=0\.518\); we treat this as a suggestive rather than confirmatory finding\.
#### Why are some components not consistently helpful?
Extended ablations \(Section[6\.4](https://arxiv.org/html/2608.24381#S6.SS4)\) initially suggested that consistency regularization was consistently harmful on datasets with native missingness\. However, this effect did not survive averaging over multiple independent pretraining seeds: across three seeds per variant, the sign of the effect varied by task and the aggregate delta was statistically indistinguishable from zero\. The same was true for group masking and mask\-aware pretraining\. We take this as a caution against over\-interpreting single\-seed ablation comparisons in small\-sample tabular SSL studies, rather than as evidence that these design choices are inert—a larger task sample or more pretraining seeds per variant would be needed to reliably detect effects of this magnitude, if they exist at all\.
## 8Limitations and Broader Impact
#### Limitations\.
While self\-supervised pretraining demonstrates strong average performance and test\-time robustness, our empirical evaluation highlights several limitations\. First, the mask\-and\-recover objective does not universally benefit all tabular manifolds; on datasets with high native missingness, the injection of additional artificial missingness during pretraining can occasionally cause the model to underfit the available signal\. Second, the computational overhead of training a deep tabular encoder for 50 pretraining epochs and 30 fine\-tuning epochs is significantly higher than fitting a classical decision tree\. Third, there is an asymmetry in evaluation rigor: our deep tabular models are averaged over 3 independent seeds for the label\-scarcity, ablation, and SSL\-baseline results, but the MCAR and MNAR robustness deltas \(Section[6\.3](https://arxiv.org/html/2608.24381#S6.SS3)\) use a single fixed fine\-tuning seed \(42\) rather than a 3\-seed average, so those two numbers carry additional seed\-selection noise not present in our other primary tests; for the same reason, the "Δ\\Delta10%" column of Table[4](https://arxiv.org/html/2608.24381#A1.T4)\(single\-seed\) need not exactly match the 3\-seed\-averaged \+0\.0055 figure reported in Section[6\.3](https://arxiv.org/html/2608.24381#S6.SS3)\. The classical baselines in Table[1](https://arxiv.org/html/2608.24381#S6.T1)are likewise fit with a single fixed seed \(Appendix[C](https://arxiv.org/html/2608.24381#A3)\), so their reported scores do not reflect seed\-to\-seed variance the way the 3\-seed deep\-model numbers do\. Finally, our group masking strategy relies on linear correlation metrics \(Spearman rank\), which may fail to capture complex, non\-linear dependencies between features\. Relatedly, three tasks \(15 and 32 from 1% labels onward, and 6 from 5% labels onward\) sit at or above roughly 0\.98 AUC \(Task 6 is the exception at 1% labels, where all methods score below 0\.93\); these near\-ceiling tasks necessarily contribute small, low\-variance deltas to our paired statistical tests, which slightly dilutes the effective sample even thoughn=14n=14is used nominally throughout\. We also tested, and could not confirm, a specific mechanistic explanation for the Missing vs\. Clean Paradox \(mask\-visibility during pretraining, Section[6\.4](https://arxiv.org/html/2608.24381#S6.SS4)\); the paradox’s root cause remains an open question for future work\. Additionally, our ablation study is limited to four native\-missingness tasks; even after averaging over three independent pretraining seeds per variant, this sample size affords limited statistical power to detect small effects \(a Wilcoxon test on four paired samples cannot reachp<0\.05p<0\.05regardless of effect size\), so our null findings for the consistency loss, group masking, and mask\-aware components should be read as an absence of detected effect rather than definitive proof of no effect\. Similarly, our comparison against prior tabular SSL objectives \(Section[6\.6](https://arxiv.org/html/2608.24381#S6.SS6)\) is restricted to 10% labels and does not rule out differences that might emerge at other label fractions or with per\-method hyperparameter tuning; we deliberately did not tune hyperparameters per dataset or per method \(Appendix[D](https://arxiv.org/html/2608.24381#A4)\) in order to isolate the pretraining objective as the variable of interest, which is a defensible but real constraint on how far these comparisons generalize\. Finally, all p\-values in this paper are corrected for multiple comparisons within their respective families \(Section[5](https://arxiv.org/html/2608.24381#S5)\); readers should interpret the raw, uncorrected effect sizes \(e\.g\., the \+0\.0245 AUC gain under MCAR\) as suggestive rather than confirmatory, given that none of our primary robustness comparisons remain significant after correction\.
#### Broader Impact\.
The ability to train robust models under extreme label scarcity has profound implications for domains where labeling requires expensive human expertise, such as medical diagnostics and fraud detection\. Furthermore, achieving zero\-shot robustness to test\-time sensor degradation \(MNAR shifts\) reduces the likelihood of catastrophic algorithmic failures in production environments\. However, practitioners must remain cautious; deep tabular models are inherently less interpretable than shallow decision trees, and deploying them in high\-stakes environments requires secondary explainability audits to ensure predictions do not rely on spurious correlations\.
## 9Conclusion
In this paper, we presented a rigorous empirical evaluation of self\-supervised pretraining for tabular deep learning under extreme label scarcity and test\-time missingness\. Our analysis across diverse OpenML datasets revealed that while mask\-and\-recover SSL objectives generally outperform scratch training and are highly competitive with classical tree ensembles, these performance gains exhibit significant inter\-task variance\. Notably, we uncovered a "clean vs\. missing" paradox: imputation\-based pretext tasks yield the most reliable downstream improvements on natively clean datasets, rather than those with inherent missingness\. Despite this variance, SSL\-pretrained models achieved a higher average AUC under test\-time degradation, degrading more gracefully than scratch\-trained models under both random \(MCAR\) missingness injections and structured \(MNAR\) test\-time shifts on average, though neither difference remains significant after correcting for multiple comparisons, nor is the effect uniform across all 14 tasks\. We additionally found that our mask\-and\-recover objective performs comparably to three established tabular SSL baselines—VIME, SCARF, and SubTab—under a controlled, architecture\-matched comparison, indicating our findings are not an artifact of an unusually weak or strong pretraining objective\. Finally, our extended, seed\-averaged ablations found no reliable effect of view\-consistency regularization, group masking, or mask\-aware pretraining on native\-missingness tasks—an initial single\-seed comparison had suggested a consistent harmful effect of the consistency loss, but this did not replicate once properly averaged across independent pretraining seeds, underscoring the importance of multi\-seed evaluation in small\-sample tabular SSL ablation studies\. Ultimately, our findings establish a more nuanced understanding of when and why self\-supervised learning succeeds, highlighting the need for dataset\-adaptive SSL designs in real\-world tabular data mining\.
#### Future Work\.
Several directions follow naturally from these findings\. First, given that our seed\-averaged ablations found no reliable effect for the consistency loss or group masking on the four native\-missingness tasks studied here, a larger and more diverse pool of native\-missingness datasets would provide the statistical power needed to determine whether these components have a genuine, if small, effect, or whether they are truly inert with respect to missingness handling\. Second, the root cause of the Missing vs\. Clean Paradox remains open: future work could investigate whether alternative corruption schemes, such as explicitly modeling the joint distribution of synthetic and native missingness rather than treating them as interchangeable, might resolve the paradox where our mask\-aware ablation did not\. Third, extending this evaluation protocol to a broader set of pretraining objectives beyond mask\-and\-recover, such as contrastive or generative approaches, would help determine whether the robustness gains and paradoxical behavior we observe are specific to this pretraining paradigm or are a more general property of self\-supervised tabular representations\. We hope the dataset\-adaptive perspective and the seed\-averaging methodology adopted here prove useful to future empirical studies in this area\.
## Declarations
Funding:This research was conducted independently; no external funding or grants were received\. Conflicts of Interest:The author declares no competing interests\. Data Availability:All datasets utilized in this study are publicly accessible via the OpenML repository\. Code Availability:The complete PyTorch implementation, experimental pipeline, and evaluation scripts used to generate the results in this manuscript are available at[https://github\.com/shnd23/ssl\-tabular\-label\-scarcity](https://github.com/shnd23/ssl-tabular-label-scarcity)\.
## References
- \[1\]Chen, T\., & Guestrin, C\. \(2016\)\. XGBoost: A scalable tree boosting system\.*Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining*, 785–794\.
- \[2\]Ke, G\., Meng, Q\., Finley, T\., Wang, T\., Chen, W\., Ma, W\., … & Liu, T\. Y\. \(2017\)\. LightGBM: A highly efficient gradient boosting decision tree\.*Advances in Neural Information Processing Systems*, 30\.
- \[3\]Breiman, L\. \(2001\)\. Random forests\.*Machine Learning*, 45\(1\), 5–32\.
- \[4\]Arik, S\. O\., & Pfister, T\. \(2021\)\. TabNet: Attentive interpretable tabular learning\.*Proceedings of the AAAI Conference on Artificial Intelligence*, 35\(8\), 6679–6687\.
- \[5\]Gorishniy, Y\., Rubachev, I\., Khrulkov, V\., & Babenko, A\. \(2021\)\. Revisiting deep learning models for tabular data\.*Advances in Neural Information Processing Systems*, 34, 18932–18943\.
- \[6\]Somepalli, G\., Goldblum, M\., Schwarzschild, A\., Bruss, C\. B\., & Goldstein, T\. \(2021\)\. SAINT: Improved neural networks for tabular data via row attention and contrastive pre\-training\.*arXiv preprint arXiv:2106\.01342*\.
- \[7\]Grinsztajn, L\., Oyallon, E\., & Varoquaux, G\. \(2022\)\. Why do tree\-based models still outperform deep learning on typical tabular data?*Advances in Neural Information Processing Systems*, 35\.
- \[8\]Gorishniy, Y\., Rubachev, I\., & Babenko, A\. \(2022\)\. On embeddings for numerical features in tabular deep learning\.*Advances in Neural Information Processing Systems*, 35, 24991–25004\.
- \[9\]Huang, X\., Khetan, A\., Cvitkovic, M\., & Karnin, Z\. \(2020\)\. TabTransformer: Tabular data modeling using contextual embeddings\.*arXiv preprint arXiv:2012\.06678*\.
- \[10\]Kossen, J\., Band, N\., Lyle, C\., Gomez, A\. N\., Rainforth, T\., & Gal, Y\. \(2021\)\. Self\-attention between datapoints: Going beyond individual input\-output pairs in deep learning\.*Advances in Neural Information Processing Systems*, 34\.
- \[11\]Borisov, V\., Leemann, T\., Seßler, K\., Haug, J\., Pawelczyk, M\., & Kasneci, G\. \(2024\)\. Deep neural networks and tabular data: A survey\.*IEEE Transactions on Neural Networks and Learning Systems*, 35\(6\), 7499–7519\.
- \[12\]Yoon, J\., Zhang, Y\., Jordon, J\., & van der Schaar, M\. \(2020\)\. VIME: Extending the success of self\-and semi\-supervised learning to tabular domain\.*Advances in Neural Information Processing Systems*, 33, 11033–11043\.
- \[13\]Bahri, D\., Jiang, H\., Tay, Y\., & Metzler, D\. \(2021\)\. SCARF: Self\-supervised contrastive learning using random feature corruption\.*International Conference on Learning Representations*\.
- \[14\]Ucar, T\., Hajiramezanali, E\., & Edwards, L\. \(2021\)\. SubTab: Subsetting features of tabular data for self\-supervised representation learning\.*Advances in Neural Information Processing Systems*, 34, 18853–18865\.
- \[15\]Vanschoren, J\., van Rijn, J\. N\., Bischl, B\., & Torgo, L\. \(2013\)\. OpenML: Networked science in machine learning\.*SIGKDD Explorations Newsletter*, 15\(2\), 49–60\.
- \[16\]Pedregosa, F\., Varoquaux, G\., Gramfort, A\., Michel, V\., Thirion, B\., Grisel, O\., … & Duchesnay, E\. \(2011\)\. Scikit\-learn: Machine learning in Python\.*Journal of Machine Learning Research*, 12, 2825–2830\.
- \[17\]Devlin, J\., Chang, M\. W\., Lee, K\., & Toutanova, K\. \(2018\)\. BERT: Pre\-training of deep bidirectional transformers for language understanding\.*arXiv preprint arXiv:1810\.04805*\.
- \[18\]He, K\., Chen, X\., Xie, S\., Li, Y\., Dollár, P\., & Girshick, R\. \(2022\)\. Masked autoencoders are scalable vision learners\.*Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition*, 16000–16009\.
- \[19\]Little, R\. J\. A\., & Rubin, D\. B\. \(2002\)\.*Statistical analysis with missing data*\(2nd ed\.\)\. John Wiley & Sons\.
- \[20\]Demšar, J\. \(2006\)\. Statistical comparisons of classifiers over multiple data sets\.*Journal of Machine Learning Research*, 7\(1\), 1–30\.
- \[21\]Troyanskaya, O\., Cantor, M\., Sherlock, G\., Brown, P\., Hastie, T\., Tibshirani, R\., … & Altman, R\. B\. \(2001\)\. Missing value estimation methods for DNA microarrays\.*Bioinformatics*, 17\(6\), 520–525\.
- \[22\]van Buuren, S\., & Groothuis\-Oudshoorn, K\. \(2011\)\. mice: Multivariate imputation by chained equations in R\.*Journal of Statistical Software*, 45\(3\), 1–67\.
- \[23\]Yoon, J\., Jordon, J\., & van der Schaar, M\. \(2018\)\. GAIN: Missing data imputation using generative adversarial nets\.*Proceedings of the 35th International Conference on Machine Learning*, 80, 5689–5698\.
- \[24\]Stekhoven, D\. J\., & Bühlmann, P\. \(2012\)\. MissForest—non\-parametric missing value imputation for mixed\-type data\.*Bioinformatics*, 28\(1\), 112–118\.
- \[25\]Loshchilov, I\., & Hutter, F\. \(2019\)\. Decoupled weight decay regularization\.*International Conference on Learning Representations \(ICLR\)*\.
- \[26\]Smith, L\. N\., & Topin, N\. \(2019\)\. Super\-convergence: Very fast training of neural networks using large learning rates\.*Artificial Intelligence and Machine Learning for Multi\-Domain Operations Applications*, 11006, 369–386\.
## Appendix APer\-task Results and Robustness Table
The following tables provide the exact, per\-task evaluation metrics used to compute the aggregate summaries presented in the main text\. Table[3](https://arxiv.org/html/2608.24381#A1.T3)details the test AUC\-ROC performance of both the Self\-Supervised Learning \(SSL\) pretrained encoder and the Scratch baseline across four distinct label scarcity regimes \(1%, 5%, 10%, and 20%\)\. Table[4](https://arxiv.org/html/2608.24381#A1.T4)details the exactΔ\\DeltaAUC \(SSL−\-Scratch\) under the normal 10% labeled setting, alongside the two test\-time degradation settings: Missing Completely At Random \(MCAR \+30%\) and the structured Missing Not At Random \(MNAR\) shift\.
Table 3:Per\-task AUC\-ROC results for SSL pretraining vs\. scratch training across four label scarcity fractions\. Metrics represent the mean over 3 random seeds and are rounded to 3 decimals for readability; statistical tests use full\-precision values, available in the accompanying code\.Table 4:Robustness deltas at 10% labels \(seed 42\)\. Values representΔ\\DeltaAUC \(SSL−\-Scratch\) under normal, MCAR \(\+30% missingness\), and MNAR structured shift\.
## Appendix BData Dictionary and Context
To ensure the broad applicability of our empirical findings, we curated 14 datasets from the OpenML repository spanning diverse domains, feature dimensionalities, and missingness properties\. A brief context for each dataset is provided below:
Task 2079 \(Eucalyptus\):An ecological dataset measuring the survival and growth of Eucalyptus species across different environments, containing categorical soil and genetic features alongside substantial missing sensor readings \(4\.3% missingness\)\.
Task 3021 \(Sick\):A medical dataset aimed at identifying sick patients based on thyroid disease indicators\. It contains a high degree of missingness \(22\.2%\) due to patients frequently lacking full diagnostic workups\.
Task 14954 \(Cylinder Bands\):An industrial manufacturing dataset predicting rotogravure printing press delays \(cylinder banding\)\. It features a mix of nominal and numeric features with 5\.3% missingness\.
Task 146800 \(MiceProtein\):A biological dataset tracking expression levels of 77 proteins in the cerebral cortex of mice \(1\.7% missingness\), used to identify subpopulations subject to specific learning contexts\.
Task 2 \(Anneal\):A metallurgical dataset used to predict the annealing process of steel\. Although the raw dataset is infamous for heavy structural missingness in its categorical variables, thepreprocessednumerical subset evaluated here is clean \(0\.0% missingness\)\.
Task 15 \(Breast\-W\):The original Wisconsin Breast Cancer dataset, using nine cytological characteristics \(e\.g\., clump thickness, uniformity of cell size and shape, bare nuclei\) manually scored on a 1–10 scale by a cytologist, rather than features computed via image processing\.
Task 29 \(Credit\-Approval\):A financial dataset analyzing credit card applications, featuring mixed data types and minor missingness related to anonymized applicant history\.
Task 3904 \(JM1\):A software engineering dataset predicting software defects based on Halstead and McCabe complexity metrics extracted from C code\.
Task 7592 \(Adult\):A census\-based demographic dataset used to predict whether a person’s income exceeds $50K/year\. It is a highly cited benchmark in both tabular learning and algorithmic fairness\.
Task 6 \(Letter\):An image\-derived tabular dataset where the objective is to identify English capital letters from 16 statistical features extracted from raster scan images\.
Task 32 \(Pendigits\):Pen\-based recognition of handwritten digits, utilizing coordinate information collected from a pressure\-sensitive tablet\.
Task 37 \(Diabetes\):The Pima Indians Diabetes dataset, containing diagnostic measurements to predict the onset of diabetes mellitus within 5 years\.
Task 43 \(Spambase\):A cybersecurity dataset used to classify emails as spam or non\-spam based on word frequencies and character run\-length metrics\.
Task 53 \(Vehicle\):A computer vision\-derived tabular dataset classifying 3D vehicle silhouettes \(e\.g\., Saab, Van, Bus\) based on geometric features\.
## Appendix CBaseline Implementation Details
In Section[6\.5](https://arxiv.org/html/2608.24381#S6.SS5), we compared our SSL pretraining against state\-of\-the\-art tree\-based ensembles and linear models\. To ensure maximum reproducibility, we detail the exact parameter configurations below\.
For all classical baselines, categorical variables were strictly isolated, cast to string data types to avoid float coercion errors, and processed using Scikit\-Learn’s\[[16](https://arxiv.org/html/2608.24381#bib.bib16)\]OrdinalEncoder\.Logistic Regression:We utilized Scikit\-Learn’sLogisticRegressionclass\. Due to the high dimensionality and variance of some datasets, themax\_iterparameter was explicitly raised to 1000 to ensure gradient descent convergence\.
Random Forest:We employed Scikit\-Learn’sRandomForestClassifier\. To provide a highly competitive baseline, we fixed the ensemble size ton\_estimators=100and fully utilized parallel processing vian\_jobs=\-1\. Node splitting criteria remained at the default Gini impurity\.
XGBoost:We used theXGBClassifierfrom the nativexgboostPython package\. The evaluation metric was explicitly set toeval\_metric=’logloss’\. The label encoder deprecation warning was bypassed viause\_label\_encoder=False\. All other boosting hyperparameters \(e\.g\., maximum depth, learning rate\) were kept at their highly optimized default values\.
LightGBM:We employed theLGBMClassifier\. To prevent memory fragmentation and console spam during the iteration over the 14 datasets, the model was executed withverbose=\-1\.
All classical baselines \(Logistic Regression, Random Forest, XGBoost, LightGBM\) were initialized with a fixed random seed of 42 to guarantee deterministic behavior; this is distinct from the 3\-seed protocol used for the deep tabular models \(Section[5](https://arxiv.org/html/2608.24381#S5)\)\. We note that our classical baselines use simple statistical imputation rather than learned imputation methods such as GAIN\[[23](https://arxiv.org/html/2608.24381#bib.bib23)\]or MissForest\[[24](https://arxiv.org/html/2608.24381#bib.bib24)\]; the choice of imputation strategy for classical baselines, and the broader question of why tree ensembles remain competitive with deep tabular models\[[7](https://arxiv.org/html/2608.24381#bib.bib7)\], are surveyed at length elsewhere\[[11](https://arxiv.org/html/2608.24381#bib.bib11)\]and are outside the scope of this comparison\.
## Appendix DExtended Hyperparameter Configurations
To ensure fair comparison and reproducibility, we detail the complete hyperparameter configurations and initialization bounds used for both the deep tabular models and the classical baselines\.
### D\.1Deep Tabular Encoder and SSL Objective
The deep tabular architecture was kept strictly uniform across all 14 datasets to isolate the impact of the self\-supervised pretraining objective\. No dataset\-specific architecture tuning was performed\. Our dual\-branch MLP encoder is deliberately simple relative to attention\-based tabular architectures such as TabTransformer\[[9](https://arxiv.org/html/2608.24381#bib.bib9)\], SAINT\[[6](https://arxiv.org/html/2608.24381#bib.bib6)\], and Non\-Parametric Transformers\[[10](https://arxiv.org/html/2608.24381#bib.bib10)\], and relative to learned numerical\-feature embeddings\[[8](https://arxiv.org/html/2608.24381#bib.bib8)\]; this choice keeps the pretraining objective, rather than architectural capacity, as the primary variable under study\. Hyperparameter values \(corruption rateρ=0\.3\\rho=0\.3, embedding dimension 32, hidden dimension 128, learning rate10−310^\{\-3\}\) follow common choices in the tabular SSL literature rather than being tuned per dataset or per method; we did not perform a hyperparameter search on any individual task, since doing so would confound the cross\-task comparisons that are the focus of this study\. This is a deliberate methodological choice rather than an oversight, though we acknowledge in Section[8](https://arxiv.org/html/2608.24381#S8)that dataset\-specific tuning could interact with the Missing vs\. Clean Paradox we report\.
- •Numeric Multi\-Layer Perceptron:2 layers, hidden dimension 128, Layer Normalization applied before ReLU activations, Dropout ratep=0\.1p=0\.1\.
- •Categorical Embeddings:DimensionalityE=32E=32per feature\. Unknown categories at test time mapped to an index of 0\.
- •Fusion Multi\-Layer Perceptron:2 layers, hidden dimension 128, Layer Normalization, Dropout ratep=0\.1p=0\.1\.
- •Pretraining Optimizer:AdamW\[[25](https://arxiv.org/html/2608.24381#bib.bib25)\], Learning Rate=10−3=10^\{\-3\}, Weight Decay=10−4=10^\{\-4\}, Batch Size=256=256, Grad\. Clip1\.01\.0, OneCycleLR\[[26](https://arxiv.org/html/2608.24381#bib.bib26)\]\.
- •Corruption Rate:Bernoulli masking parameterρ=0\.3\\rho=0\.3\.
- •Consistency Weight:λ=1\.0\\lambda=1\.0\(ablated to0\.00\.0; see Section[6\.4](https://arxiv.org/html/2608.24381#S6.SS4)for results averaged over 3 independent pretraining seeds\)\.
### D\.2Tree\-Based Ensembles
The classical baselines were instantiated using their respective Python libraries\. To provide a rigorous baseline, we ensured all tree ensembles utilized 100 estimators, providing sufficient capacity to model the tabular manifolds without excessive computational overhead\.
- •Random Forest \(Scikit\-Learn\):n\_estimators= 100,criterion= ’gini’,max\_depth= None,min\_samples\_split= 2,min\_samples\_leaf= 1,max\_features= ’sqrt’,bootstrap= True\.
- •XGBoost:n\_estimators= 100,learning\_rate= 0\.3,max\_depth= 6,subsample= 1\.0,colsample\_bytree= 1\.0,eval\_metric= ’logloss’\.
- •LightGBM:n\_estimators= 100,learning\_rate= 0\.1,num\_leaves= 31,boosting\_type= ’gbdt’\.Similar Articles
When, Where, and How: Adaptive Binning for Tabular Self-Supervised Learning
This paper proposes Adaptive Binning, a learning-coupled feature-wise coarse-to-fine curriculum for tabular self-supervised learning that adaptively discretizes features, improving representations on medical datasets and establishing a unified benchmark.
When Tabular Foundation Models Meet Strategic Tabular Data: A Prior Alignment Approach
This paper studies whether tabular foundation models based on pretrained prior-data fitted networks (PFNs) can generalize to strategic tabular data where individuals modify features after deployment. It proposes Strategic Prior-data Fitted Network (SPN), an inference-time framework that aligns PFN predictions with the post-manipulation distribution without retraining.
The Good, the Bad, and the Ugly of Markov Boundary for Tabular Prediction
This paper evaluates the practical effectiveness of Markov boundaries for tabular prediction, finding that while theoretically optimal, current causal discovery methods fail to consistently improve predictive performance due to computational limitations and mismatched optimization goals.
Can Tabular In-Context Learners Generalize to Biomolecular Property Prediction?
This paper investigates whether tabular in-context learning models, pretrained on synthetic causal tables, can generalize to predict biomolecular properties from limited labeled data. The authors find that these models are competitive for protein fitness regression but that representation choice is crucial for small-molecule classification.
Why Large Language Models Fail at Tabular Prediction
A new arXiv paper systematically tests five hypotheses for why large language models fail at tabular prediction, finding that dimensionality is the decisive factor: LLM accuracy degrades as input dimension grows, unlike classical baselines that stay flat or improve.