Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CGPT: Cluster-Guided Partial Tables with LLM-Generated Supervision for Table Retrieval

About

General-purpose embedding models have demonstrated strong performance in text retrieval but remain suboptimal for table retrieval, where highly structured content leads to semantic compression and query-table mismatch. Recent LLM-based retrieval augmentation methods mitigate this issue by generating synthetic queries, yet they often rely on heuristic partial-table selection and seldom leverage these synthetic queries as supervision to improve the embedding model. We introduce CGPT, a training framework that enhances table retrieval through LLM-generated supervision. CGPT constructs semantically diverse partial tables by clustering table instances using K-means and sampling across clusters to broaden semantic coverage. An LLM then generates synthetic queries for these partial tables, which are used in hard-negative contrastive fine-tuning to refine the embedding model. Experiments across four public benchmarks (MimoTable, OTTQA, FetaQA, and E2E-WTQ) show that CGPT consistently outperforms retrieval baselines, including QGpT, with an average R@1 improvement of 16.54 percent. In a unified multi-domain corpus setting, CGPT further demonstrates strong cross-domain generalization and remains effective even when using smaller LLMs for synthetic query generation. These results indicate that semantically guided partial-table construction, combined with contrastive training from LLM-generated supervision, provides an effective and scalable paradigm for large-scale table retrieval. Our code is available at https://github.com/yumeow0122/CGPT.

Tsung-Hsiang Chou, Chen-Jui Yu, Shui-Hsiang Hsu, Yao-Chung Fan• 2026

Related benchmarks

TaskDatasetResultRank
Table RetrievalMimo ch
R@156.8
15
Table RetrievalMimo en
R@160.13
15
Table RetrievalOTTQA
R@186.86
15
Table RetrievalE2E-WTQ
R@172.2
15
Table RetrievalFeTaQA
R@134.9
15
Showing 5 of 5 rows

Other info

GitHub

Follow for update