ronak's picture
Update README.md
aa11d9d verified
metadata
tags:
  - generated_from_trainer
license: mit
language:
  - en
base_model: mistralai/Mistral-7B-v0.1
RankZephyr Logo

Model Card for RankZephyr 7B V1 - Full

RankZephyr is a series of language models trained to act as helpful reranking assistants built on the Zephyr-7B-β model. RankZephyr Base is the model that follows single-stage fine-tuning on the RankGPT-3.5 model, while RankZephyr Full is the model that is further fine-tuned on RankGPT-4 reorderings of OpenAI's Ada2 orderings for 5K queries.

Model description

  • Model type: A 7B parameter GPT-like model initially fine-tuned on a mix of publicly available, synthetic datasets, followed by task-specific listwise reranking data.
  • Language(s) (NLP): Primarily English
  • License: MIT
  • Fine-tuned from model: HuggingFaceH4/zephyr-7b-beta

Model Sources

Effectiveness

At the time of release, RankZephyr-7B-Full is the state-of-the-art open-source reranking model on various datasets like DL19/20/21/22 and TREC-COVID and TREC-News.

With the MS MARCO v1 collection:

Model Size First Stage DL19 DL20
RankZephyr-7b-v1-full-rho 🪁 7B SPLADE++ ED 0.7855 0.8255
RankZephyr-7b-v1-full 🪁 7B SPLADE++ ED 0.7803 0.8211
RankGPT-4 (PSC) - SPLADE++ ED 0.7601 0.7514
RankGPT-4 - SPLADE++ ED 0.7464 0.7076
RankZephyr-7b-v1-base 🪁 7B SPLADE++ ED 0.7341 0.7213
RankGPT-3.5 - SPLADE++ ED 0.7504 0.7120

More details can be found in the paper.

Intended uses & limitations

The model is to be used in conjunction with the RankLLM repository. While rank-llm exists as a PyPI package, we are currently in the early stages of development and encourage users to directly check install from source.

The original Zephyr model is trained for chat. In our case, RankZephyr is fine-tuned to act as a listwise reranking agent. You provide it with a query and documents and get back a reordered list of document identifiers.

Bias, Risks, and Limitations

The following is an excerpt from the Zephyr-7B-β model card:

Zephyr-7B-β has not been aligned to human preferences for safety within the RLHF phase or deployed with in-the-loop filtering of responses like ChatGPT, so the model can produce problematic outputs (especially when prompted to do so). It is also unknown what the size and composition of the corpus was used to train the base model (mistralai/Mistral-7B-v0.1), however it is likely to have included a mix of Web data and technical sources like books and code. See the Falcon 180B model card for an example of this.

Our model is trained specifically on monolingual English data, effectiveness on multilingual sets is not guaranteed.

Citation

If you find RankZephyr is useful in your work, please cite the following paper:

@ARTICLE{pradeep2023rankzephyr,
  title   = {{RankZephyr}: Effective and Robust Zero-Shot Listwise Reranking is a Breeze!},
  author  = {Ronak Pradeep and Sahel Sharifymoghaddam and Jimmy Lin},
  year    = {2023},
  journal = {arXiv:2312.02724}
}