Liang Wang, Nan Yang, Xiaolong Huang, Linjun Yang, Rangan Majumder, Furu Wei, arXiv 2024 This model has 24 layers and the embedding size is 1024. Below is an example to encode queries and passages from the MS-MARCO passage ranking dataset. This model is initialized from xlm-roberta-large and continually trained on a mixture of multilingual datasets. It supports 100 languages from xlm-roberta, but low-resource languages may see performance degradation. For all labeled datasets, we only use its training set for fine-tuning. For other training details, please refer to our paper at https://arxiv.org/pdf/2402.05672. Check out unilm/e5 to reproduce evaluation results on the BEIR and MTEB…
Open-weight model · Feature extraction
bge-reranker-large
by Beijing Academy of Artificial Intelligence BAAI/bge-reranker-large
We have updated the new reranker, supporting larger lengths, more languages, and achieving better performance. More details please refer to our Github: FlagEmbedding.
Runs On
What it takes to serve bge-reranker-large (560M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.
| Precision | Weights | Memory needed | Cheapest setup | Per hour | Also fits |
|---|---|---|---|---|---|
| 16-bit | 1.1 GB | 1.3 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
| 8-bit | 0.6 GB | 0.7 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
| 4-bit | 0.3 GB | 0.3 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Sep 18, 2026.
SAVRN's Notes on bge-reranker-large
Nobody gives a reranker its own accelerator, and at 1.3 GB in 16-bit bge-reranker-large shows why. This 560M-parameter XLM-RoBERTa sequence classifier from BAAI ranks retrieved passages for a retrieval-augmented pipeline, so we put it on the same card as that stack. On the cheapest listing the Index shows, a single MI300X with 192 GB at $1.85 an hour on-demand, it occupies under one percent of the memory.
MIT terms apply, so commercial use, modification and redistribution are allowed as long as the copyright and permission notices go with the files. The number to check is 514, the context length in tokens; whatever you hand it per call must fit that window, which sets chunk sizes upstream. The publisher's own page says it released newer rerankers with larger inputs and more languages on 3/18/2024; weigh that against this model's 2024-05-11 last update before standardizing.
Model Card
By Beijing Academy of Artificial Intelligence, published under mit, revision 55611d7bca2a.
We have updated the new reranker, supporting larger lengths, more languages, and achieving better performance.
FlagEmbedding
Model List | FAQ | Usage | Evaluation | Train | Citation | License
More details please refer to our Github: FlagEmbedding.
English | 中文
FlagEmbedding focuses on retrieval-augmented LLMs, consisting of the following projects currently:
- Long-Context LLM: Activation Beacon
- Fine-tuning of LM : LM-Cocktail
- Embedding Model: Visualized-BGE, BGE-M3, LLM Embedder, BGE Embedding
- Reranker Model: llm rerankers, BGE Reranker
- Benchmark: C-MTEB
News
Configuration
- Architecture
- XLMRobertaForSequenceClassification
- Context length (tokens)
- 514
- Layers
- 24
- Hidden size
- 1,024
- Feed-forward size
- 4,096
- Attention heads
- 16
- Vocabulary size
- 250,002
- Stored precision
- float32
- Model type
- xlm-roberta
Identity and Version
- Repository
- BAAI/bge-reranker-large
- Publisher
- Beijing Academy of Artificial Intelligence
- Task
- Feature extraction
- Modality
- Text
- Library
- transformers
- Parameters
- 560M parameters
- Languages
- en, zh
- Revision
- 55611d7bca2a7133960a6d3b71e083071bbfc312
- First published
- 2023-09-12
- Last updated
- 2024-05-11
Files and Weights
11 files, 6.7 GB in total. The weights are 3 files totalling 4.5 GB in bin, onnx, safetensors.
Every file
| File | Type | Size | SHA-256 |
|---|---|---|---|
| model.safetensors | Weights | 2.2 GB | c5ae4e262c60 |
| onnx/model.onnx | Weights | 618.5 KB | 0528834a83cb |
| pytorch_model.bin | Weights | 2.2 GB | 62129e841464 |
| config.json | Configuration | 801 B | — |
| special_tokens_map.json | Configuration | 279 B | — |
| README.md | Documentation | 34.0 KB | — |
| onnx/model.onnx_data | Other | 2.2 GB | e88670fb9065 |
| sentencepiece.bpe.model | Other | 5.1 MB | cfc8146abe2a |
| .gitattributes | Repository | 1.6 KB | — |
| tokenizer.json | Tokenizer | 17.1 MB | 9eb652ac4e40 |
| tokenizer_config.json | Tokenizer | 443 B | — |
License and Download
- License
- mit
- Access
- Open weights, no gate
- Download size
- 4.5 GB
Released by Beijing Academy of Artificial Intelligence through its official repository on Hugging Face. Read the license.
Built From
- Described by arXiv:2309.07597
- Described by arXiv:2310.07554
- Described by arXiv:2311.13534
- Described by arXiv:2312.15503
- Described by arXiv:2401.03462
Evaluations
Each result is shown as reported, with the conditions its reporter stated. None is a SAVRN measurement. A comparison lines two results up only when their configuration, unit and setup are all stated and identical.
| Benchmark | Conditions | Result | Reported by | Revision | Date |
|---|---|---|---|---|---|
| MTEB CMedQAv1 | Configuration defaultTask RerankingMetric mapComparison conditions not established | 81.2721 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB CMedQAv1 | Configuration defaultTask RerankingMetric mrrComparison conditions not established | 84.1424 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB CMedQAv2 | Configuration defaultTask RerankingMetric mapComparison conditions not established | 84.1037 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB CMedQAv2 | Configuration defaultTask RerankingMetric mrrComparison conditions not established | 86.7938 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB MMarcoReranking | Configuration defaultTask RerankingMetric mapComparison conditions not established | 35.4601 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB MMarcoReranking | Configuration defaultTask RerankingMetric mrrComparison conditions not established | 34.6024 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB T2Reranking | Configuration defaultTask RerankingMetric mapComparison conditions not established | 67.2773 | BAAI Publisher reported |
Evaluated revision not stated | — |
| MTEB T2Reranking | Configuration defaultTask RerankingMetric mrrComparison conditions not established | 77.1315 | BAAI Publisher reported |
Evaluated revision not stated | — |
Memory Requirements
| Precision | Weights in memory |
|---|---|
| As published | 4.5 GB |
| 16-bit | 1.1 GB |
| 8-bit | 0.6 GB |
| 4-bit | 0.3 GB |
Weights only, from the published parameter count; the key-value cache and runtime add to this.
Compare bge-reranker-large
Questions About bge-reranker-large
How much GPU memory does bge-reranker-large need?
About 1.3 GB at 16-bit and 0.3 GB at 4-bit: the weights (560M parameters) plus a working margin. A long context needs more.
What is the cheapest GPU to run bge-reranker-large on?
At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.
Can I use bge-reranker-large commercially?
Yes. bge-reranker-large is released under MIT License. The MIT License is a short permissive license. It permits commercial use, modification and redistribution, provided the copyright notice and permission notice are included.
What is bge-reranker-large's context length?
514 tokens, from the maximum position embeddings in its published configuration.
Similar Models
Liang Wang, Nan Yang, Xiaolong Huang, Linjun Yang, Rangan Majumder, Furu Wei, arXiv 2024 This model has 24 layers and the embedding size is 1024. Below are examples to encode queries and passages from the MS-MARCO passage ranking dataset. Usage with Infinity: This model is initialized from xlm-roberta-large and continually trained on a mixture of multilingual datasets. It supports 100 languages from xlm-roberta, but low-resource languages may see performance degradation. First stage: contrastive pre-training with 1 billion weakly supervised text pairs. Check out unilm/e5 to reproduce evaluation results on the BEIR and MTEB benchmark. 1. Do I need to add instructions to the query? Yes, this…
jina-embeddings-v3 is a multilingual multi-task text embedding model designed for a variety of NLP applications. Based on the Jina-XLM-RoBERTa architecture, this model supports Rotary Position Embeddings to handle long input sequences up to 8192 tokens. Additionally, it features 5 LoRA adapters to generate task-specific embeddings efficiently. - retrieval.query: Used for query embeddings in asymmetric retrieval tasks - retrieval.passage: Used for passage embeddings in asymmetric retrieval tasks - separation: Used for embeddings in clustering and re-ranking applications - classification: Used for embeddings in classification tasks - text-matching: Used for embeddings in tasks that quantify…
We are open-sourcing our Conformer-based W2v-BERT 2.0 speech encoder as described in Section 3.2.1 of the paper, which is at the core of our Seamless models. This model was pre-trained on 4.5M hours of unlabeled audio data covering more than 143 languages. It requires finetuning to be used for downstream tasks such as Automatic Speech Recognition (ASR), or Audio Classification. This model and its training are supported by Transformers, more on it in the docs. This is a bare checkpoint without any modeling head, and thus requires finetuning to be used for downstream tasks such as ASR. You can however use it to extract audio embeddings from the top layer with this code snippet: To learn more…
The Qwen3 Embedding model series is the latest proprietary model of the Qwen family, specifically designed for text embedding and ranking tasks. Building upon the dense foundational models of the Qwen3 series, it provides a comprehensive range of text embeddings and reranking models in various sizes (0.6B, 4B, and 8B). This series inherits the exceptional multilingual capabilities, long-text understanding, and reasoning skills of its foundational model. The Qwen3 Embedding series represents significant advancements in multiple text embedding and ranking tasks, including text retrieval, code retrieval, text classification, text clustering, and bitext mining. Exceptional Versatility: The…
For more details please refer to our Github: FlagEmbedding. If you are looking for a model that supports more languages, longer texts, and other retrieval methods, you can try using bge-m3. FlagEmbedding focuses on retrieval-augmented LLMs, consisting of the following projects currently: - 1/30/2024: Release BGE-M3, a new member to BGE model series! M3 stands for Multi-linguality (100+ languages), Multi-granularities (input length up to 8192), Multi-Functionality (unification of dense, lexical, multi-vec/colbert retrieval). It is the first embedding model that supports all three retrieval methods, achieving new SOTA on multi-lingual (MIRACL) and cross-lingual (MKQA) benchmarks. Technical…