Work

RetroLLM: Empowering Large Language Models to Retrieve Fine-grained Evidence within Generation

Year: 2025

Type: conference-paper

Abstract: Large language models (LLMs) exhibit remarkable generative capabilities but often suffer from hallucinations.Retrieval-augmented generation (RAG) offers an effective solution by incorporating external knowledge, but existing methods still face several limitations: additional deployment costs of separate retrievers, redundant input tokens from retrieved text chunks, and the lack of joint optimization of retrieval and generation.To address these issues, we propose RetroLLM, a unified framework that integrates retrieval and generation into a single, auto-regressive process, enabling LLMs to directly generate fine-grained evidence from the corpus with constrained decoding.Moreover, to mitigate false pruning in the process of constrained evidence generation, we introduce (1) hierarchical FM-Index constraints, which generate corpus-constrained clues to identify a subset of relevant documents before evidence generation, reducing irrelevant decoding space; and (2) a forward-looking constrained decoding strategy, which considers the relevance of future sequences to improve evidence accuracy.Extensive experiments on five open-domain QA datasets demonstrate RetroLLM's superior performance across both in-domain and out-ofdomain tasks.The code is available at https: //github.com/sunnynexus/RetroLLM.

Source: Annual Meeting of the Association for Computational Linguistics (ACL)

DOI: 10.18653/v1/2025.acl-long.819