Model reference · open weights
bge-multilingual-gemma2 is an open-weight embedding model from BAAI, listed in the AxForge catalogue. AxForge can bring it up on EU-owned hardware for you on request — with the licence handled where one is required.
About
For more details please refer to our Github: FlagEmbedding. BGE-Multilingual-Gemma2 is a LLM-based multilingual embedding model. It is trained on a diverse range of languages and tasks based on google/gemma-2-9b. BGE-Multilingual-Gemma2 primarily demonstrates the following advancements: - Diverse training data: The model's training data spans a broad range of languages, including English, Chinese, Japanese, Korean, French, and more.Additionally, the data covers a variety of task types, such as retrieval, classification, and clustering. - Outstanding performance: The model exhibits state-of-the-art (SOTA) results on multilingual benchmarks like MIRACL, MTEB-pl, and MTEB-fr. It also achieves excellent performance on other major evaluations, including MTEB, C-MTEB and AIR-Bench. 📑 Open-source Plan - [x] Checkpoint - [x] Training Data The training data of BGE-Multilingual-Gemma2 is available at this link. Usage Using FlagEmbedding By default, FlagLLMModel will use all available GPUs when encoding. Please set os.environ["CUDAVISIBLEDEVICES"] to select specific GPUs. You also can set os.environ["CUDAVISIBLEDEVICES"]="" to make all GPUs unavailable. Using Sentence Transformers Using HuggingFace Transformers Evaluation bge-multilingual-gemma2 exhibits state-of-the-art (SOTA) results on benchmarks like MIRACL, MTEB-pl, and MTEB-fr. It also achieves excellent performance on other major evaluations, including MTEB, C-MTEB and AIR-Bench. - MIRACL nDCG@10: Recall@100: - MTEB-fr/pl - MTEB - BEIR - C-MTEB - AIR-Bench Long-Doc (en, Recall@10): QA (en&zh, nDCG@10): Model List bge is short for BAAI general embedding. Citation If you find this repository useful, please consider giving a star :star: and citation
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | BAAI |
|---|---|
| Type | Embedding models |
| Parameters (lead) | 9.2B |
| Context | 8k tokens |
| Variants | 1 |
| Runs with | sentence-transformers |
| Released | 2024-07-25 |
| Popularity | 333k downloads / month |
| Likes | 203 |
| Licence | Open, with conditions |
How it works
Variants
Open weights ship in several sizes and precisions. One page, all the variants — pick the one that fits your GPU. VRAM figures are estimates from model size.
| Variant | Params | Precision | VRAM | Fits 16 GB | Weights |
|---|---|---|---|---|---|
| bge-multilingual-gemma2 | 9.2B | BF16 | ~21.3 GB | ✓ | Weights ↗ |
Benchmarks
As published on the model card — the maker's own numbers, not measured by AxForge.
| Task | Dataset | Metric | Score |
|---|---|---|---|
| Retrieval | MTEB NFCorpus | main_score | 38.114 |
| Retrieval | MTEB NFCorpus | ndcg_at_1 | 48.452 |
| Retrieval | MTEB NFCorpus | ndcg_at_3 | 44.451 |
| Retrieval | MTEB NFCorpus | ndcg_at_5 | 41.139 |
| Retrieval | MTEB NFCorpus | ndcg_at_10 | 38.114 |
| Retrieval | MTEB NFCorpus | ndcg_at_100 | 35.361 |
| Retrieval | MTEB NFCorpus | ndcg_at_1000 | 44.011 |
| Retrieval | MTEB NFCorpus | map_at_1 | 5.638 |
| Retrieval | MTEB NFCorpus | map_at_3 | 10.376 |
| Retrieval | MTEB NFCorpus | map_at_5 | 11.821 |
| Retrieval | MTEB NFCorpus | map_at_10 | 14.087 |
| Retrieval | MTEB NFCorpus | map_at_100 | 18.254 |
| Retrieval | MTEB NFCorpus | map_at_1000 | 19.868 |
| Retrieval | MTEB NFCorpus | precision_at_1 | 49.845 |
| Retrieval | MTEB NFCorpus | precision_at_3 | 41.899 |
| Retrieval | MTEB NFCorpus | precision_at_5 | 35.356 |
| Retrieval | MTEB NFCorpus | precision_at_10 | 28.514 |
| Retrieval | MTEB NFCorpus | precision_at_100 | 9.337 |
| Retrieval | MTEB NFCorpus | precision_at_1000 | 2.21 |
| Retrieval | MTEB NFCorpus | recall_at_1 | 5.638 |
| Retrieval | MTEB NFCorpus | recall_at_3 | 11.938 |
| Retrieval | MTEB NFCorpus | recall_at_5 | 14.062 |
| Retrieval | MTEB NFCorpus | recall_at_10 | 18.506 |
| Retrieval | MTEB NFCorpus | recall_at_100 | 35.97 |
Using it via the API
Once AxForge deploys bge-multilingual-gemma2 for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (bge-multilingual-gemma2 below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/embeddings \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"bge-multilingual-gemma2","input":"text to embed"}'
Licence
Open weights under gemma, which carries conditions (e.g. attribution or an acceptable-use clause). Worth a read before production use — we can walk you through it. Read the licence ↗
Explore