Skip to content
Preprint

Test-Time Augmentation for Tabular-to-Image Classifiers under Distribution Shifts

Aug 2026 · 0 citations · 17 references
Computer Science

TL;DR

Results indicate that TTA improves OOD performance, with composite and photometric strategies providing the best trade-off between robustness and variance, in contrast to frequency-domain transformations that alter the encoder's feature-to-intensity mapping consistently degrade performance.

Abstract

Tabular-to-image methods that convert tabular data into visual representations have emerged as a novel paradigm for leveraging the high performance of deep learning models. Despite their advantages, the robustness of these methods under distribution shifts remains under explored. Test-Time Augmentation (TTA) is an effective approach in image classification to improve model generalization and robustness, where predictions over multiple transformed views of each input are aggregated. This work evaluates the impact of TTA techniques on predictive performance under Out-Of-Distribution (OOD) for representations generated by tabular-to-image methods. Six tabular-to-image encoding methods were considered: TINTO, IGTD, DeepInsight, BIE, DistanceMatrix, Fotomics. Twenty-five TTA techniques were used, organized into six types: Geometric, Photometric, Structural, Frequency/Encoding, Mixup, and Composite. We employed two datasets from the TableShift benchmark (HELOC and Voting) that provide in-distribution and OOD test subsets designed to evaluate the effect of distribution shifts on tabular data. The results indicate that TTA improves OOD performance, with composite and photometric strategies providing the best trade-off between robustness and variance. In contrast, frequency-domain transformations that alter the encoder's feature-to-intensity mapping consistently degrade performance. These findings highlight TTA as a promising approach for improving the robustness and generalization of classifiers trained on image representations derived from tabular data, particularly under distribution shifts.

View source

Similar papers

Review Aug 2026

Image Augmentation as Test Generation for Deep Learning-Based Image Retrieval Systems

Ensuring the reliability of deep learning-based image retrieval systems is a software engineering challenge. This paper presents a dual contribution: (1) a literature review of augmentation and generation techniques which resulted in the identification of 50 techniques which we organized into a ten-category taxonomy, and (2) a large-scale empirical study that evaluates these techniques as test generators for embedding-based image retrieval systems. Augmented images are embedded using Amazon Titan and OpenCLIP, and evaluated across four analytical dimensions: (1) embedding-space similarity, (2) embedding uncertainty measured via four estimators, (3) semantic realism scored by LLaVA, and (4) retrieval failure rate. Experiments are performed on three datasets: CIFAR-10, ImageNet-1K, and a dataset from an industrial partner (March Networks). Across all evaluated datasets and embedding models, and under the single severity level tested for each technique, weather simulation and SaSPA are the image augmentation/generation techniques that produce the highest embedding uncertainty and failure rates while maintaining a favorable balance between performance stability, visual realism, and augmentation effectiveness. The results we discuss are configuration-specific and may shift under milder or stronger perturbation settings. In contrast, GAN-based augmentation techniques are among the lowest in realism, indicating the presence of synthetic artifacts and perceptual inconsistencies that reduce their suitability to produce realistic test inputs. Overall, our findings provide practical guidelines for selecting augmentation techniques that maximize test diversity while preserving realistic image characteristics, thereby enabling the construction of comprehensive and effective test suites for image retrieval systems while reducing the cost of manual data labeling through the use of metamorphic testing.

Yehan De Silva, Anirudh Sridhar, Armin Lotfy et al. · 0 citations
Preprint Aug 2026

Scalable Black-Box Model Attribution for Images

The rapid proliferation of generative models raises the model attribution problem: given only an image, can we determine which model produced it? Existing methods have grown as elaborate as the generators they target, on the as- sumption that a more sophisticated model demands a more sophisticated attributor. We show it does not. RPA (Raw- Patch Attribution) attributes images in the strictest black- box setting with a lightweight CNN. Despite its simplicity, it attributes more models at higher accuracy than prior work, reaching 98.0% on 25-class DRAGON and 92.9% on 27- class OpenFake; it is data-efficient and runs at a cost inde- pendent of the number of candidate models; and it stays ro- bust to the compression, blur, and resizing images undergo in the wild. Training for closed-set attribution yields a ver- satile feature extractor: the same representation recovers model lineage without supervision, flags and groups unseen generators, and admits new models through few-shot adap- tation rather than retraining.

Asaf Livne, Amir Jevnisek, S. Avidan · 0 citations

Rethinking BCE Loss for Multi-Label Image Recognition with Fine-Tuning

Class-wise Covariance Regularization is proposed, which aligns the predicted covariance structure of class confidences with the semantic correlations encoded in pretrained text embed-dings with the geometric consistency of the class space throughout fine-tuning, resulting in more stable and interpretable confidence distributions across categories.

Ao Zhou, Zhiwei Jiang, Zifeng Cheng et al. · 0 citations
Open access Aug 2026

Dataset Distillation by Tabular Alignment via Moment Embeddings

Dataset distillation has achieved strong results in computer vision, but is largely underexplored in the tabular domain. We introduce a tabular dataset distillation method that projects the data through many random embedders (views) to achieve invariance to transformation and to focus on the consistency between real and synthetic sets. The synthetic set is determined through a formulation of distribution matching between the many-view projection of the original and distilled dataset. Building on this approach, our proposal achieves three goals: (1) we formulate the Tabular Alignment via Moment Embeddings (TAME) method and, by extensive empirical evaluation, we show its efficiency; (2) we evaluate TAME on a benchmark of 18 tabular datasets, with strong baselines and evaluation metrics; and (3) we present a structured set of studies analyzing the impact of losses, dataset geometry, embedder architecture, instances per class (IPC) budget and downstream classifiers. We show that the proposed TAME method consistently surpasses baselines on neural classifiers, while remaining competitive with strong coreset baselines on tree-based classifiers (RF, XGBoost). Performance is further increased, especially for tree-based classifiers with a lightweight validation method. Extensive evaluation, including on additional large-scale sets and ablation experiments, allow a better understanding of the method.

Eduard Barnoviciu, C. Florea · 0 citations
Preprint Aug 2026

OPAL: Orthonormal Prototype Alignment Learning for Interpretable Image Classification

Prototypical part-based models provide explainable predictions by comparing input regions to learned prototypes. However, current approaches are burdened by complex, multi-stage training pipelines and heavily rely on auxiliary regularization to prevent prototype collapse. To overcome these limitations, we introduce Orthonormal Prototype Alignment Learning (OPAL), a single-stage, end-to-end framework that simplifies interpretable classification. Our approach anchors the latent space using predefined orthonormal bases, embedding each class within a dedicated subspace spanned by fixed part-prototypes. To achieve precise part localization, OPAL enforces spatial competition across feature maps. This mechanism isolates sparse, discriminative regions, directing each prototype to consistently attend to the same semantic concept across different images. By framing classification as a direct representation alignment task, our method eliminates the need for auxiliary losses. Extensive experiments on fine-grained benchmarks demonstrate that OPAL outperforms both its non-interpretable counterparts and state-of-the-art part-prototype methods, delivering granular visual explanations by explicitly revealing the specific image regions driving every prediction. Code is available at https://github.com/ilancarretero/OPAL.

I. Carretero, G. Angulo, R. del Amor et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.