Secure Systems Group
Menu×

Wiki Home

Dense retrieval

Definition

Dense retrieval finds relevant documents by mapping text into a continuous vector space rather than matching words. A learned encoder turns each passage in a corpus into an embedding vector; at query time another encoder (sometimes the same one) embeds the query, and relevance is the similarity of the two vectors, usually cosine similarity or dot product. The retriever returns the top-k passages with the highest similarity scores. This is the neural alternative to sparse lexical retrieval such as BM25, which matches on shared terms; dense retrieval can match paraphrases that share no keywords. Dense Passage Retrieval (Karpukhin et al., 2020) is the canonical formulation, training separate query and passage encoders so that a question and its answer-bearing passage land close together.

In a retrieval-augmented system the dense retriever is the component that selects which corpus text reaches the model’s context window. Two consequences follow for security. Whether a piece of text is retrieved for a given query is decided entirely by embedding similarity, so an attacker who controls a passage’s text controls its position in that vector space. And when the encoder parameters are public, that similarity is a differentiable function an attacker can optimize against directly.

Papers that use this concept

See also

References

  • Karpukhin, V., Oguz, B., Min, S., Lewis, P., Wu, L., Edunov, S., Chen, D., and Yih, W.-t. “Dense Passage Retrieval for Open-Domain Question Answering.” EMNLP, 2020.