--- 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 that are 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](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta) ### Model Sources - **Repository:** https://github.com/castorini/rank_llm - **Paper:** https://arxiv.org/abs/2312.02724 ## 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| ## Intended uses & limitationspacka The following is an excerpt from the [Zephyr-7B-β model card](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta/blob/main/README.md#intended-use--limitations): 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](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta/blob/main/README.md#bias-risks--limitations): > 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](https://huggingface.co/tiiuae/falcon-180B#training-data) 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} } ```