atlas

Research map / Graph learning / Knowledge and temporal graphs

CARTE: Pretraining and Transfer for Tabular Learning

ICML 2024 · Graph learning › Knowledge and temporal graphs · approach: Heterogeneous graph learning · 1 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.

Proposes CARTE, a graph-attention transformer that represents each table row as a star graphlet with language-model embeddings of entries and column names, pretrained contrastively on YAGO3, enabling transfer across tables without schema or entity matching.

Problem

Transfer learning and pretraining on tabular data is hard because it requires data integration (schema matching across columns and entity matching across string entries), so tree-based models remain the standard.

Approach

Represent each row as a star-like graphlet whose leaves are cell values and column names embedded by a language model (FastText or e5), and use a graph-attention network with edge-aware attention to contextualize entries; pretrain self-supervised with an InfoNCE contrastive loss on graphlets sampled from the YAGO3 knowledge graph, so unmatched tables can share one representation space.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use CARTE instead of the strongest alternatives for single-table learningCARTE · single-table tasks across regression and classification datasetsnormalized score and critical-difference rankingHelped
Add bagging with different train/validation splits for early stopping (ablation)neural network models · single-table tasksprediction performanceHelped
Add source tables for joint learningCARTE · related multi-table taskstarget-table performanceHelped
Use CARTE rather than CatBoost, TabVec-XGB, or TabVec-RF when features are missingCARTE · 51 datasets with randomly dropped featurespercentage decrease in performance: CARTE 13.28%, 38.35%, 10.19%, 24.42%; CatBoost 21.70%, 53.32%, 12.23%, 29.70%Helped
Use entity string representations without manual entity matchingCARTE · company employees, movies, US accidents, and US electionperformance robustness to entity variantsNo clear effect
Remove attention from the CARTE architecture (ablation)CARTE · downstream tabular tasksprediction performanceHurt
Remove string-level representations that capture semantic similarity (ablation)CARTE · downstream tabular tasksprediction performanceHurt
Use CARTE instead of CatBoost for single-table learningCARTE · 51 datasets, learning time at n=64 and n=512learning time: 0.98$\pm$1.19 vs 85.43$\pm$60.30; 1.05$\pm$1.06 vs 315.49$\pm$119.84Hurt

Surprising findings

Limitations

Experimental setup

Datasets: YAGO3 51 tabular datasets from Kaggle, OpenML and prior ML studies TabLLM datasets

Benchmarks: 51 tabular datasets (40 regression, 11 classification) TabLLM nine datasets

Metrics: R2 AUROC normalized score critical difference diagram

Baselines: CatBoost XGBoost HistGradientBoosting RandomForest MLP ResNet TabPFN S-LLM TabLLM

Builds on: TabPFN TabLLM KEN FastText InfoNCE

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