Polaris: Learning to Generate Table Descriptions from Retrieval Feedback
Organizations: University of Wisconsin-Madison
Abstract
Many table-centric NLP tasks such as NL2SQL first retrieve relevant tables from large collections using keyword search. Recent work uses LLMs to generate natural-language table descriptions to improve retrieval, but they are typically optimized for fluency rather than retrieval effectiveness. We present Polaris, a system that trains an LLM to generate table descriptions directly from retrieval feedback. Our key insight is that existing table retrieval benchmarks already contain the supervision needed for this task: given query-table relevance judgments, we generate multiple candidate descriptions for each table, rank them by their BM25 retrieval effectiveness, and use the resulting preference pairs to fine-tune the LLM with Direct Preference Optimization (DPO). Polaris further expands abbreviated table and column names before generation to reduce vocabulary mismatch. Extensive experiments show that Polaris outperforms the state-of-the-art AutoDDG solution, often by a significant margin. More broadly, our results demonstrate that retrieval benchmarks can be repurposed as supervision for training LLMs to generate retrieval-oriented metadata.
Figures & tables
| Dataset | Domain | #Tables | #Queries | #Gold |
|---|---|---|---|---|
| AW | Enterprise | 96 | 15 | 41 |
| Arctic | Science | 251 | 20 | 121 |
| LTER | Science | 2,015 | 15 | 374 |
| ECIR | Government | 2,100 | 12 | 1,411 |
| WTR | Web | 4,634 | 60 | 2,190 |
| WikiTables | Web | 3,361 | 57 | 845 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .5551 | .6192 | .6692 | .6969 | .7030 |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| Arctic | AutoDDG | .6274 | .5942 | .5942 | .6360 | .6812 |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| LTER | AutoDDG | .4372 | .4194 | .4011 | .3656 | .3559 |
| Polaris | .7218 | .6546 | .6013 | .5239 | .5145 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .49 | .64 | .76 | .83 | .85 |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | AutoDDG | .24 | .34 | .48 | .68 | .77 |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| LTER | AutoDDG | .08 | .14 | .22 | .32 | .35 |
| Polaris | .18 | .25 | .32 | .39 | .43 |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Value |
|---|---|
| Base model | Llama 3.1 8B Instruct |
| Quantization | 4-bit |
| LoRA rank / | 64 / 64 |
| Trainable parameters | 167M / 8.2B (2.05%) |
| DPO | 0.1 |
| Learning rate |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .5551 | .6192 | .6692 | .6969 | .7030 |
| .7667 | .7948 | .8129 | .8129 | .8129 | ||
| Arctic | AutoDDG | .6274 | .5942 | .5942 | .6360 | .6812 |
| .7041 | .6651 | .6700 | .7008 | .7226 | ||
| LTER | AutoDDG | .4372 | .4194 | .4011 | .3656 | .3559 |
| .6042 | .5856 | .5353 | .4358 | .4188 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .49 | .64 | .76 | .83 | .85 |
| .69 | .78 | .82 | .82 | .82 | ||
| Arctic | AutoDDG | .24 | .34 | .48 | .68 | .77 |
| .29 | .40 | .55 | .74 | .78 | ||
| LTER | AutoDDG | .08 | .14 | .22 | .32 | .35 |
| .12 | .18 | .26 | .33 | .37 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | Polaris | .7312 | .7582 | .7869 | .7869 | .7869 |
| .6681 | .7287 | .7794 | .7794 | .7794 | ||
| .6940 | .7485 | .7847 | .7847 | .7847 | ||
| Arctic | Polaris | .7133 | .6743 | .6939 | .7416 | .7678 |
| .7209 | .6785 | .6891 | .7334 | .7535 | ||
| .6690 | .6493 | .6737 | .6872 | .7106 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | Polaris | .66 | .77 | .83 | .83 | .83 |
| .61 | .76 | .89 | .89 | .89 | ||
| .65 | .79 | .87 | .87 | .87 | ||
| Arctic | Polaris | .29 | .41 | .58 | .77 | .82 |
| .30 | .40 | .56 | .76 | .80 | ||
| .27 | .39 | .57 | .72 | .77 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .8265 | .8390 | .8733 | .8733 | .8733 | |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| Arctic | .7303 | .6952 | .7071 | .7425 | .7690 | |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| LTER | .6269 | .5964 | .5393 | .4439 | .4259 | |
| Polaris | .7218 | .6546 | .6013 | .5239 | .5145 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .73 | .81 | .88 | .88 | .88 | |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | .29 | .42 | .60 | .77 | .82 | |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| LTER | .11 | .18 | .27 | .34 | .36 | |
| Polaris | .18 | .25 | .32 | .39 | .43 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .4853 | .5286 | .5946 | .6306 | .6306 | |
| .8265 | .8390 | .8733 | .8733 | .8733 | ||
| Arctic | .6861 | .6586 | .6786 | .7055 | .7437 | |
| .7303 | .6952 | .7071 | .7425 | .7690 | ||
| LTER | .3379 | .3269 | .3275 | .3380 | .3440 | |
| .6269 | .5964 | .5393 | .4439 | .4259 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .42 | .55 | .73 | .83 | .83 | |
| AutoDDG | .49 | .64 | .76 | .83 | .85 | |
| .73 | .81 | .88 | .88 | .88 | ||
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | .29 | .41 | .58 | .74 | .81 | |
| AutoDDG | .24 | .34 | .48 | .68 | .77 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .7398 | .7662 | .8051 | .8051 | .8051 | |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| Arctic | .7266 | .6874 | .6837 | .7380 | .7585 | |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| LTER | .6760 | .6151 | .5806 | .4929 | .4732 | |
| Polaris | .7218 | .6546 | .6013 | .5239 | .5145 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .68 | .77 | .86 | .86 | .86 | |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | .30 | .42 | .56 | .78 | .82 | |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| LTER | .13 | .20 | .29 | .37 | .40 | |
| Polaris | .18 | .25 | .32 | .39 | .43 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .8019 | .8313 | .8601 | .8601 | .8601 | |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| Arctic | .6843 | .6593 | .7013 | .7446 | .7645 | |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| LTER | .6364 | .5987 | .5567 | .4805 | .4694 | |
| Polaris | .7218 | .6546 | .6013 | .5239 | .5145 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .73 | .84 | .90 | .90 | .90 | |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | .29 | .41 | .61 | .80 | .84 | |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| LTER | .13 | .20 | .28 | .35 | .39 | |
| Polaris | .18 | .25 | .32 | .39 | .43 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .6908 | .7522 | .7881 | .7968 | .7968 | |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| Arctic | .8080 | .7131 | .7059 | .7595 | .7951 | |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| LTER | .7064 | .6535 | .6020 | .5105 | .4940 | |
| Polaris | .7218 | .6546 | .6013 | .5239 | .5145 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .65 | .83 | .91 | .93 | .93 | |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| Arctic | .32 | .42 | .59 | .80 | .86 | |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| LTER | .16 | .23 | .31 | .38 | .42 | |
| Polaris | .18 | .25 | .32 | .39 | .43 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .5551 | .6192 | .6692 | .6969 | .7030 |
| Polaris | .7312 | .7582 | .7869 | .7869 | .7869 | |
| .8133 | .8282 | .8677 | .8677 | .8677 | ||
| Arctic | AutoDDG | .6274 | .5942 | .5942 | .6360 | .6812 |
| Polaris | .7133 | .6743 | .6939 | .7416 | .7678 | |
| .7303 | .7146 | .7330 | .7755 | .8058 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | AutoDDG | .49 | .64 | .76 | .83 | .85 |
| Polaris | .66 | .77 | .83 | .83 | .83 | |
| .75 | .83 | .92 | .92 | .92 | ||
| Arctic | AutoDDG | .24 | .34 | .48 | .68 | .77 |
| Polaris | .29 | .41 | .58 | .77 | .82 | |
| .32 | .46 | .63 | .82 | .88 |
| Dataset | Method | @5 | @10 | @20 | @50 | @100 |
|---|---|---|---|---|---|---|
| AW | .8133 | .8282 | .8677 | .8677 | .8677 | |
| .7594 | .7668 | .7895 | .7895 | .7895 | ||
| .7312 | .7582 | .7869 | .7869 | .7869 | ||
| Arctic | .7303 | .7146 | .7330 | .7755 | .8058 | |
| .7602 | .7162 | .6726 | .7139 | .7255 | ||
| .7133 | .6743 | .6939 | .7416 | .7678 |
Table Retrieval
Table retrieval finds relevant tables and schema elements for question answering, analytics, and text-to-SQL.
See all 19 papers