MarkedBERT: Integrating Traditional IR Cues in Pre-trained Language Models for Passage Retrieval

By Lila Boualili, Jose G. Moreno, Mohand Boughanem. 2020

28 citations recorded by the source corpus.

The Information Retrieval (IR) community has witnessed a flourishing development of deep neural networks, however, only a few managed to beat strong baselines. Among them, models like DRMM and DUET were able to achieve better results thanks to the proper handling of exact match signals. Nowadays, the application of pre-trained language models to IR tasks has achieved impressive results exceeding all previous work. In this paper, we assume that established IR cues like exact term-matching, proven to be valuable for deep neural models, can be used to augment the direct supervision from labeled data for training these pre-trained models. To study the effectiveness of this assumption, we propose MarkedBERT a modified version of one of the most popular pre-trained models via language modeling tasks, BERT. MarkedBERT integrates exact match signals using a marking technique that locates and highlights Exact Matched query-document terms using marker tokens. Experiments on MS MARCO Passage Ranking task show that our rather simple approach is actually effective. We find that augmenting the input with marker tokens allows the model to focus on valuable text sequences for IR.

Subfields: Artificial Intelligence, Information Systems, Information Retrieval & Search, Large Language Models

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. Jimmy J. Lin
  2. Andrew Yates
  3. Rodrigo Nogueira
  4. Zeynep Akkalyoncu Yilmaz
  5. Sebastian Hofstätter
  6. Allan Hanbury
  7. Zhuyun Dai
  8. Jamie Callan

In Information Systems

  1. Rodrigo Nogueira
  2. Andrew Yates
  3. Jimmy J. Lin
  4. Sebastian Hofstätter
  5. Allan Hanbury
  6. Sean MacAvaney
  7. Jamie Callan
  8. Nazli Goharian

In Information Retrieval & Search

  1. Jimmy J. Lin
  2. Rodrigo Nogueira
  3. Andrew Yates
  4. Zeynep Akkalyoncu Yilmaz
  5. Sebastian Hofstätter
  6. Allan Hanbury
  7. Zhuyun Dai
  8. Wei Yang

In Large Language Models

  1. Jimmy J. Lin
  2. Andrew Yates
  3. Rodrigo Nogueira
  4. Zhuyun Dai
  5. Jamie Callan
  6. Zeynep Akkalyoncu Yilmaz
  7. Sebastian Hofstätter
  8. Allan Hanbury

Citations in the corpus