CARTE: Pretraining and Transfer for Tabular Learning - Archive ouverte HAL
Pré-Publication, Document De Travail (Preprint/Prepublication) Année : 2024

CARTE: Pretraining and Transfer for Tabular Learning

Résumé

Pretrained deep-learning models are the go-to solution for images or text. However, for tabular data the standard is still to train tree-based models. Indeed, transfer learning on tables hits the challenge of data integration: finding correspondences, correspondences in the entries (entity matching) where different words may denote the same entity, correspondences across columns (schema matching), which may come in different orders, names... We propose a neural architecture that does not need such correspondences. As a result, we can pretrain it on background data that has not been matched. The architecture --CARTE for Context Aware Representation of Table Entries-- uses a graph representation of tabular (or relational) data to process tables with different columns, string embedding of entries and columns names to model an open vocabulary, and a graph-attentional network to contextualize entries with column names and neighboring entries. An extensive benchmark shows that CARTE facilitates learning, outperforming a solid set of baselines including the best tree-based models. CARTE also enables joint learning across tables with unmatched columns, enhancing a small table with bigger ones. CARTE opens the door to large pretrained models for tabular data.
Fichier principal
Vignette du fichier
CARTE_ICML_2024.pdf (1.1 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-04596816 , version 1 (31-05-2024)
hal-04596816 , version 2 (24-09-2024)

Licence

Identifiants

  • HAL Id : hal-04596816 , version 1

Citer

Myung Jun Kim, Léo Grinsztajn, Gaël Varoquaux. CARTE: Pretraining and Transfer for Tabular Learning. 2024. ⟨hal-04596816v1⟩
141 Consultations
69 Téléchargements

Partager

More