Co-adaptive Rule Routing and Constraining for Structured Knowledge Retrieval
Organizations: University of Georgia · Michigan State University · Georgia Institute of Technology · Amazon ·
Abstract
Structured knowledge retrieval acquires task-relevant evidence from structured knowledge sources for downstream tasks. Existing methods rely on unconstrained retrieval or predefined query-rule supervision, such as author-paper rules for scientific discovery and anatomy-drug rules for precision medicine. Such supervision is rarely available in practice. We propose Co-Adaptive Rule Routing and Constraining Retrieval (CoRE), which jointly learns which reasoning rule to assign to each query and how to retrieve evidence under that rule, without predefined query-rule pairs. A routing agent selects a rule, and its corresponding retriever traverses the knowledge base and ranks evidence that satisfies the rule's constraints. Because routing and retrieval are interdependent, CoRE alternates between reinforcement learning for the routing policy and contrastive learning for the rule-constrained retriever. The selected rules provide supervision for retriever training, while retrieval performance rewards the routing policy. This closed-loop process enables both components to co-adapt without explicit query-rule supervision. Experiments on STaRK-QA datasets across scientific discovery, biomedical reasoning, and e-commerce knowledge management demonstrate CoRE's effectiveness. Ablations show the benefit of co-adapting the router and retriever. Generation-based routing transfers across datasets and benefits from multi-domain pretraining followed by target-specific fine-tuning. Analysis of rule usage across queries also informs automatic rule refinement and discovery.
Figures & tables
| Category | Retrieval Baseline | AMAZON | MAG | PRIME | AVERAGE | ||||||||||||
| H@1 | H@5 | R@20 | MRR | H@1 | H@5 | R@20 | MRR | H@1 | H@5 | R@20 | MRR | H@1 | H@5 | R@20 | MRR | ||
| Textual | BM25 | 40.62 | 62.18 | 50.89 | 50.59 | 23.11 | 44.47 | 44.02 | 33.23 | 11.96 | 25.67 | 28.31 | 18.35 | 25.23 | 44.11 | 41.07 | 34.06 |
| Ada | 39.16 | 62.73 | 53.29 | 50.35 | 29.08 | 49.61 | 48.36 | 38.62 | 12.63 | 31.49 | 36.00 | 21.41 | 26.96 | 47.94 | 45.88 | 36.79 | |
| Multi-Ada | 40.07 | 64.98 | 55.12 | 51.55 | 25.92 | 50.43 | 50.80 | 36.94 | 15.10 | 33.56 | 38.05 | 23.49 | 27.03 | 49.66 | 47.99 | 37.33 | |
| DPR | 15.29 | 47.93 | 44.49 | 30.20 | 10.51 | 35.23 | 42.11 | 21.34 | 4.46 | 21.85 | 30.13 | 12.38 | 10.09 | 35.00 | 38.91 | 21.31 | |
| Graph | QAGNN | 26.56 | 50.01 | 52.05 | 37.75 | 12.88 | 39.01 | 46.97 | 29.12 | 8.85 | 21.35 | 29.63 | 14.73 | 16.10 | 36.79 | 42.88 | 27.20 |
| Method | AMAZON | MAG | PRIME | Average | ||||
| H@1 | MRR | H@1 | MRR | H@1 | MRR | H@1 | MRR | |
| CoRE-BERT-C | 50.71 | 62.46 | 63.37 | 74.17 | 49.34 | 61.27 | 54.47 | 65.97 |
| w/o Agent-Adapt (unadapted BERT) | 14.63 | 19.83 | 28.23 | 35.05 | 5.15 | 8.15 | 16.00 | 21.01 |
| w/o Agent-Adapt (GPT-5-mini) | 26.70 | 32.72 | 28.23 | 31.87 | 4.14 | 6.66 | 19.69 | 23.75 |
| w/o Retriever-Adapt (BM25-traversal) | 49.87 | 61.45 | 55.51 | 65.45 | 38.86 | 49.70 | 48.08 | 58.87 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| STaRK-Amazon | STaRK-MAG | STaRK-Prime | |
| #entity types | 4 | 4 | 10 |
| #relation types | 5 | 4 | 18 |
| #entities | 1,035,542 | 1,872,968 | 129,375 |
| #relations | 9,443,802 | 39,802,116 | 8,100,498 |
| #tokens | 592,067,882 | 212,602,571 | 31,844,769 |
| #queries | 9,181 | 13,407 | 11,302 |
| Key | Traversal Path |
| A | Product |
| B | Brand Product |
| C | Product Product |
| D | Category Product |
| E | Product Category Product |
| F | Product Brand Product |
| Key | Traversal Path |
| A | Paper |
| B | Author Paper |
| C | Paper Paper |
| D | Field of Study Paper |
| E | Institution Author Paper |
| F | Paper Author Paper |
| Key | Traversal Path |
|---|---|
| R00 | Anatomy Expression Present Gene/Protein |
| R01 | Anatomy Expression Absent Gene/Protein |
| R02 | Pathway Interacts With Gene/Protein Expression Absent Anatomy |
| R03 | Gene/Protein Target Drug |
| R04 | Disease Associated With Gene/Protein |
| R05 | Gene/Protein PPI Gene/Protein |
Information Retrieval
Information Retrieval studies how to find, rank, and organize relevant information in large collections of data.
See all 474 papers