Better RAG Results with Reciprocal Rank Fusion and Hybrid Search

249 点作者 johnjwang12 个月前

24 条评论

If you're looking for an example of RRF + Hybrid Search with PostgreSQL, I've put together a FastAPI app here that uses RAG with those options:<a href="https://github.com/Azure-Samples/rag-postgres-openai-python/">https://github.com/Azure-Samples/rag-postgres-openai-python/</a>Here's the RRF+Hybrid part: <a href="https://github.com/Azure-Samples/rag-postgres-openai-python/blob/e30ea96ca11ca6578ca38d3428594bd98d704900/src/fastapi_app/postgres_searcher.py#L38">https://github.com/Azure-Samples/rag-postgres-openai-python/...</a>That's largely based off a sample from the pgvector repo, with a few tweaks.Agreed that Hybrid is the way to go, it's what the Azure AI Search team also recommends, based off their research:<a href="https://techcommunity.microsoft.com/t5/ai-azure-ai-services-blog/azure-ai-search-outperforming-vector-search-with-hybrid/ba-p/3929167" rel="nofollow">https://techcommunity.microsoft.com/t5/ai-azure-ai-services-...</a>

评论 #40538286 未加载

thefourthchime12 个月前

I also found pure RAG with vector search to not work. I was creating a bot that could find answers to questions about things by looking at Slack discussions.At first, I downloaded entire channels, loaded them into a vector DB, and did RAG. The results sucked. Vector searches don't understand things very well, and in this world, specific keywords and error messages are very searchable.Instead, I take the user's query, ask an LLM (Claude / Bedrock) to find keywords, then search Slack using the API, get results, and use an LLM to filter for discussions that are relevant, then summarize them all in a response.This is slow, of course, so it's very multi-threaded. A typical response will be within 30 seconds.

评论 #40533618 未加载

评论 #40529308 未加载

评论 #40532639 未加载

评论 #40530710 未加载

edude0312 个月前

Thanks for sharing, I like the approach and it makes a lot of sense for the problem space. Especially using existing products vs building/hosting your own.I was however tripped up by this sentence close to the beginning:> we encountered a significant challenge with RAG: relying solely on vector search (even using both dense and sparse vectors) doesn’t always deliver satisfactory results for certain queries.Not to be overly pedantic, but that's a problem with vector similarity, not RAG as a concept.Although the author is clearly aware of that - I have had numerous conversations in the past few months alone of people essentially saying "RAG doesn't work because I use pg_vector (or whatever) and it never finds what I'm looking for" not realizing 1) it's not the only way to do RAG, and 2) there is often a fair difference between the embeddings and the vectorized query, and with awareness of why that is you can figure out how to fix it.<a href="https://medium.com/@cdg2718/why-your-rag-doesnt-work-9755726dd1e9" rel="nofollow">https://medium.com/@cdg2718/why-your-rag-doesnt-work-9755726...</a> basically says everything I often say to people with RAG/vector search problems but again, seems like the assembled team has it handled :)

评论 #40525374 未加载

评论 #40531282 未加载

eskibars12 个月前

Having worked building out a RAG SaaS platform for the past year and having worked on the vendor side of several keyword-based search systems in the past 10 years, I can say it's absolutely necessary to have some kind of hybrid search for most use cases I've seen.The problem is that most people don't have experience optimizing even 1 of the retrieval systems (vector or keyword), so a lot of users that try to DIY build end up with an awful time trying to get to prod. People are talking about things like RRF (which are needed) but then missing other big-picture things like the mistakes everyone makes when building out a keyword search (not getting the right language rules in place) and also not getting the right vector side (finding the right embedding models, chunking strategies, etc).I recognize I have a bit of a conflict of interest since I'm at a RAG vendor, but I'll abstain from the name/self-promotion and say: I've seen so many cases where people get this wrong, if you're thinking RAG you really should be hiring a consultant or looking at a complete platform from people that have done it more. Or be prepared to spend a lot of cycles learning and iterating

评论 #40536325 未加载

评论 #40534825 未加载

pmc0012 个月前

For another set of measurements that support RRF + Hybrid > vectors, we (Azure AI Search team) did a bunch of evaluations a few months ago: <a href="https://techcommunity.microsoft.com/t5/ai-azure-ai-services-blog/azure-ai-search-outperforming-vector-search-with-hybrid/ba-p/3929167" rel="nofollow">https://techcommunity.microsoft.com/t5/ai-azure-ai-services-...</a>We also included supporting data in that write up showing you can improve significantly on top of Hybrid/RRF using a reranking stage (assuming you have a good reranker model), so we shipped one as an optional step as part of our search engine.

cheesyFish12 个月前

RRF is alright, but I've had better results with relative score, or distribution-based scoring.LlamaIndex has a module for exactly this<a href="https://docs.llamaindex.ai/en/stable/examples/retrievers/relative_score_dist_fusion/" rel="nofollow">https://docs.llamaindex.ai/en/stable/examples/retrievers/rel...</a>

yingfeng12 个月前

RRF is a simple and effective means of fused ranking for multiple recall. Within our open source RAG product RAGFlow(<a href="https://github.com/infiniflow/ragflow">https://github.com/infiniflow/ragflow</a>), Elasticsearch is currently used instead of other general vector databases, because it can provide hybrid search right now. Under the default cases, embedding based reranker is not required, just RRF is enough, while even if reranker is used, keywords based retrieval is also a MUST to be hybridized with embedding based retrieval, that's just what RAGFlow's latest 0.7 release has provided.On the other hand let me introduce another database we developed, Infinity(<a href="https://github.com/infiniflow/infinity">https://github.com/infiniflow/infinity</a>), which can provide the hybrid search, you can see the performance here(<a href="https://github.com/infiniflow/infinity/blob/main/docs/references/benchmark.md">https://github.com/infiniflow/infinity/blob/main/docs/refere...</a>), both vector search and full-text search could perform much faster than other open source alternatives.From the next version(weeks later), Infinity will also provide more comprehensive hybrid search capabilities, what you have mentioned the 3-way recalls(dense vector, sparse vector, keyword search) could be provided within single request.

评论 #40532447 未加载

retakeming12 个月前

pg_search (full text search Postgres extension) can be used with pgvector for hybrid search over Postgres tables. It comes with a helpful hybrid search function that uses relative score fusion. Whereas rank fusion considers just the order of the results, relative score fusion uses the actual metrics outputted by text/vector search.

throwaway11512 个月前

I've implemented a very similar RAG hybrid solution, and it has improved LLM responses enormously. There are other things you can do too that have huge improvements, like destructuring your data and placing it into a graph structure, with queryable edge relationships. I think we're just scratching the surface.

评论 #40527846 未加载

janalsncm12 个月前

Reciprocal rank scoring is just one way of forcing scores into a fixed distribution: in this case, decaying with the reciprocal of its rank. But it also assumes fixed weight from all components, i.e. the top ranked keyword match has equal relevance to the top ranked semantic match.There are a couple ways around this. Either learning the relative importance based on the query, and/or using a separate reranking function (usually a DNN) that also takes user behavior into account.

ko_pivot12 个月前

Meilisearch has a really clean implementation of this. Can easily adjust keyword vs vector weighting per query.

gregnr12 个月前

In case you just want a single Postgres function that does RRF (pgvector+fts): <a href="https://supabase.com/docs/guides/ai/hybrid-search">https://supabase.com/docs/guides/ai/hybrid-search</a>(disclaimer: supabase dev who went down the rabbit hole with hybrid search)

SomewhatLikely12 个月前

Both of your references use RRF=1/(60+Rank)So I'm not sure why the article uses 1/Rank alone. Did you test both and find that the smoothing didn't help? My understanding is that it has been pretty important for the best results.

评论 #40526867 未加载

评论 #40526876 未加载

owen-elliott12 个月前

The composability of RRF is definitely one of its most appealing characteristics. It doesn't matter what algorithm or vendor you have, you can just fuse with ranks alone. I've seen it shine when fusing lexical and vector search results where semantic attributes like styles and exact attributes like quantities are mixed together in queries, e.g., "modern formal watch with 40mm face".While it's not such a problem in RAG, one downside is that it complicates pagination for results (there are a few different ways to tackle this).

cricketlover12 个月前

Pardon my ignorance but I was hung up on this line.> Out-of-sync document stores could lead to subtle bugs, such as a document being present in one store but not another.But then the article suggests to upload synchronously in S3/DDB and then sync asynchronously to actual document stores. How does this solve out of sync issue? It doesn't. It can't be solved is what I'm thinking.> Data, numbersHow much data are we talking about?

marcyb5st12 个月前

As soon as the indexed documents contain lingo of any kind you need hybrid search IMHO.Additionally, if you can add conditional fuzzy matching into the mix so fat fingering something still yields a workable result is even better for UX (something along the lines of "the results from the tf-idf search are garbage, let's redo the search with fuzzy matching this time).

_pdp_12 个月前

I have no doubt this probably produces better results than a simple vector search, but you cannot escape the fact that you are converting a query to a set of results, and so the quality and intent of the query matter. In fact, it matters more than the search mechanics. Anyone who has ever used a search engine or some other search mechanism knows that intuitively.

treprinum12 个月前

Hybrid might work for English but where are you going to get sparse embeddings like SPLADE or ELSERv2 for most other languages? Vector search with ada-002 or text-003-large capped to the first 500-1000 dimensions will give you a support for 100+ languages for free. If you are using BM25, then you need to train BM25 on every single separate knowledge base which is annoying and expensive.

mtbarta312 个月前

Great article. Hybrid search works well for a lot of scenarios.The tradeoffs of using existing systems vs building your own resonate with me. What we eventually experienced, however, is that periods of bad search performance often correlated to out-of-date search indices.I'd be interested in another article detailing how you monitor search. It can be tricky to keep an entire search system moving.

esafak12 个月前

1. Does anyone know a postgres reranking extension, to go beyond RRF through ML models or at least custom code?2. If anyone is observing significant gains from incorporating knowledge graphs into the retrieval step, what kind of a knowledge graph are you working with, what is your retrieval algorithm, and what technology are you using to store it?

评论 #40529063 未加载

ndricca12 个月前

May I ask you if you tried hybrid search directly on Pinecone, using Bm25 or splade?

cpursley12 个月前

Any tips on accomplishing this in Postgres with pg_vector?

评论 #40529056 未加载

评论 #40527209 未加载

评论 #40527580 未加载

armind12 个月前

how does this compare to differentiable search index (DSI) and it's updated version (DSI++)?

yding12 个月前

Absolutely makes sense!