From RAG to RAPTOR: Improving AI Retrieval with Open-Source Models

The journey from Retrieval Augmented Generation (RAG) to RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval) marks significant progress in addressing challenges like hallucinations and confidently incorrect responses.

Powered by Ollama and open-source tools, this development path increases AI accuracy and usability.

RAG

Understanding RAG (Retrieval Augmented Generation)

RAG improves AI responses by supplementing pre-trained models with user-specific data from external sources like JSON, PDFs, or GitHub repositories. This data is transformed into chunks stored in a vector store, allowing AI to retrieve relevant information, reducing hallucinations and incorrect answers.

The mechanism of RAG

Advancing to RAG Fusion

RAG Fusion builds on RAG by incorporating multi-query generation and reciprocal ranking. Multi-query generation involves AI creating multiple questions from the user’s original query, enhancing the retrieval process. Reciprocal ranking then ranks these answers, ensuring the most accurate and relevant response is provided.

RAG Fusion
The mechanism of RAG Fusion

Introducing RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval)

RAPTOR takes AI retrieval a step further. Unlike traditional RAG, RAPTOR organizes data in a tree structure, summarizing at each layer from the bottom up. This method captures broader context and enhances the representation of large-scale discourse, overcoming limitations of retrieving only short text chunks.

RAPTOR
The mechanism of RAPTOR

Practical Implementation with Ollama and Open-Source Tools

Using tools available on GitHub, RAPTOR can be implemented with ease. The process involves converting data into JSON, creating embeddings, and organizing them into a tree structure stored in a vector store. Ollama provides the infrastructure to run these processes locally, ensuring privacy and security.

Step-by-Step: From RAG to RAPTOR

  1. Data Preparation: Convert data sources into JSON and load into LangChain.
  2. Text Splitting: Use recursive or semantic chunking to divide text into manageable pieces.
  3. Embeddings Creation: Generate embeddings using local models to ensure privacy.
  4. Tree Organization: Build a tree structure by summarizing data at each layer.
  5. Storage and Retrieval: Store the tree in a vector store for efficient retrieval.

Benefits of RAPTOR

RAPTOR’s tree-organized approach provides several benefits:

  • Enhanced Context: Captures larger discourse structure, improving response accuracy.
  • Efficient Retrieval: Tree structure enables faster and more relevant information retrieval.
  • Scalability: Suitable for handling large datasets with complex structures.

Conclusion

The evolution from RAG to RAPTOR marks a significant leap in AI retrieval capabilities. This progression addresses key challenges in AI responses, making it a valuable tool for developers and researchers.

So staying informed and engaged is crucial as we navigate this exciting era. With our platform, NextBrain AI, you can harness the full power of language models to effortlessly analyze your data and gain strategic insights. Schedule your demo today to see what AI can reveal from your data.

Logo NextBrain

We are on a mission to make NextBrain a space where humans work together with the most advanced algorithms to deliver superior game changing insight from data. We love No-code Machine Learning

Offices

Europe
Paseo de la Castellana, n.º 210, 5º-8
28046 Madrid, Spain
Phone number: spain flag +34 91 991 95 65

Australia
Level 1, Pier 8/9,23 Hickson Road
Walsh Bay, NSW, 2000
Phone number: spain flag +61 410 497229

Open hours (CET)

Monday—Thursday: 8:00AM–5:30PM
Friday: 8:00AM–2:00PM


EMEA, America

Live chat support
Contact our Sales Team