FlagOpen/FlagEmbedding
Retrieval and Retrieval-augmented LLMs
What it solves
BGE (BAAI General Embedding) provides a comprehensive toolkit for improving the retrieval stage of Retrieval-Augmented Generation (RAG) and search systems. It addresses the need for high-quality text and multimodal embeddings and reranking models to ensure that the most relevant documents are retrieved for an LLM to process.
How it works
The toolkit offers a suite of models and tools for the entire retrieval pipeline:
- Embedding Models: Converts text or images into vectors (embeddings) for efficient similarity search. This includes specialized models like BGE-M3 for multi-lingual, multi-granular, and multi-functional retrieval (dense, lexical, and multi-vector).
- Rerankers: Cross-encoder models that provide more accurate, though slower, scoring of the top-k documents returned by embedding models to refine the retrieval results.
- Multimodal Capabilities: BGE-VL supports visual search applications, including text-to-image and image-to-text retrieval.
- Fine-tuning & Evaluation: Includes tools to fine-tune embedders and rerankers on custom data and evaluate their performance using benchmarks like C-MTEB.
Who it’s for
This project is for developers and researchers building RAG-based applications, search engines, and multimodal retrieval systems who need state-of-the-art embedding and reranking capabilities.
Highlights
- BGE-M3: A versatile model supporting 100+ languages, input lengths up to 8192 tokens, and three retrieval methods (dense, lexical, and multi-vector).
- Multimodal Support: Includes BGE-VL for comprehensive visual search and Visualized-BGE for hybrid image-text data.
- Comprehensive Toolkit: Provides a one-stop shop for inference, fine-tuning, and evaluation of both embedders and rerankers.
- High Performance: Many models in the series rank highly on benchmarks like MTEB and C-MTEB.
Related
- Project
- Project
- Dispatch
- Dispatch
- Project