← all repositories

pat-jj/DeepRetrieval

DeepRetrieval trains Large Language Models as search agents using reinforcement learning to generate better queries for information retrieval.

DeepRetrieval
Velocity · 7d
+1.5
★ / day
Trend
steady
star history

DeepRetrieval is a novel reinforcement learning framework that trains LLMs for query generation to enhance information retrieval. Unlike supervised approaches, the system learns through direct trial and error using retrieval metrics as rewards. The LLM generates reasoning steps in a think section followed by the final augmented query, enabling explicit chain-of-thought reasoning before query formulation. The project includes model weights and easy-to-use APIs for query rewriting.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.