GOTabPFN: From Feature Ordering to Compact Tokenization for Tabular Foundation Models on High-Dimensional Data

Al Zadid Sultan Bin Habib, Md Younus Ahamed, Prashnna Kumar Gyawali, Gianfranco Doretto, Donald A. Adjeroh
6/3/2026
cs.LGcs.AIstat.ML

No official code found. Best implementation: zadid6pretam/GOTabPFN (11 stars, MIT, last commit 2026-06-03).

Abstract

We investigate how to make small tabular foundation models effective for High-Dimensional, Low-Sample Size (HDLSS) tabular prediction without retraining large backbones. We introduce Graph-guided Ordering with Local Refinement (GO-LR), show its equivalence to weighted Minimum Linear Arrangement, and interpret the practical solver as a TSP-path-style surrogate. We propose GOTabPFN,which builds on GO-LR, and a Neuro-Inspired Subunit Compression (NSC) unit to pool locally adjacent ordered features into meta-features, yielding a compact representation that makes TabPFN-style prediction practical in HDLSS regimes. Across tabular benchmarks, GOTabPFN improves stability and accuracy under tight token budgets.

View on arXivView PDF

Code Implementations(1)

MIT

GOTabPFN: Graph-guided feature ordering and compact tokenization for TabPFN-style tabular foundation models on high-dimensional data.

110May 1, 20264 months agoMIT
compression-algorithmdimensionality-reductionfeature-orderinghigh-dimensional-datameta-features+3 more

Similar Papers

Cite this paper

@article{habib2026gotabpfn,
  title  = {GOTabPFN: From Feature Ordering to Compact Tokenization for Tabular Foundation Models on High-Dimensional Data},
  author = {Al Zadid Sultan Bin Habib and Md Younus Ahamed and Prashnna Kumar Gyawali and Gianfranco Doretto and Donald A. Adjeroh},
  year   = {2026},
  eprint = {2606.05441},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.05441v1}
}

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

Discussion