DREAM method trains dense retrievers using frozen LLMs, outperforms contrastive baselines
A new retrieval training method called DREAM derives supervision signals directly from a frozen large language model's next-token prediction objective, bypassing the need for manually constructed positive and negative document pairs. Unlike widely used dense retrievers such as DPR, ANCE, and ColBERT, DREAM avoids expensive data mining pipelines that typically require billions of training examples. The approach works by injecting retriever scores into selected attention heads of a frozen LLM, tightly coupling the retriever with the model's attention mechanism. Tested across backbone sizes of 0.5B, 1B, and 3B parameters, DREAM consistently outperformed existing baselines on the BEIR and RTEB benchmarks. If validated at scale, the method could eliminate the contrastive data construction stage from standard retrieval-augmented generation pipelines.
This is an AI-generated summary. ShortSingh links to the original source for the complete article.

Discussion (0)
Log in to join the discussion and vote.
Log in