What Do Tabular Foundation Models Compute In Context? In-Situ Representation Refinement through Attention-Gated Updates

Tian Zhou, Beverly Jin, Linxiao Yang, Xue Wang, Wenwei Wang +4 more
9/23/2026
cs.LG

No confident code implementation found for this paper.

Abstract

What reusable computation should a tabular foundation model learn when every table defines a new supervised task? We develop in-situ representation refinement: support labels guide updates to the episode's representations, and these updates transfer to unlabeled queries without changing model parameters. A regularized leave-one-out objective yields a support correction and its query extension. The leading term separates attention-based reading from state-dependent scaling, motivating RefineICL: an attention-gated, FFN-free contextual stack with selected low-rank feature interaction and typed memory. RefineICL-L24 reaches 0.93836 OVR-AUC and 0.87173 accuracy on AMLB29. A benchmark-informed continuation reaches 1644.8 Elo on the 38-dataset TabArena snapshot, 31.4 Elo above TabPFN-3 under the same evaluation. It also improves all four reported metrics over TabPFN-v3 on both TabZilla views. In a matched 100K-update depth grid, an expanded FFN gives no consistent validation benefit and uses 60.2% more peak inference memory at L8. Internal interventions show that support representations are more than a static source of labels: removing one intermediate support update, while preserving the query output, increases final query cross-entropy in all 72 tested episodes. Together, the derivation and interventions explain how attention-gated updates can construct a task-specific predictor in context.

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{zhou2026what,
  title  = {What Do Tabular Foundation Models Compute In Context? In-Situ Representation Refinement through Attention-Gated Updates},
  author = {Tian Zhou and Beverly Jin and Linxiao Yang and Xue Wang and Wenwei Wang and Bingqing Peng and Mengni Ye and Jinjie Gu and Liang Sun},
  year   = {2026},
  eprint = {2609.27679},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.27679v1}
}

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/2609.27679.svg)](https://codeofpaper.com/paper/2609.27679)

Discussion