{LoopICL}: {Looping} a single transformer block to solve tabular tasks

Tabular foundation models using in-context learning have recently surpassed gradient-boosted trees on predictive tabular tasks. However, recent mechanistic insights suggest that parameters in these models are largely redundant. We introduce LoopICL, a looped transformer whose core design decouples parameter count from computational depth. LoopICL consists of a single block, processing data through two coupled streams: a cell stream capturing per-cell feature representations and a row stream capturing in-context example representations, jointly refined through within-column and cross-column attention. During pre-training, we vary loop counts, allowing the block to be unrolled for a varying number of iterations at test-time and use a learned exit-gate to automatically exit. In its standard setting, LoopICL performs competitively with TabICLv2 on TabArena and TALENT at the same computational cost (FLOPs), while using nearly 90\% fewer parameters. Furthermore, its recurrent design enables users to also trade off inference cost and performance, providing a resource-aware TFM.

Informationen zur Zitierung

Balef, Amir Rezaei; Eggensperger, Katharina: {LoopICL}: {Looping} a single transformer block to solve tabular tasks, arXiv, 2026, September, arXiv, http://arxiv.org/abs/2609.36108, Balef.Eggensperger.2026a,

Assoziierte Lamarr-ForscherInnen

Photo. Portrait of Katharina Eggensperger.

Prof. Dr. Katharina Eggensperger

Principal Investigator Ressourcenbewusstes ML zum Profil