Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

TABBIE: Pretrained Representations of Tabular Data

About

Existing work on tabular representation learning jointly models tables and associated text using self-supervised objective functions derived from pretrained language models such as BERT. While this joint pretraining improves tasks involving paired tables and text (e.g., answering questions about tables), we show that it underperforms on tasks that operate over tables without any associated text (e.g., populating missing cells). We devise a simple pretraining objective (corrupt cell detection) that learns exclusively from tabular data and reaches the state-of-the-art on a suite of table based prediction tasks. Unlike competing approaches, our model (TABBIE) provides embeddings of all table substructures (cells, rows, and columns), and it also requires far less compute to train. A qualitative analysis of our model's learned cell, column, and row representations shows that it understands complex table semantics and numerical trends.

Hiroshi Iida, Dung Thai, Varun Manjunatha, Mohit Iyyer• 2021

Related benchmarks

TaskDatasetResultRank
Column Type AnnotationPublicBI to GitTables
SW F166.2
32
Column Type AnnotationSemtab low-resource 2019
SW F158
26
Column PopulationWikipedia tables (test)
MAP54.5
15
Measure Type PredictionAnaMeta
Accuracy77.39
14
Row PopulationWikipedia Tables Row Population
MAP44.4
12
Column Type PredictionVizNet
Support-weighted F196.9
11
Common Measure IdentificationAnaMeta
HR@171.16
10
Measure Pair IdentificationAnaMeta
Accuracy77.7
10
Natural Key IdentificationAnaMeta
Accuracy94.42
10
Common Breakdown IdentificationAnaMeta
HR@162.36
10
Showing 10 of 23 rows

Other info

Code

Follow for update