RAG But Better: Rerankers with Cohere AI

Ғылым және технология

Rerankers have been a common component of retrieval pipelines for many years. They allow us to add a final "reranking" step to our retrieval pipelines - like with Retrieval Augmented Generation (RAG) - that can be used to dramatically optimize our retrieval pipelines and improve their accuracy.
In this video we'll learn about rerankers, how they compare to the more common embedding retrieval only setup, and how we can create retrieval pipelines with reranking using Cohere AI reranking model. We'll also be using the (more typical) OpenAI text-embedding-ada-002 model with the Pinecone Vector Database.
📌 Code (08:32):
github.com/pinecone-io/exampl...
📚 Article:
www.pinecone.io/learn/series/...
🌲 Subscribe for Latest Articles and Videos:
www.pinecone.io/newsletter-si...
👋🏼 AI Consulting:
aurelio.ai
👾 Discord:
/ discord
Twitter: / jamescalam
LinkedIn: / jamescalam
00:00 RAG and Rerankers
01:25 Problems of Retrieval Only
04:32 How Embedding Models Work
06:34 How Rerankers Work
08:20 Implementing Reranking in Python
13:11 Testing Retrieval without Reranking
15:21 Retrieval with Cohere Reranking
21:54 Tips for Reranking
#artificialintelligence #nlp #ai #openai

