Pyserini: A Python Toolkit for Reproducible Information Retrieval Research with Sparse and Dense Representations

By Jimmy J. Lin, Xueguang Ma, Sheng-Chieh Lin, Jheng-Hong Yang, Ronak Pradeep, Rodrigo Nogueira, David R. Cheriton. 2021

673 citations recorded by the source corpus.

Pyserini is a Python toolkit for reproducible information retrieval research with sparse and dense representations. It aims to provide effective, reproducible, and easy-to-use first-stage retrieval in a multi-stage ranking architecture. Our toolkit is self-contained as a standard Python package and comes with queries, relevance judgments, pre-built indexes, and evaluation scripts for many commonly used IR test collections. We aim to support, out of the box, the entire research lifecycle of efforts aimed at improving ranking with modern neural approaches. In particular, Pyserini supports sparse retrieval (e.g., BM25 scoring using bag-of-words representations), dense retrieval (e.g., nearest-neighbor search on transformer-encoded representations), as well as hybrid retrieval that integrates both approaches. This paper provides an overview of toolkit features and presents empirical results that illustrate its effectiveness on two popular ranking tasks. Around this toolkit, our group has built a culture of reproducibility through shared norms and tools that enable rigorous automated testing.

Subfields: Artificial Intelligence, Materials Chemistry, 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. Luke S. Zettlemoyer
  2. Hannaneh Hajishirzi
  3. Jimmy J. Lin
  4. Sebastian Riedel
  5. Mohit Bansal
  6. Wen-tau Yih
  7. Jianfeng Gao
  8. Graham Neubig

In Materials Chemistry

  1. Jimmy J. Lin
  2. Noah A. Smith
  3. Joelle Pineau
  4. Luke S. Zettlemoyer
  5. Armand Joulin
  6. Wayne Xin Zhao
  7. Edouard Grave
  8. Michael Lewis

In Information Retrieval & Search

  1. Jimmy J. Lin
  2. Wen-tau Yih
  3. Sebastian Riedel
  4. Sebastian Hofstätter
  5. Jamie Callan
  6. Patrick Lewis
  7. Jing Liu
  8. Ruiyang Ren

Citations in the corpus