On the Robustness of Tabular Foundation Models: Test-Time Attacks and In-Context Defenses

Mohamed Djilani, Thibault Simonetto, Karim Tit, Florian Tambon, Paul Récamier +3 more
6/3/2025
cs.LG

No confident code implementation found for this paper.

Abstract

Recent tabular Foundational Models (FM) such as TabPFN and TabICL, leverage in-context learning to achieve strong performance without gradient updates or fine-tuning. However, their robustness to adversarial manipulation remains largely unexplored. In this work, we present a comprehensive study of the adversarial vulnerabilities of tabular FM, focusing on both their fragility to targeted test-time attacks and their potential misuse as adversarial tools. We show on three benchmarks in finance, cybersecurity and healthcare, that small, structured perturbations to test inputs can significantly degrade prediction accuracy, even when training context remain fixed. Additionally, we demonstrate that tabular FM can be repurposed to generate transferable evasion to conventional models such as random forests and XGBoost, and on a lesser extent to deep tabular models. To improve tabular FM, we formulate the robustification problem as an optimization of the weights (adversarial fine-tuning), or the context (adversarial in-context learning). We introduce an in-context adversarial training strategy that incrementally replaces the context with adversarial perturbed instances, without updating model weights. Our approach improves robustness across multiple tabular benchmarks. Together, these findings position tabular FM as both a target and a source of adversarial threats, highlighting the urgent need for robust training and evaluation practices in this emerging paradigm.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{djilani2025robustness,
  title  = {On the Robustness of Tabular Foundation Models: Test-Time Attacks and In-Context Defenses},
  author = {Mohamed Djilani and Thibault Simonetto and Karim Tit and Florian Tambon and Paul Récamier and Salah Ghamizi and Maxime Cordy and Mike Papadakis},
  year   = {2025},
  eprint = {2506.02978},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2506.02978v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2506.02978.svg)](https://codeofpaper.com/paper/2506.02978)

Discussion