E5-Mistral: Improving Text Embeddings with Large Language Models (Microsoft, Dec 2023)
Source paper: Improving Text Embeddings with Large Language Models (arXiv:2401.00368)
Overview
This work shows that high-quality text embedding models can be trained entirely on synthetic data generated by a large language model, eliminating the need for human-annotated pairs. The resulting model, E5-Mistral-7B-Instruct, achieves state-of-the-art zero-shot performance on BEIR and strong results on MTEB.
Key Contributions
- Synthetic data generation: GPT-4 is prompted to generate diverse training data covering two main task families:
- *Short-text matching* (query ↔ passage retrieval, e.g., for search and RAG)
- *Long-text matching* (summarization, sentiment classification, etc.)
- Multi-lingual coverage: Data is generated in 25 languages with varied task formats, instruction prompts, and GPT-4 temperature settings for diversity.
- Quality filtering: An LLM-based filtering pipeline verifies relevance of generated pairs, removing noisy or trivial examples.
- Two-step training: 1. *Weakly supervised pre-training* on the synthetic mix with contrastive learning (InfoNCE, in-batch negatives). 2. *Supervised fine-tuning* with hard negatives mined via retrieval (mixing BM25 and dense-retrieved negatives).
- Backbone: Mistral-7B-Instruct, fine-tuned with LoRA and a last-token pooling strategy over instruction-augmented inputs.
- BEIR (zero-shot): State-of-the-art nDCG@10 at time of release, outperforming prior strong baselines such as E5-large-v2 and generic LLM-embedding approaches.
- MTEB: Strong average scores across retrieval, clustering, classification, and reranking tasks.
- BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity
- Arctic-Embed 2.0: Multilingual Retrieval Without Compromise
- BGE-en-ICL / BGE-ICL: Making Text Embedders Few-Shot Learners
Results
Takeaways
1. LLM-generated data can replace expensive human annotation for embedding training. 2. Two-stage training (synthetic pre-training → supervised with hard negatives) is a simple but effective recipe. 3. Instruction-aware embeddings enable one model to serve multiple task types (retrieval, STS, classification).
> Note: Exact benchmark numbers should be verified against the original PDF.