Privacy Vulnerabilities of Attention Layers in Tabular Foundation Models and Protection of High-Risk Queries
On the map: LLM safety and security → Membership inference attacks
Abstract (arXiv)
Tabular foundation models are commonly assumed to present limited privacy concerns as they are often pre-trained on large collections of synthetic data. However, these models leverage in-context learning, where sensitive records may be provided directly at inference time as labelled context examples. In this paper, we demonstrate that predictions generated via the attention mechanism leak sufficient information to enable effective Membership Inference Attacks (MIAs). To highlight this vulnerability, we propose AMIA (Attention-based Membership Inference Attack), a shadow-model-free attack that exploits the concentration of transformer attention patterns. Our results show that attention mechanisms reveal strong membership signals, which exceed classical confidence-based attacks, achieving an average gain of 7.7%, specially in low false-positive regimes. To mitigate this risk, we introduce an inference-time defence inspired by k-anonymity principles. This approach reduces the uniqueness of context-key representations without introducing random noise or retraining the model. By targeting only high-risk queries identified through AMIA scores, the defence substantially reduces membership leakage of this attack by an average of 50% and 25% against confidence-based attacks, while preserving predictive utility with only 3.9% performance degradation. Beyond showing that context examples are vulnerable, we further demonstrate that fine-tuning introduces an additional source of privacy risk. In particular, samples whose prediction confidence increases after fine-tuning become more susceptible to MIAs, indicating that fine-tuning can amplify memorisation and expose sensitive training information through confidence shifts.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Membership inference attacks, one of 12 directions in LLM safety and security. Builds on 35 of its 50 references that are in the map. First posted on arXiv 24 Jun 2026.
Builds on
35 of 50 references.
Closest ancestors · 11
Same direction, newest first
- 2026AttenMIA: LLM Membership Inference Attack through Attention Signals26 Jan
- 2026SoK: Challenges in Tabular Membership Inference Attacks22 Jan
- 2025LoRA-Leak: Membership Inference Attacks Against LoRA Fine-tuned Language Models24 Jul
- 2025Tab-MIA: A Benchmark Dataset for Membership Inference Attacks on Tabular Data in LLMs23 Jul
- 2024On the Privacy Risk of In-context Learning15 Nov
- 2024Context-Aware Membership Inference Attacks against Pre-trained Large Language Models11 Sep
- 2024Membership Inference Attacks Against In-Context Learning2 Sep
- 2023Low-Cost High-Power Membership Inference Attacks6 Dec
- 2023Overconfidence is a Dangerous Thing: Mitigating Membership Inference Attacks by Enforcing Less Confident Prediction4 Jul
- 2021Membership Inference Attacks From First Principles7 Dec
- 2021Enhanced Membership Inference Attacks against Machine Learning Models18 Nov
Other references · 24
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 643
- 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 267
- 2025TabArena: A Living Benchmark for Machine Learning on Tabular Datacited by 152
- 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 115
- 2024TabDPT: Scaling Tabular Foundation Models on Real Datacited by 97
- 2023Privacy-Preserving In-Context Learning with Differentially Private Few-Shot Generationcited by 94
- 2023Flocks of Stochastic Parrots: Differentially Private Prompt Learning for Large Language Modelscited by 80
- 2025Mitra: Mixed Synthetic Priors for Enhancing Tabular Foundation Modelscited by 67
12 more
- 2023Privacy-Preserving In-Context Learning for Large Language Modelscited by 50
- 2024Private prediction for large-scale synthetic text generationcited by 24
- 2023Understanding Emergent In-Context Learning from a Kernel Regression Perspectivecited by 23
- 2025TabSTAR: A Tabular Foundation Model for Tabular Data with Text Fieldscited by 15
- 2025TabFlex: Scaling Tabular Learning to Millions with Linear Attentioncited by 13
- 2024DP-TabICL: In-Context Learning with Differentially Private Tabular Datacited by 11
- 2024TabularBench: Benchmarking Adversarial Robustness for Tabular Deep Learning in Real-world Use-casescited by 10
- 2024Locally Differentially Private In-Context Learningcited by 5
- 2024In-Context Learning for Extreme Multi-Label Classificationcited by 4
- 2025ContextLeak: Auditing Leakage in Private In-Context Learning Methodscited by 4
- 2025When Tables Leak: Attacking String Memorization in LLM-Based Tabular Data Generationcited by 3
- 2025Privacy Preserving In-Context-Learning Framework for Large Language Modelscited by 2
Built on by
Not cited yet.