IMPORTANT: No additional bug fixes or documentation updates will be released for this version. For the latest information, see the current release documentation.

› ›

Semantic re-ranking

edit

Semantic re-ranking

edit

This functionality is in technical preview and may be changed or removed in a future release. Elastic will work to fix any issues, but features in technical preview are not subject to the support SLA of official GA features.

This overview focuses more on the high-level concepts and use cases for semantic re-ranking. For full implementation details on how to set up and use semantic re-ranking in Elasticsearch, see the reference documentation in the Search API docs.

Re-rankers improve the relevance of results from earlier-stage retrieval mechanisms. Semantic re-rankers use machine learning models to reorder search results based on their semantic similarity to a query.

Semantic re-ranking requires relatively large and complex machine learning models and operates in real-time in response to queries. This technique makes sense on a small top-k result set, as one the of the final steps in a pipeline. This is a powerful technique for improving search relevance that works equally well with keyword, semantic, or hybrid retrieval algorithms.

The next sections provide more details on the benefits, use cases, and model types used for semantic re-ranking. The final sections include a practical, high-level overview of how to implement semantic re-ranking in Elasticsearch and links to the full reference documentation.

Use cases

edit

Semantic re-ranking enables a variety of use cases:

Lexical (BM25) retrieval results re-ranking
- Out-of-the-box semantic search by adding a simple API call to any lexical/BM25 retrieval pipeline.
- Adds semantic search capabilities on top of existing indices without reindexing, perfect for quick improvements.
- Ideal for environments with complex existing indices.
Semantic retrieval results re-ranking
- Improves results from semantic retrievers using ELSER sparse vector embeddings or dense vector embeddings by using more powerful models.
General applications
- Supports automatic and transparent chunking, eliminating the need for pre-chunking at index time.
- Provides explicit control over document relevance in retrieval-augmented generation (RAG) uses cases or other scenarios involving language model (LLM) inputs.

Now that we’ve outlined the value of semantic re-ranking, we’ll explore the specific models that power this process and how they differ.

Cross-encoder and bi-encoder models

edit

At a high level, two model types are used for semantic re-ranking: cross-encoders and bi-encoders.

In this version, Elasticsearch only supports cross-encoders for semantic re-ranking.

A cross-encoder model can be thought of as a more powerful, all-in-one solution, because it generates query-aware document representations. It takes the query and document texts as a single, concatenated input.
A bi-encoder model takes as input either document or query text. Documents and query embeddings are computed separately, so they aren’t aware of each other.
- To compute a ranking score, an external operation is required. This typically involves computing dot-product or cosine similarity between the query and document embeddings.

In brief, cross-encoders provide high accuracy but are more resource-intensive. Bi-encoders are faster and more cost-effective but less precise.

In future versions, Elasticsearch will also support bi-encoders. If you’re interested in a more detailed analysis of the practical differences between cross-encoders and bi-encoders, untoggle the next section.

Comparisons between cross-encoder and bi-encoder

The following is a non-exhaustive list of considerations when choosing between cross-encoders and bi-encoders for semantic re-ranking:

Because a cross-encoder model simultaneously processes both query and document texts, it can better infer their relevance, making it more effective as a reranker than a bi-encoder.
Cross-encoder models are generally larger and more computationally intensive, resulting in higher latencies and increased computational costs.
There are significantly fewer open-source cross-encoders, while bi-encoders offer a wide variety of sizes, languages, and other trade-offs.
The effectiveness of cross-encoders can also improve the relevance of semantic retrievers. For example, their ability to take word order into account can improve on dense or sparse embedding retrieval.
When trained in tandem with specific retrievers (like lexical/BM25), cross-encoders can “correct” typical errors made by those retrievers.
Cross-encoders output scores that are consistent across queries. This enables you to maintain high relevance in result sets, by setting a minimum score threshold for all queries. For example, this is important when using results in a RAG workflow or if you’re otherwise feeding results to LLMs. Note that similarity scores from bi-encoders/embedding similarities are query-dependent, meaning you cannot set universal cut-offs.
Bi-encoders rerank using embeddings. You can improve your re-ranking latency by creating embeddings at ingest-time. These embeddings can be stored for re-ranking without being indexed for retrieval, reducing your memory footprint.

Semantic re-ranking in Elasticsearch

edit

In Elasticsearch, semantic re-rankers are implemented using the Elasticsearch Inference API and a retriever.

To use semantic re-ranking in Elasticsearch, you need to:

Choose a re-ranking model. Currently you can:
- Integrate directly with the Cohere Rerank inference endpoint using the rerank task type
- Integrate directly with the Google Vertex AI inference endpoint using the rerank task type
- Upload a model to Elasticsearch from Hugging Face with Eland. You’ll need to use the text_similarity NLP task type when loading the model using Eland. Refer to the Elastic NLP model reference for a list of third party text similarity models supported by Elasticsearch for semantic re-ranking.
  - Then set up an Elasticsearch service inference endpoint with the rerank task type
Create a rerank task using the Elasticsearch Inference API. The Inference API creates an inference endpoint and configures your chosen machine learning model to perform the re-ranking task.
Define a text_similarity_reranker retriever in your search request. The retriever syntax makes it simple to configure both the retrieval and re-ranking of search results in a single API call.

Example search request with semantic reranker

The following example shows a search request that uses a semantic reranker to reorder the top-k documents based on their semantic similarity to the query.

resp = client.search(
    retriever={
        "text_similarity_reranker": {
            "retriever": {
                "standard": {
                    "query": {
                        "match": {
                            "text": "How often does the moon hide the sun?"
                        }
                    }
                }
            },
            "field": "text",
            "inference_id": "my-cohere-rerank-model",
            "inference_text": "How often does the moon hide the sun?",
            "rank_window_size": 100,
            "min_score": 0.5
        }
    },
)
print(resp)

POST _search
{
  "retriever": {
    "text_similarity_reranker": {
      "retriever": {
        "standard": {
          "query": {
            "match": {
              "text": "How often does the moon hide the sun?"
            }
          }
        }
      },
      "field": "text",
      "inference_id": "my-cohere-rerank-model",
      "inference_text": "How often does the moon hide the sun?",
      "rank_window_size": 100,
      "min_score": 0.5
    }
  }
}

Learn more

edit

Read the retriever reference documentation for syntax and implementation details
Learn more about the retrievers abstraction
Learn more about the Elastic Inference APIs
Check out our Python notebook for using Cohere with Elasticsearch

« Re-ranking Learning To Rank »