Overview
BGE Reranker Base from BAAI is a cross-encoder model that reranks a list of retrieved passages against a query, consistently improving nDCG@10 by 3-8 points over first-stage embedding retrieval alone. It is the missing second stage in most RAG pipelines: your vector search narrows candidates to 20-100 passages, the reranker scores each (query, passage) pair directly and returns a precise ranked order.
Cross-encoders see both query and passage together in a single forward pass -- unlike bi-encoders that embed each independently -- which lets them capture fine-grained relevance signals that embedding similarity misses. The trade-off is higher per-passage latency, which is why rerankers are applied to a small candidate set, not the full corpus.
Deploy as a SageMaker endpoint. The endpoint accepts {"query": "...", "passages": ["...", "..."]} and returns a relevance score per passage. Rerank your top-20 vector search results and pass only the top-3 to the LLM -- better answers, lower token cost.
Primary use cases: RAG pipeline precision improvement, enterprise document search, e-commerce product ranking, legal case retrieval, and any two-stage retrieval architecture where first-stage recall is high but precision needs improvement.
Highlights
- Cross-encoder architecture sees query and passage together in one forward pass -- captures relevance signals that bi-encoder embedding similarity misses
- Two-stage retrieval: vector search narrows to 20-100 candidates, reranker scores each (query, passage) pair and returns a precise ranked order
- Flat $0.08/hr with no per-query charges -- integrate as middleware between your vector DB and LLM to cut token costs and improve answer quality
Details
Introducing multi-product solutions
You can now purchase comprehensive solutions tailored to use cases and industries.
Features and programs
Financing for AWS Marketplace purchases
Pricing
Dimension | Description | Cost/host/hour |
|---|---|---|
ml.m5.xlarge Inference (Real-Time) Recommended | Model inference on the ml.m5.xlarge instance type, real-time mode | $0.10 |
ml.m5.xlarge Inference (Batch) Recommended | Model inference on the ml.m5.xlarge instance type, batch mode | $0.10 |
Vendor refund policy
No refunds.
How can we make this page better?
Legal
Vendor terms and conditions
Content disclaimer
Delivery details
Amazon SageMaker model
An Amazon SageMaker model package is a pre-trained machine learning model ready to use without additional training. Use the model package to create a model on Amazon SageMaker for real-time inference or batch processing. Amazon SageMaker is a fully managed platform for building, training, and deploying machine learning models at scale.
Version release notes
Initial release
Additional details
Inputs
- Summary
BAAI/bge-reranker-base cross-encoder on SageMaker. Second-stage reranking lifts RAG answer precision by 3-8 nDCG@10 points over embedding-only retrieval. Flat $0.08/hr, no per-query charges.
- Input MIME type
- application/json
Support
Vendor support
Contact support@waltsoft.net for deployment assistance.
AWS infrastructure support
AWS Support is a one-on-one, fast-response support channel that is staffed 24x7x365 with experienced and technical support engineers. The service helps customers of all sizes and technical abilities to successfully utilize the products and features provided by Amazon Web Services.
Similar products
