Can retrieval learn what actually helps answer questions?
Standard RAG trains retrievers to find similar documents and generators to produce answers separately. But does surface similarity match what genuinely helps generate correct responses? This explores whether retrieval can receive feedback from answer quality.
Standard RAG trains the retriever and generator separately. The retriever optimizes for document relevance — returning chunks that look like what was asked. The generator optimizes for answer quality — producing correct, coherent responses from whatever the retriever provides. The two objectives are decoupled, which means the retriever can learn to retrieve documents that are semantically similar but not actually useful for answering.
The fundamental problem: the retriever cannot receive a gradient signal from the generator without a differentiable interface between them. Text is discrete — you cannot backpropagate through "select these k chunks from a vocabulary of millions."
CLaRa (Continuous Latent Reasoning) solves this with shared continuous document representations. Documents are encoded once into compact memory-token vectors. The reranker and generator both operate in this continuous space. During training, the next-token prediction loss from the generator propagates back through both modules via a differentiable top-k estimator. The retriever learns from the generator's success and failure.
The learned alignment: the retriever stops optimizing for surface similarity and starts optimizing for "does including this document improve the answer?" Documents that look relevant but do not contribute get deprioritized. Documents that seem tangential but bridge a reasoning gap get upweighted.
This matters because the gap between "similar to query" and "useful for generating the answer" is large in practice. Retrieval trained on human relevance labels is approximating what humans think is relevant. Retrieval trained on generation loss is learning what is actually useful for the downstream task.
Inquiring lines that use this note as a source 13
This note is a source for these synthesized inquiries. Follow a line forward into its question, or open it to trace back to all of its sources.
- Can task-aware ranking replace similarity scoring in other RAG systems?
- Why does retrieval quality sometimes conflict with final answer quality?
- What documents improve answers beyond surface query similarity?
- Why does standard RAG succeed for evidence-based but fail for debate questions?
- Can factually wrong generated documents still improve retrieval accuracy?
- Can retrieval strategies drive both draft refinement and new research question generation?
- Can other RAG hyperparameters like chunk size be learned through generator feedback?
- Can generator feedback backpropagate through the entire retrieval pipeline?
- Why does semantic similarity retrieval enable skill transfer to novel situations?
- How does response content compare to model confidence as a retrieval trigger?
- Can adaptive retrieval triggered by model uncertainty improve RAG reliability?
- What are the 27 external features that predict retrieval need?
- How much training data teaches retrieval models to follow instructions?
Related concepts in this collection 2
This note in its neighbourhood — explore the map, then jump to a related concept in the list below.
Click a node to walk · click center to open · click Open in graph to see this note in the full knowledge graph
-
Can document count be learned instead of fixed in RAG?
Standard RAG systems use a fixed number of documents regardless of query complexity. Can an RL agent learn to dynamically select both how many documents and their order based on what helps the generator produce correct answers?
DynamicRAG uses RL with generator output as reward signal; CLaRa uses differentiable joint training; both solve the retrieval-generation alignment problem via generator feedback
-
Do language models actually use their encoded knowledge?
Probes can detect that LMs encode facts internally, but do those encoded facts causally influence what the model generates? This explores the gap between knowing and doing.
the decoupled optimization problem in standard RAG is an instance of this: retrieval encodes information that does not causally improve generation; CLaRa is a fix
Related papers in this collection 8
Papers most semantically related to this note, ranked by cosine similarity in the embedding space.
- CLaRa: Bridging Retrieval and Generation with Continuous Latent Reasoning
- Generator-Retriever-Generator: A Novel Approach to Open-domain Question Answering
- UR2: Unify RAG and Reasoning through Reinforcement Learning
- Searching for Best Practices in Retrieval-Augmented Generation
- Chain-of-Retrieval Augmented Generation
- Towards Agentic RAG with Deep Reasoning: A Survey of RAG-Reasoning Systems in LLMs
- R1-Searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learning
- Retrieval-augmented reasoning with lean language models
Original note title
joint optimization of retriever and generator through shared continuous representations aligns retrieval with answer quality