FGTR: Fine-Grained Multi-Table Retrieval via Hierarchical LLM Reasoning
Organizations: Zhejiang University of Technology Hangzhou, China · University of Aberdeen Aberdeen, UK
Abstract
With the rapid advancement of large language models (LLMs), growing efforts have been made on LLM-based table retrieval. However, existing studies typically focus on single-table query, and implement it by similarity matching after encoding the entire table. These methods usually result in low accuracy due to their coarse-grained encoding which incorporates much query-irrelated data, and are also inefficient when dealing with large tables, failing to fully utilize the reasoning capabilities of LLM. Further, multi-table query is under-explored in retrieval tasks. To this end, we propose a hierarchical multi-table query method based on LLM: Fine-Grained Multi-Table Retrieval FGTR, a new retrieval paradigm that employs a human-like reasoning strategy. Through hierarchical reasoning, FGTR first identifies relevant schema elements and then retrieves the corresponding cell contents, ultimately constructing a concise and accurate sub-table that aligns with the given query. To comprehensively evaluate the performance of FGTR, we construct two new benchmark datasets based on Spider and BIRD . Experimental results show that FGTR outperforms previous state-of-the-art methods, improving the F_2 metric by 18% on Spider and 21% on BIRD, demonstrating its effectiveness in enhancing fine-grained retrieval and its potential to improve end-to-end performance on table-based downstream tasks.
Figures & tables
| Method | Spider | BIRD | ||||||
| R | P | SR | R | P | SR | |||
| Full-schema | 100 | 11.94 | 100 | 100 | 100 | 8.87 | 100 | 100 |
| Gold-Based | 100 | 100 | 100 | 100 | 100 | 100 | 100 | 100 |
| Backbone (Llama3-8b) | ||||||||
| C3 | 70.74 | 74.83 | 71.52 | 32.45 | 52.00 | 75.00 | 55.40 | 28.34 |
| FGTR | 74.12 | 80.32 | 75.28 | 39.54 | 72.28 | 76.04 | 73.00 | 35.19 |
| Method | Spider | BIRD | ||||||
| R | P | SR | R | P | SR | |||
| Full BD | 100 | 8.07 | - | 100 | 100 | 2.94 | - | 100 |
| Gold-based | 100 | 100 | 100 | 100 | 100 | 100 | 100 | 100 |
| TableRAG | 92.84 | 8.82 | 31.96 | 63.08 | 87.40 | 5.70 | 22.60 | 42.97 |
| FGTR (k=3) | 94.98 | 32.74 | 68.82 | 72.73 | 96.74 | 26.17 | 62.85 | 76.13 |
| FGTR (k=5) | 97.73 | 24.31 | 69.93 | 84.43 | 97.42 | 18.19 | 52.06 | 83.20 |
| Method | Components | Downstream Accuracy (%) | ||||
| SR | CR | BirdQA | SpiderQA | TabFACT | WTQ | |
| FGTR | ✓ | ✓ | 53.76 | 78.26 | 76.25 | 60.27 |
| ✓ | ✗ | – | 75.21 | 74.84 | 55.47 | |
| ✗ | ✓ | 46.72 | 74.80 | 67.20 | 56.28 | |
| ✗ | ✗ | – | 66.08 | 62.90 | 53.00 | |
| TableRAG | – | 45.50 | 48.61 | 69.42 | 56.63 | |
| Method | Spider | BIRD | ||||||||||
| Schema Retrieval | Cells Retrieval | Schema Retrieval | Cells Retrieval | |||||||||
| R | P | R | P | R | P | R | P | |||||
| FGTR | 98.32 | 70.72 | 91.20 | 97.78 | 84.75 | 94.86 | 97.27 | 72.00 | 90.89 | 92.48 | 90.54 | 92.09 |
| -w/o column explaining | 96.63 | 68.41 | 89.27 | - | - | - | 92.17 | 70.30 | 86.84 | - | - | - |
| -w/o question parsing | 95.71 | 59.73 | 85.42 | - | - | - | 93.54 | 65.20 | 86.06 | - | - | - |
| -w/o schema mapping | 91.62 | 76.38 | 88.21 | - | - | - | 89.08 | 81.33 | 87.43 | - | - | - |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Overall | Per Question | |||||
| # DBs | # Questions | Tables/DB | Cols/Table | Rows/Table | Tables/Q | Cols/Q | |
| SpiderQA | 166 | 965 | 5.13 | 5.01 | 17 | 1.52 | 2.99 |
| BIRDQA | 95 | 1,182 | 7.25 | 10.66 | 55,448 | 2.17 | 4.95 |
| WTQ | 1,876 | 2,000 | 1.0 | 6.41 | 28.01 | 1.00 | - |
| TabFACT | 1,695 | 2,000 | 1.0 | 6.25 | 13.43 | 1.00 | - |
Table Retrieval
Table retrieval finds relevant tables and schema elements for question answering, analytics, and text-to-SQL.
See all 19 papers