Model reference · open weights
bge-small-en is an open-weight embedding model from unsloth, 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
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 of BGE has been released - 09/15/2023: The 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
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | unsloth |
|---|---|
| Type | Embedding models |
| Parameters (lead) | 33M |
| Context | 512 tokens |
| Variants | 2 |
| Runs with | sentence-transformers |
| Released | 2026-06-01 |
| Popularity | 349k downloads / month |
| Likes | 2 |
| 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.
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 | 73.791 |
| Classification | MTEB AmazonCounterfactualClassification (en) | ap | 37.219 |
| Classification | MTEB AmazonCounterfactualClassification (en) | f1 | 68.091 |
| Classification | MTEB AmazonPolarityClassification | accuracy | 92.754 |
| Classification | MTEB AmazonPolarityClassification | ap | 89.468 |
| Classification | MTEB AmazonPolarityClassification | f1 | 92.739 |
| Classification | MTEB AmazonReviewsClassification (en) | accuracy | 46.986 |
| Classification | MTEB AmazonReviewsClassification (en) | f1 | 46.559 |
| Retrieval | MTEB ArguAna | map_at_1 | 35.846 |
| Retrieval | MTEB ArguAna | map_at_10 | 51.388 |
| Retrieval | MTEB ArguAna | map_at_100 | 52.133 |
| Retrieval | MTEB ArguAna | map_at_1000 | 52.141 |
| Retrieval | MTEB ArguAna | map_at_3 | 47.037 |
| Retrieval | MTEB ArguAna | map_at_5 | 49.579 |
| Retrieval | MTEB ArguAna | mrr_at_1 | 36.558 |
| Retrieval | MTEB ArguAna | mrr_at_10 | 51.658 |
| Retrieval | MTEB ArguAna | mrr_at_100 | 52.402 |
| Retrieval | MTEB ArguAna | mrr_at_1000 | 52.41 |
| Retrieval | MTEB ArguAna | mrr_at_3 | 47.345 |
| Retrieval | MTEB ArguAna | mrr_at_5 | 49.798 |
| Retrieval | MTEB ArguAna | ndcg_at_1 | 35.846 |
| Retrieval | MTEB ArguAna | ndcg_at_10 | 59.55 |
| Retrieval | MTEB ArguAna | ndcg_at_100 | 62.596 |
| Retrieval | MTEB ArguAna | ndcg_at_1000 | 62.759 |
Using it via the API
Once AxForge deploys unsloth-bge-small-en for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (unsloth-bge-small-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":"unsloth-bge-small-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