AdapTable: Test-Time Adaptation for Tabular Data
via Shift-Aware Uncertainty Calibrator and Label Distribution Handler
Abstract
In real-world scenarios, tabular data often suffer from distribution shifts that threaten the performance of machine learning models. Despite its prevalence and importance, handling distribution shifts in the tabular domain remains underexplored due to the inherent challenges within the tabular data itself. In this sense, test-time adaptation (TTA) offers a promising solution by adapting models to target data without accessing source data, crucial for privacy-sensitive tabular domains. However, existing TTA methods either 1) overlook the nature of tabular distribution shifts, often involving label distribution shifts, or 2) impose architectural constraints on the model, leading to a lack of applicability. To this end, we propose AdapTable, a novel TTA framework for tabular data. AdapTable operates in two stages: 1) calibrating model predictions using a shift-aware uncertainty calibrator, and 2) adjusting these predictions to match the target label distribution with a label distribution handler. We validate the effectiveness of AdapTable through theoretical analysis and extensive experiments on various distribution shift scenarios. Our results demonstrate AdapTable’s ability to handle various real-world distribution shifts, achieving up to a 16% improvement on the HELOC dataset.
1 Introduction
Tabular data is one of the most abundant forms across various industries, including healthcare (Johnson et al. 2016), finance (Studies 2019), manufacturing (Hein et al. 2017), and public administration (Gardner, Popovic, and Schmidt 2023). However, tabular learning models often face challenges in real-world applications due to distribution shifts, which severely degrade their integrity and reliability. In this regard, test-time adaptation (TTA) (Lee 2013; Liu et al. 2021; Wang et al. 2021a; Niu et al. 2022, 2023; Boudiaf et al. 2022) offers a promising solution to address this issue by adapting models under unknown distribution shifts using only unlabeled test data without access to training data.
Despite its potential, direct application of TTA without the consideration of characteristics of tabular data, results in limited performance gain or model collapse. We identify two primary reasons for this. First, representation learning in the tabular domain is often hindered by the entanglement of covariate shifts and concept drifts (Liu et al. 2023). Consequently, TTA methods leveraging unsupervised objectives, which rely on cluster assumption often fail or lead to model collapse. Second, these approaches often do not take label distribution shifts into account, a key factor in the performance decline within the tabular domain. This issue is further aggravated by the tendency for predictions in the target domain to be biased towards the source label.
To address these issues, we propose AdapTable, a novel TTA approach tailored for tabular data. AdapTable consists of two main components: 1) a shift-aware uncertainty calibrator and 2) a label distribution handler. Our shift-aware uncertainty calibrator utilizes graph neural networks to assign per-sample temperature for each model prediction. By treating each column as a node, it captures not only individual feature shifts but also complex patterns across features. Our label distribution handler then adjusts the calibrated model probabilities by estimating the label distribution of the current target batch. This process aligns predictions with the target label distribution, addressing biases towards the source distribution. AdapTable requires no parameter updates, making it model-agnostic and thus compatible with both deep learning models and gradient-boosted decision trees, offering high versatility for tabular data.
We evaluate AdapTable under various distribution shifts and demonstrate AdapTable consistently outperforms baselines, achieving up to 16% gains on the HELOC dataset. Furthermore, we provide theoretical insights into AdapTable’s performance, supported by extensive ablation studies. We hereby summarize our contributions:
-
•
We analyze the challenges of tabular distribution shifts to reveal why existing TTA methods fail, highlighting the entanglement of covariate, concept shifts, and label distribution shifts as key factors in performance degradation.
-
•
Building on these analyses, we introduce AdapTable, a first model-agnostic TTA method specifically designed for tabular data. AdapTable addresses label distribution shifts by estimating and adjusting the label distribution of the current test batch, while also calibrating model predictions with a shift-aware uncertainty calibrator.
-
•
Our extensive experiments demonstrate that AdapTable exhibits robust adaptation performance across various model architectures and under diverse natural distribution shifts and common corruptions, further supported by extensive ablation studies.
2 Analysis of Tabular Distribution Shifts
In this section, we examine why prior TTA methods struggle with distribution shifts in the tabular domain. First, we note that deep learning models’ latent representations do not follow label-based cluster assumptions due to the entanglement of covariate and concept shifts, causing TTA methods relying on these assumptions (Wang et al. 2021a; Lee 2013; Niu et al. 2022) to falter in tabular data. Second, we identify label distribution shift as a key driver of performance degradation under distribution shifts, as discussed further in Section 2.1 and Section 2.2.
2.1 Indistinguishable Representations
We first reveal that deep tabular models fail to learn distinguishable embeddings. In Figures 1 (a) and (b), we visualize the embedding spaces of models trained on two datasets: HELOC (Gardner, Popovic, and Schmidt 2023), a pure tabular dataset, and Optdigits (Bischl et al. 2021), a linearized image dataset. Notably, the deep learning models’ representations adhere to the cluster assumption by labels only in the image data, not in the tabular data.
We attribute this unique behavior of deep tabular models to the high-frequency nature of tabular data. In the tabular domain, weak causality from inputs to outputs due to latent confounders often leads to vastly different labels for similar inputs (Grinsztajn, Oyallon, and Varoquaux 2022; Liu et al. 2023). For instance, cardiovascular disease risk predictions based on cholesterol, blood pressure, age, and smoking history are influenced by gender as a latent confounder, resulting in different risk levels for men and women despite identical inputs (Mosca, Barrett-Connor, and Kass Wenger 2011; DeFilippis and Van Spall 2021). This leads to high-frequency functions that are difficult for deep neural networks, which are biased toward low-frequency functions, to accurately model (Beyazit et al. 2024).
Consequently, prior TTA methods, which rely on cluster assumptions and primarily target input covariate shifts, show limited performance gains. Figure 5 demonstrates that these methods fail to improve beyond the vanilla performance of the source model due to the lack of a cluster assumption.
2.2 Importance of Label Distribution Shifts
Second, we find that label distribution shift is a primary cause of performance degradation, and accurate estimation of target label distribution can lead to significant performance gains. A recent benchmark study, TableShift (Gardner, Popovic, and Schmidt 2023) have emphasized that label distribution shift is a primary cause of performance degradation in tabular data. Specifically, They investigated the relationship between three key shift factors—input covariate shift (-shift), concept shift (-shift), and label distribution shift (-shift)—and model performance, and discovered that label distribution shifts are strongly correlated with performance degradation. Our analysis in Section F further reveals that these shifts are highly prevalent in tabular data. This underscores the need for a test-time adaptation method that addresses label distribution shifts by estimating the target label distribution and adjusting predictions accordingly.
Moreover, we visualize model predictions in Figure 2 and observe that, similar to other domains (Wu et al. 2021; Hwang et al. 2022; Park, Seo, and Yang 2023), the marginal distribution of output labels is biased toward the source label distribution. Given that tabular models are often poorly calibrated (Figure 1), we conduct an experiment using a perfectly calibrated model, which yields high confidence for correct samples and low confidence for incorrect ones. As shown in Table 1, our label distribution adaptation method significantly improves under these conditions. This underscores the need for an uncertainty calibrator specific to tabular data.
3 AdapTable
This section introduces AdapTable, the first model-agnostic test-time adaptation strategy for tabular data. AdapTable uses per-sample temperature scaling to correct overconfident yet incorrect predictions; by treating each column as a graph node, it employs a shift-aware uncertainty calibrator with graph neural networks to capture both individual and complex feature shifts (Section 3.2). It also estimates the average label distribution of the current test batch and adjusts the model’s output predictions accordingly (Section 3.3). We also provide a theoretical justification for how our label distribution estimation reduces the error bound in Section 3.4. The overall framework of AdapTable is depicted in Figure 3.
3.1 Test-Time Adaptation Setup for Tabular Data
We begin by defining the problem setup for test-time adaptation (TTA) for tabular data. Let be a pre-trained classifier on the labeled source tabular domain , where each pair consists of a tabular input and its corresponding output class label . The classifier takes a row from a table and returns output logit . Here, and are the number of input features and output classes, respectively. The objective of TTA for tabular data is to adapt to the unlabeled target tabular domain during inference, without access to . Unlike most TTA methods that fine-tune model parameters with unsupervised objectives, our approach directly adjusts the output prediction .
3.2 Shift-Aware Uncertainty Calibrator
This section describes a shift-aware uncertainty calibrator designed to adjust the poorly calibrated original predictions , where normalizes the logits. Our shift-aware uncertainty calibrator lowers the confidence of overconfident yet incorrect predictions, thereby 1) facilitating better alignment of these predictions with the estimated target label distribution, and 2) mitigating their impact on the inaccurate estimation of the target label distribution.
Conventional post-hoc calibration methods (Platt 2000; Stylianou and Flournoy 2002) typically take solely the original model prediction as input and return the corresponding temperature without taking input variations into account. We argue that this can be suboptimal as it fails to account for the uncertainty arising from variations in the input itself. Instead, our not only considers but also incorporates with the shift trend of the current batch as additional inputs. Capturing the common shift patterns within the current batch enables a more accurate reflection of the uncertainty caused by the overall shift patterns within the current batch.
In detail, the shift trend is defined for a specific column index as follows:
(1) |
Here, represents the difference between the values of the -th column within the current test batch and the average values of the corresponding column in the source data. Using for each column , we define a shift trend graph where each node represents a column, and each edge captures the relationship between different columns; the node feature for each node is defined as , and the adjacency matrix is represented by an all-ones matrix.
A graph neural network (GNN) is then applied to the graph formed above, enabling the exchange of shift trends between different columns through message passing. This process generates a column-wise contextualized representation, which is then averaged to produce an overall feature representation that encompasses all columns. Finally, the averaged node representation is concatenated with the initial prediction to yield the final output temperature . This GNN-based uncertainty calibration not only captures shifts in individual columns but also sensitively detects correlation shifts occurring simultaneously across different columns, which are common in the tabular domain. A more detailed explanation of the architecture and training of the shift-aware uncertainty calibrator can be found in Section A.
3.3 Label Distribution Handler
This section introduces a label distribution handler designed to accurately estimate the target label distribution for the current test batch and adjust the model’s output predictions accordingly. This approach is empirically justified by our observation that the marginal distribution of model predictions in the target domain tends to be biased towards the source label distribution , as discussed in Section 2.2 and illustrated in Figure 2.
A straightforward solution to correct this bias is to simply multiply to align the marginal label distribution (Berthelot et al. 2020). Specifically, given , the adjusted prediction would be:
(2) |
where normalizes the unnormalized probability. However, we find two major issues: 1) is often poorly calibrated and 2) overconfident yet incorrect predictions significantly hinder the accurate estimation of the target label distribution (Section 2.2).
To tackle these challenges, we propose a simple yet effective estimator defined like below:
(3) |
The key differences between the original Equation 2 and our Equation 3 are: 1) we use the calibrated prediction instead of the original prediction to enhance uncertainty quantification, and 2) we combine the calibrated estimate with the distributionally aligned prediction for more robust estimation.
Given the already-known source label distribution , we now explain the step-by-step process for estimating and . is calibrated into through a two-stage uncertainty calibration process. Specifically, for a current test batch , we calculate shift trend using Equation 1 and get per-sample temperature using shift-aware uncertainty calibrator to capture overall distribution shifts, as well as correlation and individual column shifts within the current batch. Here, we define the uncertainty of as a reciprocal of the margin of the calibrated probability distribution . We then measure the quantiles for each instance using within the current batch and recalibrate the original probability with , resulting in . This process calibrates predictions by leveraging relative uncertainty within the batch. Our temperature is defined as:
(4) |
where is a quantile function which gives the value corresponding to the lower quantile in , is a temperature, and and represent the low and high uncertainty quantiles, respectively. This two-stage uncertainty calibration comprehensively evaluates the current batch and estimates relative uncertainty using , , and .
Meanwhile, the target label distribution is estimated as follows:
(5) |
where is a debiased target label estimator that departs from , and is an online target label estimator, initialized as uniform distribution and updated as:
(6) |
from the current batch to the next, with a smoothing factor . This online target label estimator leverages label locality between nearby test batches, making it effective for accurately estimating the next batch’s target label distribution. A more detailed explanation of AdapTable is provided in Section A.
3.4 Theoretical Insights
Theorem 3.1.
Let and be defined as follows:
(7) | ||||
(8) |
Given the error , with true labels of inputs , the error gap is upper bounded by
(9) |
where and are constants related to , and , respectively.
Theorem 3.1 extends Theorem 2.3 in ODS (Zhou et al. 2023) to cases where the source label distribution is not uniform. It decomposes the error gap between the original model on the source domain and the adapted model for the target model with on the target domain into several components. These components include , which is an error of the estimated target label distribution, , which reflects the model’s performance on the source domain, and , which measures the generalization of feature representations adapted by the TTA algorithm. Overall, Theorem 3.1 underscores the importance of tracking label distributions and efficiently adapting models to handle label distribution shifts. The detailed explanation and proof of Theorem 3.1 can be found in Section B.
4 Experiments
This section validates AdapTable’s effectiveness. We begin with an overview of our experimental setup in Section 4.1 and then address key research questions:
-
•
Is AdapTable effective across various tabular distribution shifts, including natural shifts and common corruptions across different tabular models? (Section 4.2)
-
•
Do AdapTable’s components contribute to overall performance improvements, and do they function as intended? (Section 4.3)
-
•
Does AdapTable demonstrate strengths in computational efficiency and hyperparameter sensitivity, which are crucial for test time adaptation? (Section 4.4)
4.1 Experimental Setup
Datasets.
We evaluate AdapTable on six diverse datasets—HELOC, Voting, Hospital Readmission, ICU Mortality, Childhood Lead, and Diabetes—within the tabular distribution shift benchmark (Gardner, Popovic, and Schmidt 2023), covering healthcare, finance, and politics with both numerical and categorical features. Additionally, we verify its robustness against six common corruptions—Gaussian, Uniform, Random Drop, Column Drop, Numerical, and Categorical—to ensure its efficacy beyond label distribution shifts. More details of these shifts are in Section C.
Model architectures.
To verify the proposed method under various tabular model architectures, we mainly use MLP, a widely used tabular learning architecture. Additionally, we validate AdapTable on CatBoost (Dorogush, Ershov, and Gulin 2017) and three other representative deep tabular learning models—AutoInt (Song et al. 2019), ResNet (Gorishniy et al. 2021), and FT-Transformer (Gorishniy et al. 2021).
Baselines.
We compare AdapTable with six TTA baselines—PL (Lee 2013), TTT++ (Liu et al. 2021), TENT (Wang et al. 2021a), EATA (Niu et al. 2022), SAR (Niu et al. 2023), and LAME (Boudiaf et al. 2022). TabLog (Ren et al. 2024) is excluded due to its architectural constraint on logical neural networks (Riegel et al. 2020). We also provide performance references from classical machine learning models: -nearest neighbors (-NN), logistic regression (LogReg), random forest (RandomForest), XGBoost (Chen and Guestrin 2016), and CatBoost (Dorogush, Ershov, and Gulin 2017).
Evaluation metrics.
Implementation details.
For all experiments, we use a fixed batch size of 64, a common setting in TTA baselines (Schneider et al. 2020; Wang et al. 2021a). The smoothing factor , low uncertainty quantile , and high uncertainty quantile are set to 0.1, 0.25, and 0.75, respectively. In all tables, the best and second-best results are highlighted in bold and underline, respectively. Further implementation and hyperparameter details are provided in Section E.
4.2 Main Results
Result on natural distribution shifts.
Table 2 presents results on natural distribution shifts. Existing TTA methods, successful in computer vision, struggle in the tabular domain, often failing to outperform the source model or offering limited performance gains. In contrast, AdapTable achieves state-of-the-art results across all datasets, with dramatic performance improvements of up to 26% on the HELOC dataset. Since AdapTable does not rely on model parameter tuning, it can be easily applied to classical machine learning models; when integrated with CatBoost, AdapTable consistently improves performance across all datasets, showcasing its versatility, whereas other baselines cannot be similarly integrated as they require model parameter updates.
Result on common corruptions.
We further evaluate the efficacy of AdapTable across six types of common corruptions in real-world applications by applying them to the test sets of three datasets—HELOC, Voting, and Childhood Lead. As shown in Figure 4, prior TTA methods fail considerably, showing only marginal gains over the unadapted source model across all corruption types. It is worth noting that previous TTA methods have demonstrated significant improvements when dealing with common corruptions in vision data, highlighting the difference between corruptions in the tabular domain its counterpart in vision domain. Meanwhile, Adaptable shows substantial improvements across all types of corruptions, showing more than 10% gains of accuracy on all scenarios, demonstrating its robustness across different types of corruptions.
Result across diverse model architectures.
In Figure 5, we report AdapTable’s effectiveness across three mainstream tabular learning architectures—AutoInt (Song et al. 2019), ResNet (Gorishniy et al. 2021), and FT-Transformer (Gorishniy et al. 2021). We report the average macro F1 score across three datasets—HELOC, Voting, and Childhood Lead. None of the baselines outperform the original source model, with LAME (Boudiaf et al. 2022) even showing significant performance drops. In contrast, AdapTable consistently achieves significant improvements across all architectures, highlighting its robustness and versatility.
4.3 Ablation Study
Shift-aware uncertainty calibrator.
First, we validate the shift-aware uncertainty calibrator described in Section 3.2. Figure 6 (a) and (b) present reliability diagrams before and after applying calibration. The results demonstrate that our calibrator significantly reduces both overconfidence and underconfidence. Next, we assess the shift-awareness of our calibrator in Figure 6 (c), where we plot the average temperature against the maximum mean discrepancy (MMD) with the training data. As expected, greater shifts from the training data correspond to higher temperatures, indicating increased uncertainty. The strong positive correlation between MMD and average temperature confirms the calibrator’s effectiveness in capturing prediction uncertainty under distribution shifts in tabular data. Finally, we compare our shift-aware uncertainty calibrator with classical methods, namely Platt scaling and isotonic regression, as shown in Table 3. The results indicate that prior methods exhibit inconsistent performance across different datasets. For example, Platt scaling improves performance on the HELOC dataset but degrades it on Voting and Hospital Readmissions. In contrast, our shift-aware uncertainty calibrator consistently outperforms these classical methods, which do not account for domain shift during calibration.
Label distribution handler.
In the following section, we validate the workings and efficacy of the label distribution handler. First, Figure 7 compares the Jensen–Shannon (JS) divergence between the true and estimated label distributions across each online batch. The results show that our handler significantly improves the accuracy of label distribution estimation, showing low JS divergence across all datasets. Next, we assess the robustness of our label distribution handler with respect to test batch distribution. Namely, we evaluate under 1) a severe class imbalance scenario with an imbalance ratio of 10, and 2) class-wise temporal correlation scenario, where class labels exhibit strong temporal locality. Results are shown in Table 4. AdapTable shows performance gains in all scenarios, showing up to 27% and 19% improvements in class imbalance and temporal correlation scenarios in HELOC and Childhood Lead, respectively. More experimental details are provided in Section C.
4.4 Further Analysis
Computational efficiency.
The leftmost part of Figure 8 compares the computational efficiency of AdapTable with TTA baselines. On the HELOC dataset, AdapTable’s total elapsed time is approximately 1.54 seconds, translating to about 0.0002 seconds per sample, which is highly desirable. Moreover, AdapTable achieves an optimal efficiency-efficacy trade-off.
Hyperparameter sensitivity.
Figure 8 further analyzes the hyperparameter sensitivity of AdapTable on the Childhood Lead dataset. As shown in the figure, AdapTable remains highly insensitive to changes in the smoothing factor , low uncertainty quantile , and high uncertainty quantile .
5 Related Work
Machine learning for tabular data.
The distinct nature of tabular data reduces the effectiveness of deep neural networks, making gradient-boosted decision trees (Chen and Guestrin 2016; Dorogush, Ershov, and Gulin 2017) more suitable. However, research continues to develop deep learning models tailored for tabular data (Murtagh 1991; Song et al. 2019; Arik and Pfister 2021; Gorishniy et al. 2021), including recent efforts involving large language models (Fang et al. 2024; Hegselmann et al. 2023; Hollmann et al. 2023; Dinh et al. 2022) that leverage textual prior knowledge. Notably, our method is architecture-agnostic and can be applied to any model.
Distribution shifts in the tabular domain.
Recently, distribution shift benchmarks for tabular data have been introduced (Liu et al. 2023; Gardner, Popovic, and Schmidt 2023). WhyShift (Liu et al. 2023) reveals that concept shifts (-shifts) are more prevalent and detrimental than covariate shifts (-shifts). TableShift (Gardner, Popovic, and Schmidt 2023) offers a benchmark with 15 classification tasks, highlighting a strong correlation between shift gaps and label distribution shifts (-shifts), which supports the validity of our method.
Test-time adaptation.
Over the past years, test-time adaptation (TTA) methods have been proposed across various domains, such as computer vision (Wang et al. 2021a; Gong et al. 2022; Niu et al. 2023; Shim, Kim, and Yang 2024), natural language processing (Shi et al. 2024; Liang, He, and Tan 2024), and speech processing (Kim et al. 2023). These methods adapt pre-trained models to unlabeled target domains without requiring access to source data, making them well-suited for sensitive tabular data. TabLog (Ren et al. 2024) is a recent TTA method specifically for tabular data, but it has architectural constraints and lacks a comprehensive analysis of distribution shifts. This underscores the need for model-agnostic TTA methods with a deeper understanding of tabular data, which we address in this paper.
6 Conclusion
In this paper, we have introduced AdapTable, a test-time adaptation framework tailored for tabular data. AdapTable overcomes the limitations of previous methods, which fail to address label distribution shifts, and lack versatility across architectures. Our approach, combined with a shift-aware uncertainty calibrator that enhances calibration via modeling column shifts, and a label distribution handler that adjusts the output distribution based on real-time estimates of the current batch’s label distribution. Extensive experiments show that AdapTable achieves state-of-the-art performance across various datasets and architectures, effectively managing both natural distribution shifts and common corruptions.
References
- Arik and Pfister (2021) Arik, S. Ö.; and Pfister, T. 2021. TabNet: Attentive interpretable tabular learning. In AAAI Conference on Artificial Intelligence (AAAI).
- Association (2018) Association, A. D. 2018. Economic costs of diabetes in the US in 2017. Diabetes care.
- Berthelot et al. (2020) Berthelot, D.; Carlini, N.; Cubuk, E. D.; Kurakin, A.; Sohn, K.; Zhang, H.; and Raffel, C. 2020. Remixmatch: Semi-supervised learning with distribution alignment and augmentation anchoring. In International Conference on Learning Representations (ICLR).
- Beyazit et al. (2024) Beyazit, E.; Kozaczuk, J.; Li, B.; Wallace, V.; and Fadlallah, B. 2024. An inductive bias for tabular deep learning. Conference in Neural Information Processing Systems (NeurIPS).
- Bischl et al. (2021) Bischl, B.; Casalicchio, G.; Feurer, M.; Hutter, F.; Lang, M.; Mantovani, R. G.; van Rijn, J. N.; and Vanschoren, J. 2021. OpenML Benchmarking Suites. In Conference on Neural Information Processing Systems (NeurIPS).
- Boudiaf et al. (2022) Boudiaf, M.; Mueller, R.; Ayed, I. B.; and Bertinetto, L. 2022. Parameter-free Online Test-time Adaptation. In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).
- Brown et al. (2018) Brown, K.; Doran, D.; Kramer, R.; and Reynolds, B. 2018. HELOC Applicant Risk Performance Evaluation by Topological Hierarchical Decomposition. In NeurIPS Workshop on Challenges and Opportunities for AI in Financial Services.
- Chen and Guestrin (2016) Chen, T.; and Guestrin, C. 2016. XGBoost: A Scalable Tree Boosting System. In ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD).
- Clore et al. (2014) Clore, J.; Cios, K.; DeShazo, J.; and Strack, B. 2014. Diabetes 130-US hospitals for years 1999-2008. UCI Machine Learning Repository.
- DeFilippis and Van Spall (2021) DeFilippis, E. M.; and Van Spall, H. G. 2021. Is it time for sex-specific guidelines for cardiovascular disease? Journal of the American College of Cardiology (JACC).
- Dinh et al. (2022) Dinh, T.; Zeng, Y.; Zhang, R.; Lin, Z.; Gira, M.; Rajput, S.; Sohn, J.-y.; Papailiopoulos, D.; and Lee, K. 2022. Lift: Language-interfaced fine-tuning for non-language machine learning tasks. In Conference on Neural Information Processing Systems (NeurIPS).
- Dorogush, Ershov, and Gulin (2017) Dorogush, A. V.; Ershov, V.; and Gulin, A. 2017. CatBoost: gradient boosting with categorical features support. In NeurIPS Workshop on ML Systems.
- Fang et al. (2024) Fang, X.; Xu, W.; Tan, F. A.; Zhang, J.; Hu, Z.; Qi, Y. J.; Nickleach, S.; Socolinsky, D.; Sengamedu, S.; Faloutsos, C.; et al. 2024. Large language models (LLMs) on tabular data: Prediction, generation, and understanding-a survey. Transactions on Machine Learning Research (TMLR).
- Fey and Lenssen (2019) Fey, M.; and Lenssen, J. E. 2019. Fast graph representation learning with PyTorch Geometric. In ICLR Workshop on Representation Learning on Graphs and Manifolds.
- for Disease Control, Prevention et al. (2003) for Disease Control, C.; Prevention; et al. 2003. National Health and Nutrition Examination Survey (NHANES) Data. NCfHS, editor. NCHS.
- Gandelsman et al. (2022) Gandelsman, Y.; Sun, Y.; Chen, X.; and Efros, A. 2022. Test-time training with masked autoencoders. In Conference on Neural Information Processing Systems (NeurIPS).
- Gardner, Popovic, and Schmidt (2023) Gardner, J.; Popovic, Z.; and Schmidt, L. 2023. Benchmarking Distribution Shift in Tabular Data with TableShift. In Conference on Neural Information Processing Systems (NeurIPS).
- Gong et al. (2022) Gong, T.; Jeong, J.; Kim, T.; Kim, Y.; Shin, J.; and Lee, S.-J. 2022. Note: Robust continual test-time adaptation against temporal correlation. In Conference on Neural Information Processing Systems (NeurIPS).
- Gorishniy et al. (2021) Gorishniy, Y.; Rubachev, I.; Khrulkov, V.; and Babenko, A. 2021. Revisiting Deep Learning Models for Tabular Data. In Conference on Neural Information Processing Systems (NeurIPS).
- Grinsztajn, Oyallon, and Varoquaux (2022) Grinsztajn, L.; Oyallon, E.; and Varoquaux, G. 2022. Why do tree-based models still outperform deep learning on tabular data? arXiv 2022. In Conference on Neural Information Processing Systems (NeurIPS).
- He et al. (2016) He, K.; Zhang, X.; Ren, S.; and Sun, J. 2016. Deep residual learning for image recognition. In IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).
- Hegselmann et al. (2023) Hegselmann, S.; Buendia, A.; Lang, H.; Agrawal, M.; Jiang, X.; and Sontag, D. 2023. Tabllm: Few-shot classification of tabular data with large language models. In International Conference on Artificial Intelligence and Statistics (AISTATS).
- Hein et al. (2017) Hein, D.; Depeweg, S.; Tokic, M.; Udluft, S.; Hentschel, A.; Runkler, T. A.; and Sterzing, V. 2017. A benchmark environment motivated by industrial control problems. In IEEE Symposium Series on Computational Intelligence (SSCI).
- Hollmann et al. (2023) Hollmann, N.; Müller, S.; Eggensperger, K.; and Hutter, F. 2023. TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Second. In International Conference on Learning Representations (ICLR).
- Hwang et al. (2022) Hwang, S.; Lee, S.; Kim, S.; Ok, J.; and Kwak, S. 2022. Combating label distribution shift for active domain adaptation. In European Conference on Computer Vision (ECCV).
- Johnson et al. (2021) Johnson, A.; Bulgarelli, L.; Pollard, T.; Horng, S.; Celi, L. A.; and Mark, R. 2021. MIMIC-IV.
- Johnson et al. (2016) Johnson, A. E.; Pollard, T. J.; Shen, L.; Lehman, L.-w. H.; Feng, M.; Ghassemi, M.; Moody, B.; Szolovits, P.; Anthony Celi, L.; and Mark, R. G. 2016. MIMIC-III, a freely accessible critical care database. Scientific data.
- Kim et al. (2023) Kim, C.; Park, J.; Shim, H.; and Yang, E. 2023. SGEM: Test-Time Adaptation for Automatic Speech Recognition via Sequential-Level Generalized Entropy Minimization. In Conference of the International Speech Communication Association (INTERSPEECH).
- Lee (2013) Lee, D.-H. 2013. Pseudo-Label: The Simple and Efficient Semi-Supervised Learning Method for Deep Neural Networks. In ICML Workshop on Challenges in Representation Learning.
- Liang, He, and Tan (2024) Liang, J.; He, R.; and Tan, T. 2024. A comprehensive survey on test-time adaptation under distribution shifts. International Journal of Computer Vision (IJCV).
- Lin et al. (2017) Lin, T.-Y.; Goyal, P.; Girshick, R.; He, K.; and Dollár, P. 2017. Focal loss for dense object detection. In IEEE/CVF International Conference on Computer Vision (ICCV).
- Liu et al. (2023) Liu, J.; Wang, T.; Cui, P.; and Namkoong, H. 2023. On the Need for a Language Describing Distribution Shifts: Illustrations on Tabular Datasets. In Conference on Neural Information Processing Systems (NeurIPS).
- Liu et al. (2021) Liu, Y.; Kothari, P.; Van Delft, B.; Bellot-Gurlet, B.; Mordan, T.; and Alahi, A. 2021. TTT++: When does self-supervised test-time training fail or thrive? In Conference on Neural Information Processing Systems (NeurIPS).
- Menon et al. (2021) Menon, A. K.; Jayasumana, S.; Rawat, A. S.; Jain, H.; Veit, A.; and Kumar, S. 2021. Long-tail learning via logit adjustment. In International Conference on Learning Representations (ICLR).
- Mosca, Barrett-Connor, and Kass Wenger (2011) Mosca, L.; Barrett-Connor, E.; and Kass Wenger, N. 2011. Sex/gender differences in cardiovascular disease prevention: what a difference a decade makes. Circulation.
- Murtagh (1991) Murtagh, F. 1991. Multilayer perceptrons for classification and regression. Neurocomputing.
- Niu et al. (2022) Niu, S.; Wu, J.; Zhang, Y.; Chen, Y.; Zheng, S.; Zhao, P.; and Tan, M. 2022. Efficient test-time model adaptation without forgetting. In International Conference on Machine Learning (ICML).
- Niu et al. (2023) Niu, S.; Wu, J.; Zhang, Y.; Wen, Z.; Chen, Y.; Zhao, P.; and Tan, M. 2023. Towards stable test-time adaptation in dynamic wild world. In International Conference on Learning Representations (ICLR).
- Park, Seo, and Yang (2023) Park, J.; Seo, H.; and Yang, E. 2023. Pc-adapter: Topology-aware adapter for efficient domain adaption on point clouds with rectified pseudo-label. In IEEE/CVF International Conference on Computer Vision (ICCV).
- Paszke et al. (2019) Paszke, A.; Gross, S.; Massa, F.; Lerer, A.; Bradbury, J.; Chanan, G.; Killeen, T.; Lin, Z.; Gimelshein, N.; Antiga, L.; et al. 2019. Pytorch: An imperative style, high-performance deep learning library. In Conference on neural information processing systems (NeurIPS).
- Platt (2000) Platt, J. 2000. Probabilistic Outputs for Support Vector Machines and Comparisons to Regularized Likelihood Methods. In Advances in Large Margin Classifiers.
- Ren et al. (2024) Ren, W.; Li, X.; Chen, H.; Rakesh, V.; Wang, Z.; Das, M.; and Honavar, V. G. 2024. TabLog: Test-Time Adaptation for Tabular Data Using Logic Rules. In International Conference on Machine Learning (ICML).
- Riegel et al. (2020) Riegel, R.; Gray, A.; Luus, F.; Khan, N.; Makondo, N.; Akhalwaya, I. Y.; Qian, H.; Fagin, R.; Barahona, F.; Sharma, U.; et al. 2020. Logical neural networks. arXiv preprint arXiv:2006.13155.
- Schneider et al. (2020) Schneider, S.; Rusak, E.; Eck, L.; Bringmann, O.; Brendel, W.; and Bethge, M. 2020. Improving robustness against common corruptions by covariate shift adaptation. In Conference on Neural Information Processing Systems (NeurIPS).
- Shi et al. (2024) Shi, W.; Xu, R.; Zhuang, Y.; Yu, Y.; Wu, H.; Yang, C.; and Wang, M. D. 2024. MedAdapter: Efficient Test-Time Adaptation of Large Language Models towards Medical Reasoning. arXiv preprint arXiv:2405.03000.
- Shim, Kim, and Yang (2024) Shim, H.; Kim, C.; and Yang, E. 2024. CloudFixer: Test-Time Adaptation for 3D Point Clouds via Diffusion-Guided Geometric Transformation. In European Conference on Computer Vision (ECCV).
- Song et al. (2019) Song, W.; Shi, C.; Xiao, Z.; Duan, Z.; Xu, Y.; Zhang, M.; and Tang, J. 2019. Autoint: Automatic feature interaction learning via self-attentive neural networks. In ACM International Conference on Information and Knowledge Management (CIKM).
- Studies (2019) Studies, A. N. E. 2019. FICO. The Explainable Machine Learning Challenge.
- Studies (2022) Studies, A. N. E. 2022. ANES Time Series Cumulative Data File [dataset and documentation]. September 16, 2022 version.
- Stylianou and Flournoy (2002) Stylianou, M.; and Flournoy, N. 2002. Dose Finding Using the Biased Coin Up-and-Down Design and Isotonic Regression. In Biometrics.
- Sun et al. (2020) Sun, Y.; Wang, X.; Liu, Z.; Miller, J.; Efros, A.; and Hardt, M. 2020. Test-time training with self-supervision for generalization under distribution shifts. In International Conference on Machine Learning (ICML).
- Tachet des Combes et al. (2020) Tachet des Combes, R.; Zhao, H.; Wang, Y.-X.; and Gordon, G. J. 2020. Domain adaptation with conditional distribution matching and generalized label shift. In Conference on Neural Information Processing Systems (NeurIPS).
- Wang et al. (2021a) Wang, D.; Shelhamer, E.; Liu, S.; Olshausen, B.; and Darrell, T. 2021a. Tent: Fully test-time adaptation by entropy minimization. In International Conference on Learning Representations (ICLR).
- Wang et al. (2021b) Wang, X.; Liu, H.; Shi, C.; and Yang, C. 2021b. Be confident! towards trustworthy graph neural networks via confidence calibration. In Conference on Neural Information Processing Systems (NeurIPS).
- Wu et al. (2021) Wu, R.; Guo, C.; Su, Y.; and Weinberger, K. Q. 2021. Online adaptation to label distribution shift. In Conference on Neural Information Processing Systems (NeurIPS).
- Zhou et al. (2023) Zhou, Z.; Guo, L.-Z.; Jia, L.-H.; Zhang, D.; and Li, Y.-F. 2023. ODS: Test-Time Adaptation in the Presence of Open-World Data Shift. In International Conference on Machine Learning (ICML).
Appendix
Appendix A Detailed Algorithm of AdapTable
Post-training shift-aware uncertainty calibrator.
Given a pre-trained tabular classifier on the source domain , we introduce a post-training phase for a shift-aware uncertainty calibrator . This calibrator is trained after the initial training of using the same training dataset . For a given training batch , we compute the shift trend for a specific column index as follows:
where we add a linear layer to for categorical column to transform it into a one-dimensional representation, ensuring alignment with the numerical columns. Using , we construct a shift trend graph, where each node represents a column, and edges capture the relationships between columns. The node features are given by , and the graph is connected using an all-ones adjacency matrix. A graph neural network (GNN) is applied to this graph, facilitating the exchange of shift trends between columns through message passing, which generates a contextualized column-wise representation . These representations are averaged to form a global feature representation , which is then concatenated with the initial model prediction to produce the final output temperature . With the calibrated probability , with the per-sample temperature calculated above, we define the most plausible and second plausible class indices and as follows:
The focal loss (Lin et al. 2017) and the calibration loss (Wang et al. 2021b) are used to train the shift-aware uncertainty calibrator , defined as:
(10) | ||||
(11) |
where is an indicator function:
addresses class imbalance by reducing the impact of easily classified examples, while penalizes the gap between and for correct predictions, encouraging them to converge for incorrect predictions. For all experiments, we set and .
Label distribution handler.
During the test phase after post-training , we introduce a label distribution handler using an estimator , defined as:
where represents the calibrated prediction. This approach enhances uncertainty quantification and combines the calibrated estimation with the distributionally aligned prediction for more robust estimation. To compute , we perform a two-stage uncertainty calibration. Specifically, for a given test batch , we calculate the shift trend as:
Then, a per-sample temperature , which was defined in Equation 1 is computed. The uncertainty of is defined as the reciprocal of the margin of the calibrated probability distribution :
where and are the most plausible and second plausible class indices:
Based on , the recalibrated temperature is applied:
where , and and are the low and high uncertainty quantiles, respectively. The target label distribution is then estimated using the following formula:
where serves as a debiased target label estimator, deviating from the source label distribution . The online target label estimator is initialized with a uniform distribution and updated with each new batch as follows:
where is a smoothing factor. This update process leverages information from the current batch to refine the target label distribution estimation over time. The overall procedure of the proposed AdapTable method is summarized in Algorithm 1.
Appendix B Proof of Theorem 3.1
Let’s first define the balanced source error on the source dataset and the conditional error gap between and as follows:
(12) | ||||
(13) |
Definition B.1.
(Generalized Label Shift in Tachet des Combes et al. (2020)). Both input covariate distribution and output label distribution change. Yet, there exists a hidden representation such that the conditional distribution of given remains the same across both domains, i.e., ,
(14) |
Theorem B.2.
Let and be defined as follows:
(15) | ||||
(16) |
Given the error , with true labels of inputs , the error gap is upper bounded by
(17) |
where and are constants related to , and , respectively.
Proof.
We start by applying the law of total probability and triangle inequality to derive the following inequality:
(18) |
According to Equation 8 in (Menon et al. 2021), satisfies the following condition under generalized label shift condition in Definition B.1:
(19) |
By multiplying both sides of Equation 19 by , we obtain:
(20) |
Next, by substituting Equation 20 into Equation 18, and letting , we have:
(21) |
Using Lemma A.2 from (Tachet des Combes et al. 2020), we can further estimate the upper bound of Equation 21 as follows:
(22) |
where and , holds by Hölder’s inequality. By letting and for all , and defining and as:
we finally get:
(23) |
where holds by Hölder’s inequality, holds by the definition of , and holds by the definition of . ∎
We observe that in practice, using can result in performance degradation due to an error accumulation in . However, our approach, which integrates a two-stage uncertainty calibration with and a debiased target label estimator , demonstrates empirical efficacy across various experiments.
Appendix C Dataset Descriptions
C.1 Natural Distibution Shifts
In our experiments, we verify our method across six different datasets—HELOC, Voting, Hospital Readmission, ICU Mortality, Childhood Lead, and Diabetes—within the Tableshift Benchmark (Gardner, Popovic, and Schmidt 2023), all of which include natural distribution shifts between training and test data. For all datasets, the numerical features are normalized—subtraction of mean and division by standard deviation, while categorical features are one-hot encoded. We find that different encoding types do not play a significant role in terms of accuracy, as noted in Grinsztajn, Oyallon, and Varoquaux (2022). Detailed statistics specifications of each dataset are listed in Table 5.
-
•
HELOC: This task predicts Home Equity Line of Credit (HELOC) (Brown et al. 2018) repayment using FICO data (Studies 2019), focusing on shifts in third-party risk estimates. The dataset includes 10,459 observations, and a distribution shift occurs by using the ’External Risk Estimate’ as a domain split. Estimates above 63 are used for training, while those 63 or below are held out for testing, illustrating potential biases in credit assessments.
-
•
Voting: Using ANES (Studies 2022) data, this task predicts U.S. presidential election voting behavior with 8,280 observations. Distribution shift is introduced by splitting the data based on geographic region, with the southern U.S. serving as the out-of-domain region. This simulates how voter behavior predictions might vary when polling data is collected in one region and used to predict outcomes in another.
-
•
Hospital Readmission: Hospital Readmission (Clore et al. 2014) predicts 30-day readmission of diabetic patients using data from 130 U.S. hospitals over 10 years. The distribution shift occurs by splitting the data based on admission source, with emergency room admissions held out as the target domain. This tests how well models trained on other sources perform when applied to patients admitted through the emergency room.
-
•
ICU Mortality: The task predicts ICU patient mortality using MIMIC-iii data (Johnson et al. 2016), focusing on shifts related to insurance type. The dataset includes 23,944 observations, and a distribution shift is created by excluding Medicare and Medicaid patients from the training set, designating them as the target domain. This highlights how insurance type can affect mortality predictions.
-
•
Childhood Lead: This task predicts elevated blood lead levels in children using NHANES data (for Disease Control, Prevention et al. 2003), with 27,499 observations. A distribution shift is introduced by splitting the data based on poverty using the poverty-income ratio (PIR) as a threshold. Those with a PIR of 1.3 or lower are held out for testing, simulating risk assessment in lower-income households.
-
•
Diabetes: This task predicts diabetes using BRFSS data (Association 2018), focusing on racial shifts across 1.4 million observations. Distribution shift occurs by focusing on the differences in diabetes risk between racial and ethnic groups, particularly highlighting the higher risk faced by non-white groups compared to White non-Hispanic individuals.
Statistic | HELOC | Voting | Hospital Readmission | ICU Mortality | Childhood Lead | Diabetes |
Total Instances | 9,412 | 60,376 | 89,542 | 21,549 | 24,749 | 1,299,758 |
Training Set | 2,220 | 34,796 | 34,288 | 7,116 | 11,807 | 969,229 |
Validation Set | 278 | 4,349 | 4,286 | 889 | 1,476 | 121,154 |
Test Set | 6,914 | 21,231 | 50,968 | 13,544 | 11,466 | 209,375 |
Total Features | 22 | 54 | 46 | 7491 | 7 | 25 |
Numerical Features | 20 | 8 | 12 | 7490 | 4 | 6 |
Categorical Features | 2 | 46 | 34 | 1 | 3 | 19 |
C.2 Common Corruptions
Let be the -th row of a table with columns in the test data. We define as a random variable that follows the empirical marginal distribution of the -th column in the training set , given by:
where . Additionally, let and be the mean and standard deviation of the random variable , respectively. To effectively simulate natural distribution shifts that commonly occur beyond label distribution shifts, we introduce six types of corruptions—Gaussian noise (Gaussian), uniform noise (Uniform), random missing values (Random Drop), common column missing across all test data (Column Drop), important numerical column shift (Numerical), and important categorical column shift (Categorical)—as follows:
-
•
Gaussian: For , Gaussian noise is independently injected as:
-
•
Uniform: For the , uniform noise is independently injected as:
-
•
Random Drop: For each column , a random mask is applied, and the feature is replaced by a random sample drawn from the empirical marginal distribution of the -th column of the training set:
-
•
Column Drop: For each column , a random mask is applied, and the feature is replaced by a random sample as follows:
Unlike random drop corruption, where the mask is resampled for each -th column of the -th test instance , a single random mask is sampled for each -th column and applied uniformly across all test data.
-
•
Numerical: Important numerical column shift simulates natural domain shifts where the test distribution of the most important numerical column deviates significantly from the training distribution. We first identify the most important numerical column, , using a pre-trained XGBoost (Chen and Guestrin 2016). A Gaussian distribution
is then fitted to the -th column of the training data, using and . The likelihood of each test sample is then computed as . Finally, test samples are drawn inversely proportional to their likelihood, with the sampling probability of is defined as:
-
•
Categorical: Important categorical column shift simulates natural domain shifts where the test distribution of the most important categorical column deviates significantly from the training distribution. Again, we first identify the most important categorical column, , using a pre-trained XGBoost (Chen and Guestrin 2016). A categorical distribution, which generalizes the Bernoulli distribution,
is then fitted to the -th column of the training data, where is the number of distinct categorical features in the -th column, and for . The likelihood of each test sample is then computed as . Finally, test samples are drawn inversely proportional to their likelihood, with the sampling probability of is defined as:
C.3 Label Distribution Shifts
-
•
Class Imbalance: This label distribution shift simulates a highly class-imbalanced test stream, where labels that are rare in the training set are more likely to appear frequently in the test set. Given a class imbalance ratio , we first rank the output labels for each test sample in ascending order of their frequency in the training set, assigning ranks from 1 to , where is the number of classes. Specifically, indicates that is the least frequent label in the training set, while indicates that is the most frequent. We then define the unnormalized sampling probability for each test sample as:
The normalized sampling probability for each test sample is then defined as:
-
•
Temporal Correlation: To simulate temporal correlations in test data, we employ a custom sampling strategy using the Dirichlet distribution. This approach effectively captures temporal dependencies by dynamically adjusting the label distribution over time. We begin with a uniform probability distribution , where is the number of classes. For sampling the -th test instance, a probability distribution is drawn from the Dirichlet distribution:
and then smoothed using to avoid zero probabilities for any class :
A label is subsequently sampled according to , and the corresponding test instance is randomly selected from the test data with label . After the -th sampling, the distribution is updated using the recent history of sampled labels within a sliding window of size :
Appendix D Baseline Details
D.1 Deep Tabular Learning Architectures
-
•
MLP: Multi-Layer Perceptron (MLP) (Murtagh 1991) is a foundational deep learning architecture characterized by multiple layers of interconnected nodes, where each node applies a non-linear activation function to a weighted sum of its inputs. In the tabular domain, MLP is often employed as a default deep learning model, with each input feature corresponding to a node in the input layer.
-
•
AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networks (AutoInt) (Song et al. 2019) is a model that automatically learns complex feature interactions in tasks like click-through rate (CTR) prediction, where features are typically sparse and high-dimensional. It uses a multi-head self-attentive neural network to map features into a low-dimensional space and capture high-order combinations, eliminating the need for manual feature engineering. AutoInt efficiently handles large datasets, outperforms existing methods, and provides good explainability.
-
•
ResNet: ResNet for tabular data (Gorishniy et al. 2021), is a modified version of the original ResNet architecture (He et al. 2016), tailored to capture intricate patterns within structured datasets. Although earlier efforts yielded modest results, recent studies have re-explored ResNet’s capabilities, inspired by its success in computer vision and NLP. This ResNet-like model for tabular data is characterized by a streamlined design that facilitates optimization through nearly direct paths from input to output, enabling the effective learning of deeper feature representations.
-
•
FT-Transformer: Feature Tokenizer along with Transformer (FT-Transformer) (Gorishniy et al. 2021), represents a straightforward modification of the Transformer architecture tailored for tabular data. In this model, the feature tokenizer component plays a crucial role by converting all features, whether categorical or numerical, into tokens. Subsequently, a series of Transformer layers are applied to these tokens within the Transformer component, along with the added [CLS] token. The ultimate representation of the [CLS] token in the final Transformer layer is then utilized for the prediction.
D.2 Supervised Baselines
-
•
-NN: -Nearest Neighbors (-NN) is a fundamental model in tabular learning that identifies the closest data points based on a chosen metric. It makes predictions through majority voting for classification or weighted averaging for regression. The hyperparameter influences the model’s sensitivity.
-
•
LogReg: Logistic Regression (LogReg) is a linear classification model that estimates the probability of class membership using a logistic function, which maps the linear combination of features to a range of . With proper regularization, LogReg can achieve performance comparable to state-of-the-art tabular models.
-
•
RandomForest: Random Forest is an ensemble learning algorithm that builds multiple decision trees to improve accuracy and reduce overfitting. It is particularly effective at capturing non-linear patterns and is robust against outliers.
-
•
XGBoost: Extreme Gradient Boosting (XGBoost) (Chen and Guestrin 2016) is a boosting algorithm that sequentially builds weak learners, typically decision trees, to correct errors made by previous models. XGBoost is known for its high predictive performance and ability to handle complex relationships through regularization.
-
•
CatBoost: CatBoost (Dorogush, Ershov, and Gulin 2017), like XGBoost, is a boosting algorithm that excels in handling categorical features without extensive preprocessing. It is highly effective in real-world datasets, offering strong performance, albeit at the cost of increased computational resources and the need for parameter tuning.
D.3 Test-Time Adaptation Baselines
-
•
PL: Pseudo-Labeling (PL) (Lee 2013) leverages a pseudo-labeling strategy to update model parameters during test time.
-
•
TTT++: Improved Test-Time Training (TTT++) (Liu et al. 2021) enhances test-time adaptation by using feature alignment strategies and regularization, eliminating the need to access source data during adaptation.
-
•
TENT: Test ENTropy minimization (TENT) (Wang et al. 2021a) updates the scale and bias parameters in the batch normalization layer during test time by minimizing entropy within a given test batch.
-
•
EATA: Efficient Anti-forgetting Test-time Adaptation (EATA) (Niu et al. 2022) mitigates the risk of unreliable gradients by filtering out high-entropy samples and applying a Fisher regularizer to constrain key model parameters during adaptation.
-
•
SAR: Sharpness-Aware and Reliable optimization (SAR) (Niu et al. 2023) builds on TENT by filtering samples with large entropy, which can cause model collapse during test time, using a predefined threshold.
-
•
LAME: Laplacian Adjusted Maximum-likelihood Estimation (LAME) (Boudiaf et al. 2022) employs an output adaptation strategy during test-time, focusing on adjusting the model’s output probabilities rather than tuning its parameters.
Appendix E Further Experimental Details
E.1 Further Implementation Details
All experiments are conducted on two servers. The first server is equipped with a 40-core Intel Xeon E5-2630 v4 CPU, 252GB RAM, 4 NVIDIA TITAN Xp GPUs, and runs Ubuntu 18.04.4. The second server has a 40-core Intel Xeon E5-2640 v4 CPU, 128GB RAM, 8 NVIDIA TITAN Xp GPUs, and runs Ubuntu 22.04.4. All architectures were implemented using Python 3.8.16 with PyTorch (Paszke et al. 2019) and PyTorch Geometric (Fey and Lenssen 2019). The specific versions of all software libraries and frameworks used are provided in the AdapTable/requirements.txt file of the supplementary materials. We also include our source code in AdapTable folder of the supplementary materials. Please refer to this for all experimental details and to clarify any uncertainties.
E.2 Hyperparameters for Supervised Baselines
For -NN, LogReg, RandomForest, XGBoost, and CatBoost, optimal parameters are determined for each dataset using a random search with 10 iterations on the validation set. The search space for each method is specified in Table 6.
E.3 Hyperparameters for TTA Baselines
In scenarios where the test set is unknown, tuning the hyperparameters of TTA methods on the test set would be considered cheating. Therefore, we tune all hyperparameters for each TTA method and backbone classifier architecture using the Numerical common corruption on the CMC tabular dataset, which we did not use as test data in OpenML-CC18 (Bischl et al. 2021) benchmark. PL, TENT (Wang et al. 2021a), and SAR (Niu et al. 2023) require three main hyperparameters—learning rate, number of adaptation steps per batch, and the option for episodic adaptation, where the model is reset after each batch. PL (Lee 2013) and TENT use a learning rate of 0.0001 with 1 adaptation step and episodic updates. Additionally, SAR requires a threshold to filter high-entropy samples and is configured with a learning rate of 0.001, 1 adaptation step, and episodic updates. For TTT++ (Liu et al. 2021), EATA (Niu et al. 2022), and LAME (Boudiaf et al. 2022), we follow the authors’ hyperparameter settings, except for the learning rate and adaptation steps. TTT++ and EATA were configured with a learning rate of 0.00001, 10 adaptation steps, and episodic updates. LAME, which only adjusts output logits, does not require hyperparameters related to gradient updates. For all baselines, hyperparameter choices remained consistent across different architectures, including MLP, AutoInt, ResNet, and FT-Transformer. The hyperparameter search space for each method are detailed in Table 7.
Hyperparameter | Search Space |
lr | |
# steps | |
episodic | {True, False} |
E.4 Hyperparameters for AdapTable
AdapTable requires three test-time hyperparameters—the smoothing factor , and the low and high uncertainty quantiles and . For fairness, we tune all AdapTable hyperparameters across different backbone architectures using the Numerical common corruption on the CMC dataset from the OpenML-CC18 benchmark (Bischl et al. 2021), which is not used as test data. We observe that AdapTable’s hyperparameter choices remain consistent across various architectures, including MLP, AutoInt, ResNet, and FT-Transformer. Notably, AdapTable demonstrates high insensitivity to variations in , , and , which are uniformly set to 0.1, 0.25, and 0.75, respectively, across all datasets and architectures.
Appendix F Additional Analysis
F.1 Latent Space Visualizations
In Figure 9, we further visualize latent spaces of test instances using t-SNE across six different datasets and four representative deep tabular learning architectures to illustrate the observation discussed in Section 2.1. This visualization highlights the complex decision boundaries within the latent space of tabular data, which are significantly more intricate than those observed in other domains. By comparing the upper four rows—HELOC, Voting, Hospital Readmission, and Childhood Lead—with the lower two rows—linearized image data (MFEAT-PIXEL) and homogeneous DNA string sequences (DNA)—it becomes evident that the latent space decision boundaries in the tabular domain are particularly complex. According to WhyShift (Liu et al. 2023), this complexity is primarily due to latent confounders inherent in tabular data and concept shifts, where such confounders cause output labels to vary greatly for nearly identical inputs. As discussed in Section 2.1, this further underscores the limitations of existing TTA methods (Sun et al. 2020; Gandelsman et al. 2022; Liu et al. 2021; Boudiaf et al. 2022; Zhou et al. 2023), which often depend on the cluster assumption.
F.2 Reliability Diagrams
Figure 10 presents additional reliability diagrams across five different datasets and four representative deep tabular learning architectures, illustrating that tabular data often displays a mix of overconfident and underconfident prediction patterns. This contrasts with the consistent overconfidence observed in the image domain (Stylianou and Flournoy 2002) and underconfidence in the graph domain (Wang et al. 2021b). As shown in Figure 10, the Voting and Hospital Readmission datasets consistently exhibit overconfident behavior across all architectures, while the HELOC, Childhood Lead, and Diabetes datasets demonstrate underconfident tendencies. These observations underscore the need for a tabular-specific uncertainty calibration method.
F.3 Label Distribution Shifts and Prediction Bias Towards Source Label Distributions
We demonstrate that the data distribution shift we primarily target in the tabular domain—label distribution shift—occurs frequently in practice. Figure 11 presents the source label distribution (a), target label distribution (b), pseudo label distribution for test data using the source model (c), and the estimated target label distribution after applying our label distribution handler (d) across the five datasets. Comparing (a) and (b) in each row, it is evident that label distribution shift occurs across all datasets. In (c), we observe that the marginal label distribution predicted by the source model is commonly biased towards the source label distribution. Lastly, (d) illustrates that our label distribution handler effectively estimates the target label distribution, guiding the pseudo label distribution towards the target label distribution.
F.4 Entropy Distributions
We highlight a unique characteristic of tabular data: model prediction entropy consistently shows a strong bias toward underconfidence. To illustrate this, we present entropy distribution histograms for test instances across six datasets and four representative deep tabular learning architectures in Figure 12. A clear pattern emerges when comparing the upper four rows (HELOC, Voting, Hospital Readmission, Childhood Lead) with the lower two (Optdigits, DNA). The upper rows exhibit consistently high entropy, indicating a skew toward underconfidence, while the lower rows do not, except for Childhood Lead, where extreme class imbalance causes the model to collapse to the major class. This analysis highlights the distinct bias of tabular data toward underconfident predictions, a pattern less common in other domains. This aligns with findings that applying unsupervised objectives like entropy minimization to high-entropy samples can result in gradient explosions and model collapse (Niu et al. 2023).
Appendix G Additional Experiments
G.1 Detailed Results Across Common Corruptions and Datasets
Figure 4 presents the average F1 score across six types of common corruption and three datasets. Here, we provide more detailed results, including the standard errors. As shown in Table 8, AdapTable outperforms baseline TTA methods by a large margin across all datasets and corruption types. This further highlights the empirical efficacy of AdapTable, not only in handling label distribution shifts but also in addressing various common corruptions.
G.2 All Results Across Datasets and Model Architectures
In Figure 5, we demonstrate the effectiveness of AdapTable across various tabular model architectures by reporting the average performance across three datasets. Here, we provide the mean and standard error for each dataset and architecture. As shown in Table 9, AdapTable consistently achieves state-of-the-art performance with significant improvements across all model architectures and datasets. This further underscores the versatility and robustness of AdapTable.
G.3 Additional Computational Efficiency Analysis
One may wonder whether the post-training time required for AdapTable’s shift-aware uncertainty calibrator is prohibitively long. To address this concern, we measure and report the elapsed real time for post-training our shift-aware uncertainty calibrator on the medium-scale Hospital Readmission dataset using the FT-Transformer architecture. The post-training process takes approximately 9.2 seconds. For small- and medium-scale datasets, the post-training process typically requires only a few seconds, and even in our largest experimental setting, the time remains minimal, taking at most a few minutes.
Appendix H Limitations and Broader Impacts
H.1 Limitations
Similar to other test-time training (TTT) methods (Sun et al. 2020; Liu et al. 2021; Gandelsman et al. 2022), AdapTable requires an additional post-training stage to integrate a shift-aware uncertainty calibrator during the source model’s training phase. While full test-time adaptation methods (Wang et al. 2021a; Niu et al. 2022, 2023) avoid this, our analysis in Section 2.1 and experiments in Section 4 show that they fail in the tabular domain due to their focus on input covariate shifts, which are often entangled with concept shifts. According to WhyShift (Liu et al. 2023), concept shifts, driven by changes in latent confounders, require natural language descriptions of the shift conditions, necessitating a data-centric approach. Additionally, while AdapTable performs well across various corruptions beyond label distribution shifts (Figure 4), it is primarily focused on addressing label distribution shifts. Further exploration is needed to assess its effectiveness in handling input covariate shifts or concept shifts.
H.2 Broader Impacts
Tabular data is prevalent across industries such as healthcare (Johnson et al. 2016, 2021), finance (Studies 2019, 2022), manufacturing (Hein et al. 2017), and public administration (Gardner, Popovic, and Schmidt 2023). Our research addresses the critical yet underexplored challenge of distribution shifts in tabular data, a problem that has not received sufficient attention. We believe that our approach can significantly enhance the performance of machine learning models in various industries by improving model adaptation to tabular data, thereby creating meaningful value in practical applications. Through our data-centric analysis in Section 2, we identify why existing TTA methods fail in the tabular domain and introduce a tabular-specific approach for handling label distribution shifts in Section 3. We hope this work will provide valuable insights for future research on test-time adaptation in tabular data. Additionally, by making our source code publicly available, we aim to support real-world applications across various fields, benefiting both academia and industry.
Method | Gaussian | Uniform | Random Drop | Column Drop | Numerical | Categorical | |
HELOC | Source | 33.1 0.0 | 33.0 0.0 | 31.4 0.1 | 32.3 1.4 | 33.8 0.2 | 32.3 0.3 |
PL | 31.2 0.0 | 31.2 0.0 | 30.6 0.0 | 31.1 0.7 | 32.1 0.2 | 30.4 0.2 | |
TENT | 33.1 0.0 | 33.0 0.0 | 31.4 0.1 | 32.3 1.4 | 33.8 0.2 | 32.3 0.3 | |
EATA | 33.1 0.0 | 33.0 0.0 | 31.4 0.1 | 32.3 1.4 | 33.8 0.2 | 32.3 0.3 | |
SAR | 31.9 0.1 | 32.0 0.1 | 30.7 0.2 | 31.3 0.8 | 32.4 0.4 | 31.4 0.3 | |
LAME | 30.1 0.0 | 30.1 0.0 | 30.1 0.0 | 30.1 0.0 | 30.9 0.1 | 29.4 0.2 | |
AdapTable | 57.6 0.1 | 57.8 0.0 | 53.0 0.1 | 52.1 3.2 | 58.1 0.1 | 58.9 0.4 | |
Voting | Source | 76.6 0.0 | 76.5 0.0 | 72.5 0.2 | 72.8 0.4 | 76.3 0.1 | 85.2 0.1 |
PL | 75.6 0.3 | 75.2 0.3 | 71.1 0.5 | 70.6 0.5 | 75.9 0.1 | 85.1 0.1 | |
TENT | 76.6 0.0 | 76.5 0.0 | 72.5 0.2 | 72.8 0.4 | 76.3 0.1 | 85.2 0.1 | |
EATA | 76.6 0.0 | 76.5 0.0 | 72.5 0.2 | 72.8 0.4 | 76.3 0.1 | 85.2 0.1 | |
SAR | 67.2 1.0 | 64.0 0.2 | 61.8 1.0 | 60.8 0.8 | 69.9 0.1 | 84.2 0.1 | |
LAME | 39.4 0.2 | 39.4 0.1 | 37.3 0.0 | 37.8 0.2 | 39.4 0.2 | 81.4 0.2 | |
AdapTable | 78.9 0.0 | 78.6 0.1 | 74.9 0.1 | 75.5 0.5 | 78.0 0.1 | 85.0 0.4 | |
Childhood Lead | Source | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 |
PL | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 | |
TENT | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 | |
EATA | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 | |
SAR | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 | |
LAME | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 47.9 0.0 | 48.1 0.0 | 48.8 0.0 | |
AdapTable | 61.4 0.1 | 61.5 0.0 | 58.0 0.1 | 55.9 1.6 | 62.8 0.2 | 53.1 0.2 |
Method | HELOC | Voting | Childhood Lead | |
AutoInt | Source | 34.9 0.0 | 77.5 0.0 | 47.9 0.0 |
PL | 31.6 0.0 | 76.5 0.1 | 47.9 0.0 | |
TENT | 34.9 0.0 | 77.5 0.0 | 47.9 0.0 | |
EATA | 34.9 0.0 | 77.5 0.0 | 47.9 0.0 | |
SAR | 62.0 0.4 | 31.2 0.7 | 47.9 0.0 | |
LAME | 30.1 0.0 | 37.3 0.0 | 47.9 0.0 | |
AdapTable | 56.3 0.1 | 79.2 0.0 | 61.8 0.1 | |
ResNet | Source | 52.0 0.0 | 76.6 0.0 | 47.9 0.0 |
PL | 34.3 0.1 | 73.3 0.1 | 47.9 0.0 | |
TENT | 52.0 0.0 | 76.6 0.0 | 47.9 0.0 | |
EATA | 52.0 0.0 | 76.7 0.0 | 47.9 0.0 | |
SAR | 55.1 0.5 | 52.2 0.5 | 47.9 0.0 | |
LAME | 30.1 0.0 | 75.1 0.1 | 47.9 0.0 | |
AdapTable | 61.9 0.0 | 78.7 0.0 | 61.3 0.1 | |
FT-Transformer | Source | 33.0 0.0 | 77.3 0.0 | 47.9 0.0 |
PL | 30.6 0.0 | 76.0 0.1 | 47.9 0.0 | |
TENT | 33.0 0.0 | 77.3 0.0 | 47.9 0.0 | |
EATA | 33.0 0.0 | 77.3 0.0 | 47.9 0.0 | |
SAR | 35.3 0.1 | 73.6 0.3 | 47.9 0.0 | |
LAME | 30.7 0.1 | 71.5 0.1 | 47.9 0.0 | |
AdapTable | 55.0 0.0 | 79.2 0.1 | 61.7 0.1 |