Пікірлер: 99

  • @slayermm1122
    @slayermm1122Ай бұрын

    cohere just released rerank3 and it wokred increiblely fantastic with openai's embedding 3 model; thanks for your kind intro

  • @jantuitman
    @jantuitman7 ай бұрын

    I learned a lot from this, thank you. You say you plan a series, and you were talking about other topics for the series but these other topics you mentioned were not about rerankers. I noted that this video treats rerankers as black boxes so you could even expand the series. I for sure would be interested in: what are the most recent reranking models, how doe rerankers work, is it feasible to make a reranker yourself or does this require, just like a transformer, that you scrape the entire language / internet? In other words, this video was very interesting, but now I know about rerankers I have lots and lots of questions about rerankers.

  • @gitmaxd
    @gitmaxd7 ай бұрын

    Each video gets better! Thank you for your work!

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    Thanks!

  • @adityavd97
    @adityavd97Ай бұрын

    Thanks man. You are improving my hobby projects in real time.

  • @matteomarjanovic
    @matteomarjanovic7 ай бұрын

    Hi, thank you so much for that content! Do you think that parameters like document chunks size and overlapping are important for RAG accuracy? Should we fine-tune them in some way?

  • @justinwlin
    @justinwlin7 ай бұрын

    You’ve got top notch editing + technical explanations and none of that is easy. The amt of work to create a 20 min video, and be cohesive on such a topic is amazing. Thanks! 🔥 all ur videos are so helpful and just interesting to watch and learn

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    that's awesome to hear, thanks :)

  • @samwilletts9390
    @samwilletts93907 ай бұрын

    Great video, looking forward to more on this!

  • @Shaunmcdonogh-shaunsurfing
    @Shaunmcdonogh-shaunsurfing7 ай бұрын

    Thank you for making this. Fascinating.

  • @narayangopalmaharjan
    @narayangopalmaharjan7 ай бұрын

    Thank you for this video, been stuck in RAG realm with llama index and not satisfied, I thought similar reranking but manually, i will try cohere today instead

  • @Shaunmcdonogh-shaunsurfing
    @Shaunmcdonogh-shaunsurfing7 ай бұрын

    You mentioned some better approaches than reranking. Any hints as to what that might be (curious to know if it involves fine tuning the LLM with the data too)

  • @Cdaprod
    @Cdaprod7 ай бұрын

    My god, thank you 🙏 as someone that only rebuilds the wheel, your content is very much appreciated.

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    Happy to hear it!

  • @alejandrovelez2083
    @alejandrovelez20837 ай бұрын

    Great content man!!! I have learned so much from you

  • @bonadio60
    @bonadio607 ай бұрын

    Very good content!! I will definitely try it. Thanks

  • @jellederijke
    @jellederijke7 ай бұрын

    Top notch material, James. Much appreciated 🎉🎉 Really curious to see what kind of difference this makes in my projects. Thanks!

  • @anandteerthrparvatikar5359

    @anandteerthrparvatikar5359

    7 ай бұрын

    Totally

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    Glad it helps - may want to try retrieval + reranker system for improved name recall 😅

  • @anandteerthrparvatikar5359

    @anandteerthrparvatikar5359

    7 ай бұрын

    You are literally educating corporate people. Waiting for next session, Thanks for the efforts

  • @jellederijke

    @jellederijke

    7 ай бұрын

    ​@@jamesbriggshah really Sorry James 😂 I have just reranked the names in my RAG

  • @timkoehler86
    @timkoehler864 ай бұрын

    Great video! Btw what software are you actually using to show/explain the concept? I really like the look of it.

  • @timwarren4332
    @timwarren43327 ай бұрын

    Great stuff! Thank you!

  • @real-ethan
    @real-ethan3 ай бұрын

    My approach is letting the LLM summarize the user's input first, the prompt could be written as: "The summary of the user's request to semantically search relevant documents in English." The output of the LLM's summarization can then be used to query the vector database after embedding. This approach may potentially increase the accuracy of retrieval.

  • @edwardmitchell6842

    @edwardmitchell6842

    3 ай бұрын

    How would an LLM know how to optimize for semantic search? I would expect an expansion on the ideas would be better. I need to finish my LLM Ops to get the answer,

  • @real-ethan

    @real-ethan

    3 ай бұрын

    ​@@edwardmitchell6842 The typical process we currently observe for Retrieval-Augmented Generation (RAG) is as follows: 1. After segmenting the document, embedding is directly performed, and the results are stored in the vector database. 2. For user inputs or queries, embedding is directly carried out, and a similarity search is conducted within the vector database. It is a peculiar solution to rely solely on mathematical similarity matching for a single question and multiple factual paragraphs. As humans, when addressing a customer's request or solving a problem, we are unlikely to simply copy and paste the original question into Google's search box and hit enter. Instead, we tend to abstract and summarize the question and requirements before conducting a Google search. To achieve more precise content retrieval, my personal approach involves: 1. After segmenting the document, each paragraph undergoes summarization using a Language Model (LLM). This summary is then embedded and treated as a comprehensive index for the original document. Together with the original document paragraphs, this summary index is stored in the vector database. During subsequent retrieval and matching processes, the focus is solely on this summarization index, streamlining the matching operation. 2. For user input/questions, a summary is generated using the LLM, embedded, and then queried against the vector database to match the summary index of each document paragraph. After retrieving multiple documents, the results are then evaluated and selected by the LLM, ultimately producing the answer. The current Rag demos and new services, such as rerank provided by community, are all efforts focused on QUERYING. Perhaps we should explore more in terms of how to store, organize, and index documents. An additional hidden benefit of this approach is that, assuming our documents are in English, if a user inputs a Chinese question, direct embedding would inevitably fail to retrieve any content. Through summarization and subsequent embedding, we can translate the original input into English before processing.😄

  • @andikunar7183
    @andikunar71837 ай бұрын

    Great content, thanks a lot!

  • @tushaar9027
    @tushaar90277 ай бұрын

    Hi James great video learned a lot, actually i was using multi query retriever in my approach and was seeing the slow inference speed because of overstuffing as you mentioned. Can you give more info on re ranking models, any free ones we can use in our projects.

  • @RedCloudServices
    @RedCloudServices7 ай бұрын

    James thank as always. I hope I am asking these questions with clarity. (1) You used a different encoder model ada 002 with Cohere LLM as the vector response model? (2) Huggingface have rankings for encoding models and rankings for LLMs but are there rankings for pairs of encoding:response LLMs pairs?

  • @vinsentparamanantham5756
    @vinsentparamanantham57567 ай бұрын

    Hi James, can you give as an example with openapi since we have compliance issues, we need to run against the locally hosted llama models and also locally hosted vector database. thank you

  • @pythontok4192
    @pythontok41926 ай бұрын

    Thank you for this James. I found that when I return more things in the context, the LLM also tries to make up answers that are an amalgation of several sources' context. Any ways around this, from your experience?

  • @SanjeevKumar-hj1fb
    @SanjeevKumar-hj1fb6 ай бұрын

    Thanks for the great videos. How does embedding work on numeric fields? Shall we use embeddings for non text fields?

  • @TheOfficialWoover
    @TheOfficialWoover7 ай бұрын

    That's GREAT! thank you!

  • @Jandodev
    @Jandodev7 ай бұрын

    I've been doing this with transformers I think theirs a alot to think about with doing this efficiently but it does get the best results!

  • @harisjaved1379
    @harisjaved13794 ай бұрын

    Have been doing this for few years now. Good video but you should cover bi-encoders vs cross encoder as this is one of the best reranking techniques and also talk a bit about FAISS.

  • @LoVeRSaMa
    @LoVeRSaMa6 ай бұрын

    Any chance you can show examples with OpenSource re ranking like: JinaAI-v2-base-en for example?

  • @da-bb2up
    @da-bb2up3 ай бұрын

    Great video, James! :) What do you think is better if you compare optimizing stratagies? 1.finetuning the embedding model on your domain specific language,2 . use a hybrid search, which combines dense and sparse retrieval, 3.Reranking ? Or 4.Could you combine the three optimization strategies maybe? Thank you in advance for your answer. :) and another question: is reranking not pretty much the same as the hybrid search?(because it also uses also two search strategies but in a slitght different way - first it searches the data chunk candidates and than it searches out of candidates)

  • @SinaKMaram
    @SinaKMaram2 ай бұрын

    Hey James, great great series on Retrieval Augmented Generation... One question, having looked at the notebook and the video, why don't we avoid vector embedding and have cohere's rerank to do the job for us? I did test the idea over a group of pdf documents and it seems like the performance was significantly better particularly considering that we pass the entire text altogether to cohere API instead of breaking them down into chunks. I understand there can be cost implications involved but considering the free cohere pricing isn't that a better approach? Afterall, any reranking you perform on top of results from pinecone is somewhat at the mercy of how well you retrieve the original vectors. Would appreciate your thoughts.

  • @ganj0rm0n
    @ganj0rm0n7 ай бұрын

    Awesome! I wonder if there is a way to use a re-ranker with low code tools like flowise.

  • @hughesadam87
    @hughesadam877 ай бұрын

    Have you done any videos on ETL or suggestiosn for getting data into RAG systems? I'd really love to start with an open-source project that is more opinionated and ready-made for RAG than just langchain. LLMware looks promising. Do you have any suggestions? Some framework that would have opinionated, deployable RAG systems that solve hard problems like: auth, reranking, doc ingestion/scrubbing etc... Something I can just fire up in k8s and start fiddling w/ ? Does this exist to your knowledge? Thanks for the great video

  • @elrecreoadan878
    @elrecreoadan8787 ай бұрын

    Hi, what approach would you suggest for a hotel or restaurant customer service bot? Maybe botpress + plugin like vectora + chatgpt?

  • @shaheerzaman620
    @shaheerzaman6207 ай бұрын

    awesome stuff!

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    thanks Shaheer :)

  • @GiridharReddy-hb5nv
    @GiridharReddy-hb5nv7 ай бұрын

    is there any open source way to do the reranking ? The content was great!

  • @matthewpublikum3114
    @matthewpublikum31147 ай бұрын

    Are the reranking models specifically trained for the task, or are they decoder or encoder portion of an LLM?

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    yes they're fine-tuned specifically for calculating similarity scores - you take a pretrained transformer model, add 1-2 linear layers on to the top of the output logits of the model, and fine-tune on a dataset that would contain records like [sentence A, sentence B, similarity score]

  • @GeigenAkademie
    @GeigenAkademie4 ай бұрын

    Nowadays the context length cab be ~32k - why reranking, if I could put all possible matches to the final answering step/llm? Therefore the answering itself does a kind of reranking of the contexts

  • @RabeeQasem
    @RabeeQasem7 ай бұрын

    can you do a tut on how to use falcon to chat with you data and use diffrent data loaders ( txt,pdf,json)? love you content

  • @ackiamm
    @ackiamm7 ай бұрын

    thanks sir

  • @sanchaythalnerkar9736
    @sanchaythalnerkar97367 ай бұрын

    Can we use llama index to improve the efficiency?

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    llama-index have a lot of great retrieval tooling - I haven't been able to dive too deeply into it yet but from what I've seen they (1) do support this type of retrieval (ie with reranking), and (2) can likely improve accuracy, but I don't think you can get much faster than what we do here

  • @sanchaythalnerkar9736

    @sanchaythalnerkar9736

    7 ай бұрын

    Why is the embedding taking so long?@@jamesbriggs

  • @tiagoc9754
    @tiagoc97544 ай бұрын

    How much the vector store affects the RAG responses accuracy?

  • @dato007
    @dato0075 ай бұрын

    I don’t understand how re-ranking is adding anything. you’re giving it the same query again and they’ve already been matched with a vector similarity what additional information is using your improve the ranking? Thx!

  • @edvinbeqari7551
    @edvinbeqari75517 ай бұрын

    I didn't understand how you get a similarity score from one transformer. Whats the hint?

  • @frazuppi4897
    @frazuppi48977 ай бұрын

    any benchmark? otherwise is kinda of very empirical and only seems like a sponsored video by Cohere

  • @matteomarjanovic

    @matteomarjanovic

    7 ай бұрын

    Good point. Do you know about any possibly useful metric/benchmark?

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    no sponsor from Cohere, I'm sharing what I do in production to make search better

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    there are many benchmarks comparing bi-encoders (embedding models) to crossencoders (rerankers), but I'm not aware of any for Cohere's model compared directly to ada-002. Nonetheless you can read here txt.cohere.com/rerank/ (it only shows comparison with elastic)

  • @frazuppi4897

    @frazuppi4897

    7 ай бұрын

    @@jamesbriggs thanks a lot for the reply, so in production you push things after an empirical evaluation? Would it be possible to have a link to some benchmarks? Thanks a lot again

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    @@frazuppi4897 yeah it's a lot of fast moving projects for me at the moment, so we do a lot of empirical assessments, for reranking you can see benchmarks for some of the best performing embedding + reranker models here huggingface.co/BAAI/bge-reranker-large#baai-embedding They unfortunately don't compare reranker to encoder directly beyond a few statements on rerankers being more accurate - they do explain in better language than I the reason for this though

  • @nishanthk7048
    @nishanthk70487 ай бұрын

    Hey can anyone answer my question, While reranking it calculates relevance score again, so while calculating the score does cohere inferences the LLM or uses a algorithm?

  • @bastabey2652

    @bastabey2652

    14 күн бұрын

    I believe Cohere ReRank doesn't use a separate LLM model.. it relies on its algorithm/model

  • @hetnon
    @hetnonАй бұрын

    I don't get the part that you feed both documents in the same transformer. If your transformer output is only 1 array, what are you comparing to? You have only 1 array to compare to... nothing? What did I miss?

  • @victorhenriquecollasanta4740
    @victorhenriquecollasanta47404 ай бұрын

    amazing! can you make more enterprenerial videos, maybe on how to apply this knlowdge to build a business

  • @frazuppi4897
    @frazuppi48977 ай бұрын

    well the reranker will prob use a [CLS] token so still one vector so I don't get why you say that in the normal embedding we loose info but in the reranker no - weird. If you are sending the two documents each token will able to attend to the other, this could means the info is more accurate

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    you can read about bi-encoders (embedding models) and crossencoders (reranker), there is information compression with the bi-encoder approach as you are encoding generic embeddings, with the crossencoder you are feeding the query and original text, the transformer must then decide, on that specific query, how relevant the document is

  • @frazuppi4897

    @frazuppi4897

    7 ай бұрын

    @@jamesbriggs copy that - thanks a lot!

  • @torstenkarstadt9785
    @torstenkarstadt97856 ай бұрын

    Does Canopy support this rerankng approach?

  • @jamesbriggs

    @jamesbriggs

    6 ай бұрын

    I know it's on the roadmap, but it's not in there yet

  • @bastabey2652
    @bastabey265214 күн бұрын

    how does the ReRanker know it needs to return 3 documents with relevant information to the user's query?

  • @jamesbriggs

    @jamesbriggs

    14 күн бұрын

    we set the `top_n` parameter to `3`, logically the reranking scores every document, then we take the top 3 scoring docs

  • @nicholasliu-sontag1585
    @nicholasliu-sontag15857 ай бұрын

    You describe the re-ranker transformer as more accurate because it doesn't encode the documents into vectors - but don't all transformers work off a vectors to begin with? Isn't it still working with the same vectors that are used to calculate similarity score?

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    I probably could have phrased better, there are two parts: 1. Embedding models encode the full sequence into a single vector, transformers work with vectors but they contain a single vector for each token - but these will be compressed through a single layer before producing the similarity score, so there is still compression into a single vector happening, but... 2. Reranker models have the full context, ie they see both the query and the document that they must compute similarity for. An embedding model must produce a single vector embedding for every possible query

  • @nicholasliu-sontag1585

    @nicholasliu-sontag1585

    7 ай бұрын

    makes sense. thanks for explaining!@@jamesbriggs

  • @masssurfski

    @masssurfski

    5 ай бұрын

    My first impression was that the need to rerank means that the rank was to optimal to begin with. Your explanation above helped me better understand this. Ultimately this capability should be integrated and not require a different tool.

  • @thedoctor5478
    @thedoctor54787 ай бұрын

    I believe we have better than openai embeddings now. The leaderboard says so anyway. Also, backoff library is better for retries.

  • @heywrandom8924

    @heywrandom8924

    7 ай бұрын

    I did not watch the video but I am interested in knowing what leaderhoard you are referring to

  • @thedoctor5478

    @thedoctor5478

    7 ай бұрын

    on huggingface /spaces/mteb/leaderboard@@heywrandom8924

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    @@heywrandom8924 probably this one huggingface.co/spaces/mteb/leaderboard - I'll be talking about other embedding models in upcoming video, but yes it's true, ada-002 is far from best performing

  • @heywrandom8924

    @heywrandom8924

    7 ай бұрын

    ​@@jamesbriggsthank you (:. I didn't watch that specific video as I am not sure what the keywords in the title mean and I am not sure it's relevant to me. I just checked it out a bit and the video looks cool (:.

  • @da-bb2up
    @da-bb2up3 ай бұрын

    kzread.info/dash/bejne/h5xtxLuihtW9j9Y.html isn't the similarity score also calculated at this point at the end with cosinesimililarity just like without retriever or how is the similarity score exactly calculated?

  • @HazemAzim
    @HazemAzim7 ай бұрын

    Great . Any other open source reranking transformer on Hugging Face ? other than cohere which is closed source ?

  • @haristan1960

    @haristan1960

    7 ай бұрын

    Sentence transformer has cross encoder models in hugging face you can try them but there quite old

  • @HazemAzim

    @HazemAzim

    7 ай бұрын

    @@haristan1960 Yes true .. I found many cross encoders on SBERT and hugging face . Thanks

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    check out bpe-reranker huggingface.co/BAAI/bge-reranker-large/tree/main

  • @Data_scientist_t3rmi
    @Data_scientist_t3rmi7 ай бұрын

    do you have any videos about scalability ? i mean for 1000 pdfs it could be good thing but for 100000 documents, the time to pre-process is diffcult. thanks again for the video you were the most person that introduce me to Transformers

  • @michaeldausmann6066
    @michaeldausmann60664 ай бұрын

    cool, ok. but... how does it work? what does it do? you just give it a query and it reranks for you.......wtf what is the magic sauce I want to understand the technique.

  • @nirbhaykumar4906

    @nirbhaykumar4906

    15 күн бұрын

    Yes some more detail into working of reranker would be useful.

  • @bastabey2652
    @bastabey265214 күн бұрын

    the moment showing how LLM reads the scraped concatenated text is impressive kzread.info/dash/bejne/h5xtxLuihtW9j9Y.html

  • @jamesbriggs

    @jamesbriggs

    12 күн бұрын

    it's pretty wild

  • @vanerk_
    @vanerk_4 ай бұрын

    you have explained a high level idea of reranker whereas explanation of reranker achitecture was expected, dislike.

  • @jamesbriggs

    @jamesbriggs

    4 ай бұрын

    see here kzread.info/dash/bejne/iYdl17iGd8vSh7Q.html

  • @narutocole
    @narutocole7 ай бұрын

    Do you have any thoughts or recommendations for Opensource re-rankers? I've used 'cross-encoder/mmarco-mMiniLMv2-L12-H384-v1' for re-ranking. But I'm curious as to if anyone has using some of the recent LLMs and modifying them to work for Re-ranking similar to how SGPT modified EleutherAI/gpt-neo-125M

  • @jamesbriggs

    @jamesbriggs

    7 ай бұрын

    Hey Jordan! I haven't tested the open-source cross-encoders/rerankers for a long time - so I'm not sure - they generally get less attention than the encoder models but I'm sure there must be some good rerankers out there

  • @lachlanholland7157

    @lachlanholland7157

    7 ай бұрын

    I’m looking into using bge reranker large, however haven’t gotten it to work yet.

Келесі