Model reference · open weights
bge-en 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. If you are looking for a model that supports more languages, longer texts, and other retrieval methods, you can try using bge-m3. English | 中文 FlagEmbedding focuses on retrieval-augmented LLMs, consisting of the following projects currently: - Long-Context LLM: Activation Beacon - Fine-tuning of LM : LM-Cocktail - Dense Retrieval: BGE-M3, LLM Embedder, BGE Embedding - Reranker Model: BGE Reranker - Benchmark: C-MTEB News - 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 which supports all three retrieval methods, achieving new SOTA on multi-lingual (MIRACL) and cross-lingual (MKQA) benchmarks. Technical Report and Code. :fire: - 1/9/2024: Release Activation-Beacon, an effective, efficient, compatible, and low-cost (training) method to extend the context length of LLM. Technical Report :fire: - 12/24/2023: Release LLaRA, a LLaMA-7B based dense retriever, leading to state-of-the-art performances on MS MARCO and BEIR. Model and code will be open-sourced. Please stay tuned. Technical Report :fire: - 11/23/2023: Release LM-Cocktail, a method to maintain general capabilities during fine-tuning by merging multiple language models. Technical Report :fire: - 10/12/2023: Release LLM-Embedder, a unified embedding model to support diverse retrieval augmentation needs for LLMs. Technical Report - 09/15/2023: The technical report and massive training data of BGE has been released - 09/12/2023: New models: - New reranker model: release cross-encoder models BAAI/bge-reranker-base and BAAI/bge-reranker-large, which are more powerful than embedding model. We recommend to use/fine-tune them to re-rank top-k documents returned by embedding models. - update embedding model: release bge--v1.5 embedding model to alleviate the issue of the similarity distribution, and enhance its retrieval ability without instruction. - 09/07/2023: Update fine-tune code: Add script to mine hard negatives and support adding i
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | BAAI |
|---|---|
| Type | Embedding models |
| Parameters (lead) | 109M |
| Context | 512 tokens |
| Variants | 1 |
| Runs with | sentence-transformers |
| Released | 2023-09-11 |
| Popularity | 12M downloads / month |
| Likes | 466 |
| Licence | Open weights |
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-base-en-v1.5 | 109M | BF16 | ~0.3 GB | ✓ | Weights ↗ |
Benchmarks
As published on the model card — the maker's own numbers, not measured by AxForge.
| Task | Dataset | Metric | Score |
|---|---|---|---|
| Classification | MTEB AmazonCounterfactualClassification (en) | accuracy | 76.149 |
| Classification | MTEB AmazonCounterfactualClassification (en) | ap | 39.323 |
| Classification | MTEB AmazonCounterfactualClassification (en) | f1 | 70.169 |
| Classification | MTEB AmazonPolarityClassification | accuracy | 93.387 |
| Classification | MTEB AmazonPolarityClassification | ap | 90.213 |
| Classification | MTEB AmazonPolarityClassification | f1 | 93.377 |
| Classification | MTEB AmazonReviewsClassification (en) | accuracy | 48.846 |
| Classification | MTEB AmazonReviewsClassification (en) | f1 | 48.146 |
| Retrieval | MTEB ArguAna | map_at_1 | 40.754 |
| Retrieval | MTEB ArguAna | map_at_10 | 55.761 |
| Retrieval | MTEB ArguAna | map_at_100 | 56.331 |
| Retrieval | MTEB ArguAna | map_at_1000 | 56.334 |
| Retrieval | MTEB ArguAna | map_at_3 | 51.92 |
| Retrieval | MTEB ArguAna | map_at_5 | 54.011 |
| Retrieval | MTEB ArguAna | mrr_at_1 | 41.181 |
| Retrieval | MTEB ArguAna | mrr_at_10 | 55.968 |
| Retrieval | MTEB ArguAna | mrr_at_100 | 56.538 |
| Retrieval | MTEB ArguAna | mrr_at_1000 | 56.542 |
| Retrieval | MTEB ArguAna | mrr_at_3 | 51.98 |
| Retrieval | MTEB ArguAna | mrr_at_5 | 54.209 |
| Retrieval | MTEB ArguAna | ndcg_at_1 | 40.754 |
| Retrieval | MTEB ArguAna | ndcg_at_10 | 63.605 |
| Retrieval | MTEB ArguAna | ndcg_at_100 | 66.052 |
| Retrieval | MTEB ArguAna | ndcg_at_1000 | 66.12 |
Using it via the API
Once AxForge deploys bge-en for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (bge-en 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-en","input":"text to embed"}'
Details
Languages
Tags
Papers
Licence
Open weights under mit — commercial use is permitted. Deploy it on AxForge EU hardware on request. Read the licence ↗
Explore