Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces KAMR, a knowledge-aligned multi-hop retriever that enhances graph-based retrieval-augmented generation by distinguishing between strongly constrained anchor triplets and weakly aligned connected triplets. By addressing the limitations of existing methods that rank triplets independently, KAMR employs a novel partial alignment dataset and optimizes for both pair-level and element-level matching through contrastive objectives. The results demonstrate that KAMR significantly outperforms fourteen baselines across multiple benchmarks and LLM backbones, improving both multi-hop retrieval and question answering tasks.
KAMR reveals that optimizing for query-triplet alignment can dramatically enhance multi-hop retrieval effectiveness, outperforming existing methods by a significant margin.
Graph-based retrieval-augmented generation increasingly relies on multi-hop retrieval, where answering a query requires composing multiple connected knowledge-graph triplets. However, existing retrievers often rank triplets independently via global semantic matching. Moreover, many multi-hop benchmarks provide only final answers, which limits supervision for query--triplet alignment and causes structurally necessary but weakly aligned facts to be missed. To address these issues, we propose a knowledge-aligned multi-hop retriever, KAMR, which distinguishes anchor triplets that are strongly constrained by the query from connected triplets that are weakly aligned yet structurally linked to the anchors. To mitigate the lack of query-triplet alignment supervision, we build a partial alignment dataset by masking triplet elements and prompting an LLM to generate corresponding queries, and optimize two contrastive objectives for pair-level and element-level matching. At inference time, KAMR retrieves anchors globally and then expands locally to collect connected evidence. Across four benchmarks, three LLM backbones, and fourteen baselines, KAMR consistently improves multi-hop retrieval and downstream question answering performance.