atlas

Research map / Large language models / Instruction and fine-tuning

Large Scale Transfer Learning for Tabular Data via Language Modeling

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Efficient LLM fine-tuning systems · 12 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Fine-tunes Llama 3-8B on a filtered 100B-token tabular corpus (T4 from TabLib) with a row-causal tabular attention mask to get TabuLa-8B, a training-free in-context tabular predictor that beats XGBoost and TabPFN in few-shot settings.

Problem

Tabular prediction usually requires task-specific models trained on the target data; transfer learning has not matched the impact seen in language and vision, and LLM-based tabular predictors have lacked large-scale training data and out-of-distribution competitiveness.

Approach

Build a large, filtered corpus of tables (T4 from TabLib), serialize rows as text with unsupervised target selection, and fine-tune Llama 3-8B with a row-causal tabular masking (RCTM) scheme that lets rows of the same table attend to each other within a packed batch, so the model learns to use multiple shots in context.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune Llama 3-8B on T4 for tabular predictionLlama 3-8B · Five tabular benchmarks, zero-shotaccuracy relative to random guessing: 17 percentage points above random guessingHelped
Use TabuLa-8B instead of XGBoost or TabPFNTabuLa-8B · Five tabular benchmarks, 1-32 shotsaccuracy: 5-20ppHelped
Use TabuLa-8B instead of Llama 3-8BLlama 3-8B · Five tabular benchmarks, few-shotaccuracy: 10 to 20ppHelped
Add row-causal tabular masking (RCTM) instead of per-sample causal attention (ablation)TabuLa-8B · Full test suite, few-shotaccuracyHelped
Replace informative column headers with X1, X2, ... and target header Y (ablation)TabuLa-8B · UniPredict, 16 shotsaccuracy: 3-5ppHurt
Replace informative column headers with uninformative headers (ablation)TabuLa-8B · UniPredict, 32 shotsaccuracyNo clear effect
Remove features at evaluation, especially unimportant features (ablation)TabuLa-8B · Random sample of 32 datasets, 0-, 4-, 16-, and 32-shotaccuracyMixed
Randomly permute columns (ablation)TabuLa-8B · Random subset of 32 tasks, 0, 8, 16, and 32 shotsaccuracyNo clear effect
Use full data filtering instead of minimal filtering (ablation)TabuLa-8B · Tabular test suite, 16 or more shotsaccuracyHelped
Use full data filtering instead of minimal filtering (ablation)TabuLa-8B · Tabular test suite, fewer than 16 shotsaccuracyNo clear effect
Use a stronger base language model, from Llama 1 through Llama 2 to Llama 3 (ablation)TabuLa-8B training pipeline · Tabular test suite, up to 8 shotsaccuracyHelped
Evaluate separately on potentially contaminated versus uncontaminated tablesTabuLa-8B · Evaluation test suiteaccuracyNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: T4 (filtered from TabLib) UniPredict Grinsztajn benchmark AutoML Multimodal Benchmark OpenML CC-18 OpenML CTR-23

Benchmarks: UniPredict Grinsztajn AutoML Multimodal Benchmark OpenML CC-18 OpenML CTR-23

Metrics: open-vocabulary accuracy zero-shot accuracy few-shot accuracy relative sample efficiency

Baselines: Llama 3-8B XGBoost TabPFN CatBoost Logistic Regression Claude Instant Claude 3 Sonnet

Builds on: Llama 3 TabPFN XGBoost TabLib in-context pretraining

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy