模型重新排序
重新排序器是第二階段模型,它對搜尋結果的候選清單重新評分以使其與查詢的相關性,從而在快速檢索器拉出候選者後銳化排序。
概述
It is a key ingredient in modern search and retrieval-augmented generation (RAG).
深入探討
搜尋和 RAG 系統通常分兩個階段工作。首先,快速檢索器(通常是向量/嵌入搜尋或關鍵字 BM25)從數百萬個文件中提取大約 50-100 個候選文件 - 針對召回率和速度進行了最佳化。但第一遍對查詢和文件分別進行評分,因此可能會錯過細微差別。重新排名器是精確步驟:它將查詢和每個候選者放在一起並輸出細粒度的相關性分數,然後對清單重新排序,以便最佳結果上升到頂部。主要架構是交叉編碼器:它將查詢和文檔聯合輸入到轉換器中,讓每個查詢標記處理每個文檔標記。這種深度交互作用使得重新排序比嵌入相似性更加準確,但代價是每個候選者運行一次。
技術洞察
對比是雙編碼器與交叉編碼器。雙編碼器將查詢和文件獨立地嵌入向量中,因此相似性是一種廉價的點積——快速且可預先計算,但很淺。交叉編碼器將查詢和文件連接到一個輸入中,並運行完整的變壓器傳遞,產生具有豐富標記級關注的單一相關性得分。它無法預先計算,因此保留用於對小候選清單進行重新排名。 Cohere Rerank 和 BGE-reranker 等模型就證明了這一點。
戰略影響
成本與預算
多年來,架構決策決定著效能和營運成本。
更明確的決策
技術教育幫助團隊選擇正確的堆疊,而不僅僅是最新的堆疊。
品質管控
更好的工程選擇可以減少生產中的可靠性事故。
重新排序模型的未來
重新排序器正在成為 RAG 管道中的標準,因為更好排序的上下文可以直接提高 LLM 答案品質並減少幻覺。期待更輕、更快的交叉編碼器、多語言和多模式重新排序(文本加圖像或表格)以及更長的上下文窗口,以便可以對整個文件進行評分。基於 LLM 的「列表式」重新排序器可以同時判斷整個候選集,並且一些系統將交叉編碼器的判斷提煉回更便宜的檢索器中,以獲得接近第一階段的準確性。
現實世界的實施
RAG 聊天機器人透過嵌入搜尋檢索 50 個區塊,然後重新排名以僅將前 5 個最相關的區塊提供到 LLM 的上下文中
電子商務搜尋對產品結果重新排序,以便最匹配購物者完整查詢短語的項目首先出現
Cohere Rerank 或 BGE-reranker 提高了對數千個政策 PDF 進行企業文件搜尋的精確度
客戶支援知識庫對檢索到的幫助文章進行重新排序,以便代理在頂部顯示最相關的單一答案
風險與防護欄
優化一項基準測試可以隱藏更廣泛的系統弱點。
基礎設施和維護成本常常被低估。
隨著系統變得更加複雜,安全性和可觀察性差距可能會擴大。
實施路線圖
在實施之前定義延遲、品質和成本目標。
在實際負載和資料條件下進行基準測試。
儀器監控錯誤、漂移和使用者影響。
在擴展之前準備回滾和事件回應路徑。
不斷探索
Free newsletter
Get the daily AI briefing
Three verified AI stories every weekday morning, written in plain English. Free forever, no ads.
One email each weekday. Unsubscribe in one click. We never sell or share your address.
Test yourself
Take the Reranking Models quiz
Instant feedback on every answer, and a shareable certificate with a verifiable ID once you pass a course.
Support free AI education. AI Understanding is a 501(c)(3) nonprofit — no ads, no paywall, ever. Make a donation
常見問題
What is Reranking Models?
重新排序器是第二階段模型,它對搜尋結果的候選清單重新評分以使其與查詢的相關性,從而在快速檢索器拉出候選者後銳化排序。它是現代搜尋和檢索增強生成 (RAG) 的關鍵要素。
在典型的兩階段檢索管道中,重新排序器的工作是什麼?
快速檢索器會取得候選者;然後,重新排序器會對該候選清單重新評分,以將最相關的結果推到頂部。
大多數重排序器使用什麼架構?
重新排序器通常使用交叉編碼器,共同輸入查詢和文檔,以便注意力可以對它們的交互作用進行建模。
為什麼不能使用跨編碼器重排序器直接搜尋數百萬個文件?
與可預先計算的嵌入不同,交叉編碼器必須每對運行完整的變壓器通道,因此它保留給一個小的候選清單。
雙編碼器相對於交叉編碼器的主要優點是什麼?
雙編碼器提前獨立嵌入文檔,從而能夠在龐大的集合中進行快速相似性搜尋。
重新排序器如何幫助減少 RAG 系統中的幻覺?
更有序、更相關的背景為法學碩士提供了準確的基礎,從而減少了捏造的答案。