Boosted Dense Retriever

By Patrick Lewis, Barlas Oğuz, Wenhan Xiong, Fabio Petroni, Wen-tau Yih, Sebastian Riedel. 2021

18 citations recorded by the source corpus.

We propose DrBoost, a dense retrieval ensemble inspired by boosting. DrBoost is trained in stages: each component model is learned sequentially and specialized by focusing only on retrieval mistakes made by the current ensemble. The final representation is the concatenation of the output vectors of all the component models, making it a drop-in replacement for standard dense retrievers at test time. DrBoost enjoys several advantages compared to standard dense retrieval models. It produces representations which are 4x more compact, while delivering comparable retrieval results. It also performs surprisingly well under approximate search with coarse quantization, reducing latency and bandwidth needs by another 4x. In practice, this can make the difference between serving indices from disk versus from memory, paving the way for much cheaper deployments.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Information Retrieval & Search

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Ji-rong Wen
  2. Wayne Xin Zhao
  3. Sebastian Riedel
  4. Sebastian Hofstätter
  5. Wen-tau Yih
  6. Jing Liu
  7. Patrick Lewis
  8. Ruiyang Ren

In Computer Vision and Pattern Recognition

  1. Sebastian Riedel
  2. Wen-tau Yih
  3. Patrick Lewis
  4. Ji-rong Wen
  5. Wayne Xin Zhao
  6. Fabio Petroni
  7. Jing Liu
  8. Ruiyang Ren

In Information Retrieval & Search

  1. Sebastian Hofstätter
  2. Jing Liu
  3. Ruiyang Ren
  4. Sebastian Riedel
  5. Ji-rong Wen
  6. Wayne Xin Zhao
  7. Wen-tau Yih
  8. Patrick Lewis

Citations in the corpus