|
--- |
|
license: mit |
|
--- |
|
# Cross-Encoder for MS Marco |
|
|
|
This model was trained on the [MS Marco Passage Ranking](https://github.com/microsoft/MSMARCO-Passage-Ranking) task. |
|
|
|
The model can be used for Information Retrieval: Given a query, encode the query will all possible passages (e.g. retrieved with ElasticSearch). Then sort the passages in a decreasing order. See our paper [R2ANKER](https://arxiv.org/pdf/2206.08063.pdf) for more details. |
|
|
|
## Usage with Transformers |
|
|
|
```python |
|
from transformers import AutoTokenizer, AutoModelForSequenceClassification |
|
import torch |
|
tokenizer = AutoTokenizer.from_pretrained("YCZhou/R2ANKER") |
|
model = AutoModelForSequenceClassification.from_pretrained("YCZhou/R2ANKER") |
|
features = tokenizer(['How many people live in Berlin?', 'How many people live in Berlin?'], ['Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.', 'New York City is famous for the Metropolitan Museum of Art.'], padding=True, truncation=True, return_tensors="pt") |
|
model.eval() |
|
with torch.no_grad(): |
|
scores = model(**features).logits |
|
print(scores) |
|
``` |
|
|
|
## Citation |
|
``` |
|
@inproceedings{DBLP:conf/acl/Zhou0GTXLJJ23, |
|
author = {Yucheng Zhou and |
|
Tao Shen and |
|
Xiubo Geng and |
|
Chongyang Tao and |
|
Can Xu and |
|
Guodong Long and |
|
Binxing Jiao and |
|
Daxin Jiang}, |
|
title = {Towards Robust Ranker for Text Retrieval}, |
|
booktitle = {Findings of the Association for Computational Linguistics: {ACL} 2023, |
|
Toronto, Canada, July 9-14, 2023}, |
|
pages = {5387--5401}, |
|
publisher = {Association for Computational Linguistics}, |
|
year = {2023}, |
|
url = {https://doi.org/10.18653/v1/2023.findings-acl.332}, |
|
doi = {10.18653/V1/2023.FINDINGS-ACL.332}, |
|
timestamp = {Sat, 30 Sep 2023 09:33:34 +0200}, |
|
biburl = {https://dblp.org/rec/conf/acl/Zhou0GTXLJJ23.bib}, |
|
bibsource = {dblp computer science bibliography, https://dblp.org} |
|
} |
|
``` |