Declarative Experimentation in Information Retrieval using PyTerrier

Code

• github.com/terrier-org/pyterrier
  OfficialIn papernone★ 497
• github.com/terrierteam/pyterrier_doc2query
  pytorch★ 37
• github.com/terrierteam/pyterrier_t5
  pytorch★ 16
• github.com/Xiao0728/ColBERT-PRF-VirtualAppendix
  none★ 8
• github.com/terrierteam/pyterrier_ance
  pytorch★ 5
• github.com/terrierteam/pyterrier_deepct
  tf★ 4
• github.com/cmacdonald/pyt_jpq
  pytorch★ 0
• github.com/cmacdonald/pyterrier_colbert
  pytorch★ 0
• github.com/parry-parry/pyterrier_t5
  pytorch★ 0

Abstract

The advent of deep machine learning platforms such as Tensorflow and Pytorch, developed in expressive high-level languages such as Python, have allowed more expressive representations of deep neural network architectures. We argue that such a powerful formalism is missing in information retrieval (IR), and propose a framework called PyTerrier that allows advanced retrieval pipelines to be expressed, and evaluated, in a declarative manner close to their conceptual design. Like the aforementioned frameworks that compile deep learning experiments into primitive GPU operations, our framework targets IR platforms as backends in order to execute and evaluate retrieval pipelines. Further, we can automatically optimise the retrieval pipelines to increase their efficiency to suite a particular IR platform backend. Our experiments, conducted on TREC Robust and ClueWeb09 test collections, demonstrate the efficiency benefits of these optimisations for retrieval pipelines involving both the Anserini and Terrier IR platforms.

Tasks

Reproductions