混合搜索与重新排序
结合关键词搜索和语义搜索(混合搜索),并使用重新排序模型优先处理最相关的文档
混合搜索与重新排序 是 CoddyKit 上的免费 LangChain / RAG / Vector DBs 课时。 这是第 3 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 LangChain / RAG / Vector DBs 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 LangChain / RAG / Vector DBs 课程共包含 4 节课。
本课时的部分内容尚未翻译,以英文显示。
Beyond Basic Retrieval
When building advanced Retrieval Augmented Generation (RAG) systems, simply finding documents isn't enough. We need to find the most relevant documents efficiently.
Traditional keyword or semantic searches, while powerful, each have limitations. To overcome these, we can combine their strengths.
Keyword Search: Specificity
Keyword search (also known as sparse retrieval, e.g., using BM25 or TF-IDF) is excellent for finding exact matches and specific terms.
- Strengths: Great for precise queries, proper nouns, and when you know the exact wording.
- Weaknesses: Struggles with synonyms, different phrasing, or understanding conceptual meaning.
For example, searching 'Python list append' works well, but 'add element to Python array' might miss results.
Semantic Search: Understanding Meaning
Semantic search (dense retrieval, using embeddings) understands the meaning and context of your query and documents.
- Strengths: Handles synonyms, rephrased questions, and conceptual searches effectively.
- Weaknesses: Can struggle with very specific, rare terms or highly technical jargon if not well-represented in its embedding space.
It can understand 'add element to Python array' is similar to 'Python list append'.
Introducing Hybrid Search
Hybrid search combines the best of both worlds: the precision of keyword search and the contextual understanding of semantic search.
By running both types of retrieval and intelligently merging their results, hybrid search can significantly improve the relevance and completeness of retrieved documents for your RAG system.
Merging Results: Reciprocal Rank Fusion
A common method to combine results from multiple retrievers in hybrid search is Reciprocal Rank Fusion (RRF).
RRF assigns a score to each document based on its rank in each individual retriever's result list. Documents that appear high in multiple lists get a boosted score, leading to a more robust final ranking.
Try running this simplified example of RRF:
def reciprocal_rank_fusion(rank_lists, k=60):
fused_scores = {}
for rank_list in rank_lists:
for rank, doc_id in enumerate(rank_list):
if doc_id not in fused_scores:
fused_scores[doc_id] = 0.0
fused_scores[doc_id] += 1.0 / (k + rank + 1)
sorted_docs = sorted(fused_scores.items(), key=lambda item: item[1], reverse=True)
return [doc_id for doc_id, score in sorted_docs]
if __name__ == "__main__":
# Simulate results from two retrievers
keyword_results = ["docA", "docC", "docB", "docE"]
semantic_results = ["docB", "docA", "docD", "docC"]
fused_order = reciprocal_rank_fusion([keyword_results, semantic_results])
print("Fused Order:", fused_order)Hybrid Search with LangChain
LangChain provides an EnsembleRetriever to easily implement hybrid search. It takes multiple retrievers (e.g., a keyword retriever and a vector store retriever) and combines their results, often using RRF by default.
This allows you to leverage both precise keyword matches and semantic understanding in one powerful retrieval step.
# from langchain.retrievers import EnsembleRetriever
# from langchain_community.retrievers import BM25Retriever
# from langchain_community.vectorstores import FAISS
# from langchain_openai import OpenAIEmbeddings
# # Assume you have a BM25 retriever and a vector store retriever
# bm25_retriever = BM25Retriever.from_documents(docs)
# vectorstore = FAISS.from_documents(docs, OpenAIEmbeddings())
# vectorstore_retriever = vectorstore.as_retriever()
# ensemble_retriever = EnsembleRetriever(retrievers=[
# bm25_retriever,
# vectorstore_retriever
# ], weights=[0.5, 0.5])
# # query = "What are the capital cities of Europe?"
# # docs = ensemble_retriever.invoke(query)The Need for Re-ranking
Even after hybrid search, the initial set of retrieved documents might contain some noise or documents that are not perfectly ordered by relevance.
Re-ranking is a crucial next step. It takes the top-k documents from the initial retrieval and re-evaluates their relevance to the query using a more sophisticated model.
How Re-ranking Works
A re-ranking model, often a cross-encoder, takes both the user query and each retrieved document as input.
Unlike embedding models that create separate embeddings, a cross-encoder jointly processes the query and document to generate a single relevance score. This allows for a more nuanced understanding of their interaction.
Integrating Re-rankers
Various re-ranking models and services are available, such as Cohere's Re-rank API or open-source cross-encoders from libraries like sentence-transformers.
Integrating a re-ranker typically involves passing the initial retrieval results and the query to the re-ranker, which then returns the documents in a new, optimized order.
# from langchain.retrievers.document_compressors import CohereRerank
# from langchain.retrievers import ContextualCompressionRetriever
# # Assume you have an existing base_retriever (e.g., your EnsembleRetriever)
# # base_retriever = ensemble_retriever
# # Initialize the Cohere Rerank compressor
# # cohere_re_ranker = CohereRerank(top_n=5, cohere_api_key="YOUR_COHERE_API_KEY")
# # Create a compression retriever that uses the re-ranker
# # compression_retriever = ContextualCompressionRetriever(
# # base_compressor=cohere_re_ranker,
# # base_retriever=base_retriever
# # )
# # query = "What is the capital of France?"
# # compressed_docs = compression_retriever.invoke(query)The Full Advanced Retrieval Pipeline
By combining hybrid search and re-ranking, you create a robust retrieval pipeline:
- User Query
- Hybrid Search: Combines keyword and semantic retrieval to get an initial set of relevant documents.
- Re-ranking: A specialized model re-orders the top documents from hybrid search for maximum relevance.
- Context for LLM: The highly relevant, re-ranked documents are passed to the LLM for generation.
This approach leads to more accurate and contextually rich answers from your RAG system.
Quick Check: Retrieval Steps
Which of the following statements accurately describe the roles of Hybrid Search and Re-ranking in a RAG system?
Recap: Advanced Retrieval
We've explored advanced retrieval techniques to boost RAG performance:
- Hybrid Search: Combines keyword and semantic approaches for comprehensive initial retrieval.
- Reciprocal Rank Fusion (RRF): A method to merge and score results from different retrievers.
- Re-ranking: Uses specialized models (like cross-encoders) to refine the relevance order of retrieved documents.
These techniques ensure your RAG system provides the most accurate and contextually relevant information to the LLM.
常见问题解答
「混合搜索与重新排序」课时是免费的吗?
是的 — 「混合搜索与重新排序」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 LangChain / RAG / Vector DBs 课程的其余内容,请升级到 CoddyKit PRO。 LangChain / RAG / Vector DBs 课程共包含 4 节课。
「混合搜索与重新排序」这节课中我会学到什么?
结合关键词搜索和语义搜索(混合搜索),并使用重新排序模型优先处理最相关的文档 你通过在浏览器中直接运行的动手代码来练习 LangChain / RAG / Vector DBs,全天候 AI 导师会在你学习这节课的过程中回答你的问题。
学习 LangChain / RAG / Vector DBs 需要有经验吗?
无需任何先前经验。CoddyKit 上的 LangChain / RAG / Vector DBs 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 3 节课,共 4 节。
「混合搜索与重新排序」课时需要多长时间?
大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。
我能在这节 LangChain / RAG / Vector DBs 课中编写并运行代码吗?
能。每节 LangChain / RAG / Vector DBs 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。
此课程中的所有课时
- 多查询检索策略
- 使用 LLM 进行上下文压缩
- 混合搜索与重新排序
- 父文档与句子窗口检索