Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Strengthening LLMs for Tabular Prediction with Structural Priors

About

Tabular prediction has long been dominated by gradient-boosted decision trees and specialized deep tabular models, while large language models (LLMs) remain difficult to make competitive despite their cross-task adaptability and transparent reasoning traces. We address this gap by incorporating tabular structural priors into LLM post-training. Specifically, we propose Permutation Relative Policy Optimization (PRPO), which operationalizes column-permutation invariance through label-preserving column permutations and two-level advantage estimation. This design converts sparse outcome rewards into denser and more stable optimization signals. Extensive experiments on 139 OpenML datasets show that our 8B model reaches a genuinely competitive regime against strong specialized tabular baselines. It achieves strong fully supervised performance, dominates zero-shot settings, and performs on par with 32-shot strong baselines. Moreover, it substantially outperforms much larger general-purpose and reasoning LLMs, including up to a 53.17% improvement over DeepSeek-R1 (685B). These results show that structural-prior RL post-training is an effective route for making LLMs competitive in tabular prediction.

Pengxiang Cai, Zihao Gao, Wanchen Lian, Guocong Li, Jintai Chen• 2025

Related benchmarks

TaskDatasetResultRank
Tabular Classification53 classification datasets (unseen)
Mean Accuracy75.42
18
Tabular Regression21 Regression Datasets
Mean NMAE0.111
15
Tabular Classification50 classification datasets
Mean Accuracy84.36
10
Tabular Regression15 regression datasets
Mean NMAE0.1499
10
Showing 4 of 4 rows

Other info

Follow for update