Corpus ID: 215828505

Learning-to-Rank with BERT in TF-Ranking

@article{Han2020LearningtoRankWB,
  title={Learning-to-Rank with BERT in TF-Ranking},
  author={Shuguang Han and Xuanhui Wang and Michael Bendersky and Marc Najork},
  journal={ArXiv},
  year={2020},
  volume={abs/2004.08476}
}
This paper describes a machine learning algorithm for document (re)ranking, in which queries and documents are firstly encoded using BERT [1], and on top of that a learning-to-rank (LTR) model constructed with TF-Ranking (TFR) [2] is applied to further optimize the ranking performance. This approach is proved to be effective in a public MS MARCO benchmark [3]. Our first two submissions achieve the best performance for the passage re-ranking task [4], and the second best performance for the… Expand

Figures and Tables from this paper

Generalizing Discriminative Retrieval Models using Generative Tasks
Learning Representations for Axis-Aligned Decision Forests through Input Perturbation
...
1
2
...

References

SHOWING 1-10 OF 21 REFERENCES
Multi-Stage Document Ranking with BERT
TF-Ranking: Scalable TensorFlow Library for Learning-to-Rank
Learning to rank for information retrieval
Passage Re-ranking with BERT
Context-Aware Sentence/Passage Term Importance Estimation For First Stage Retrieval
MS MARCO: A Human Generated MAchine Reading COmprehension Dataset
...
1
2
3
...