mlsanityData to 14 Oct 2026

Support-Set Target Leakage in Relational Foundation Models during In-Context Learning: Impact, Detection, and Mitigation

arXiv 2609.36384 ↗First posted 28 Sep 2026cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Relational in-context learning (ICL) conditions predictions on the labeled support examples and their linked tables, creating a failure mode when the support set contains target-derived features that are unavailable for the query. We formulate this problem as support-set target leakage, distinct from leakage during dataset construction, temporal splitting, or representation learning. Here, the target-derived (leaker) columns are present only in the labeled support set during relational in-context inference, while queries remain clean. We construct 14 synthetic leaker types, corresponding to 20 columns, spanning proxies with different noise levels, coverage, modalities, semantic transparency, and relational distances. We evaluate a frozen relational encoder with an ICL head on held-out RelBench databases and use Integrated Gradients (IG) to rank and remove suspicious columns. Our results show that the effect of support-set leakage varies across tasks and relational distances. Target-table leakers cause the clearest degradation, while one- and two-hop leakers are not consistently used by the model. IG ranks target-table leakers highly across datasets and partially recovers performance in settings where leakage has the largest effect.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 27 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

18 of 27 references.

References · 18

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  2. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  3. 2019Attention is not Explanationcited by 929
  4. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  5. 2019Attention is not not Explanationcited by 636
  6. 2019Explaining individual predictions when features are dependent: More accurate approximations to Shapley valuescited by 208
  7. 2020Probing the Probing Paradigm: Does Probing Accuracy Entail Task Relevance?cited by 151
  8. 2024RelBench: A Benchmark for Deep Learning on Relational Databasescited by 96
  9. 2025Griffin: Towards a Graph-Centric Relational Database Foundation Modelcited by 47
  10. 2025Relational Transformer: Toward Zero-Shot Foundation Models for Relational Datacited by 39
  11. 2026PluRel: Synthetic Data unlocks Scaling Laws for Relational Foundation Modelscited by 19
  12. 2026KumoRFM-2: Scaling Foundation Models for Relational Learningcited by 19
6 more
  1. 2026Relational In-Context Learning via Synthetic Pre-training with Structural Priorcited by 14
  2. 2025ICLShield: Exploring and Mitigating In-Context Learning Backdoor Attackscited by 9
  3. 2026Noise Immunity in In-Context Tabular Learning: An Empirical Robustness Analysis of TabPFN's Attention Mechanismscited by 6
  4. 2026OpenRFM: Dissecting Relational In-Context Learningcited by 4
  5. 2026Task Scarcity and Label Leakage in Relational Transfer Learningcited by 2
  6. 2026GRAFT: Auditing Graph Neural Networks via Global Feature Attributioncited by 2

Built on by

Not cited yet.