ArXiv TLDR

TabEmbed: Benchmarking and Learning Generalist Embeddings for Tabular Understanding

🐦 Tweet
2605.04962

Minjie Qiang, Mingming Zhang, Xiaoyi Bao, Xing Fu, Yu Cheng + 3 more

cs.CLcs.IR

TLDR

TabEmbed introduces a generalist embedding model for tabular data, unifying classification and retrieval, alongside TabBench for evaluation.

Key contributions

  • Introduces TabBench, a comprehensive benchmark for evaluating tabular embedding models.
  • Proposes TabEmbed, the first generalist embedding model for tabular data.
  • Unifies tabular classification and retrieval within a shared embedding space.
  • Leverages contrastive learning with hard negative mining for fine-grained nuances.

Why it matters

Foundation models for tabular data are largely unexplored. This paper bridges the gap by providing TabBench, a benchmark, and TabEmbed, a generalist model that unifies classification and retrieval. It sets a new baseline for universal tabular representation learning, addressing limitations of existing LLM and text embedding approaches.

Original Abstract

Foundation models have established unified representations for natural language processing, yet this paradigm remains largely unexplored for tabular data. Existing methods face fundamental limitations: LLM-based approaches lack retrieval-compatible vector outputs, whereas text embedding models often fail to capture tabular structure and numerical semantics. To bridge this gap, we first introduce the Tabular Embedding Benchmark (TabBench), a comprehensive suite designed to evaluate the tabular understanding capability of embedding models. We then propose TabEmbed, the first generalist embedding model that unifies tabular classification and retrieval within a shared embedding space. By reformulating diverse tabular tasks as semantic matching problems, TabEmbed leverages large-scale contrastive learning with positive-aware hard negative mining to discern fine-grained structural and numerical nuances. Experimental results on TabBench demonstrate that TabEmbed significantly outperforms state-of-the-art text embedding models, establishing a new baseline for universal tabular representation learning. Code and datasets are publicly available at https://github.com/qiangminjie27/TabEmbed and https://huggingface.co/datasets/qiangminjie27/TabBench.

📬 Weekly AI Paper Digest

Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.