Skip to content

Modular RAG implementations: reranker, multi-retrievers #16

Closed

Description

@RobinQu

For better evaluation result in HF QA dataset.

  • Reranking: BCE, BGE-M3 scoring
  • Query rewrite
    • to generate SQL filter for given prompt
    • to generate hyperthecial queries

Activity

  1. converted this from a draft issue on Apr 3, 2024
  2. added this to the v0.1.2 milestone on Apr 3, 2024
  3. RobinQu commented on Apr 5, 2024

    @RobinQu
    OwnerAuthor
  4. RobinQu commented on May 20, 2024

    @RobinQu
    OwnerAuthor

    Rerank, Colbert, ...

    Related work

    Methods

    Opensourced Projects

    image

    image

    image

    Conlcusion

    • Both BCE and BGE family can be regarded as SOTA.
    • For multilanguage use case, valillan RAGatouille lags behind.
    • As Late-interaction models are faster in inference. bge-m3 is prefered as first ranking methods in RAG pipeline.
  5. modified the milestones: v0.1.2, v0.1.3 on May 21, 2024
  6. modified the milestones: v0.1.3, v0.1.4 on May 21, 2024
  7. RobinQu commented on May 21, 2024

    @RobinQu
    OwnerAuthor

    Timeline

    • BGE M3 in GGUF format #21
      • experiemntes with BERT + BGE M3 model for better understanding of forward pass, 3 days , 1 day.
      • bge-m3 inference code. A standalone repo may be needed. 7 days, 3 days.
    • First version of file-search tool for assistant-api #20
      • file and filebatch controllers, services, mappers. 3 days.
      • duckdb vectordb operator, retriever factory, 2 days
      • DuckDB: search with filter, delete with filter, 1 day
      • summarization pipeline for files, 2 days
      • file object handler, 2 days
      • file batch background jobs to update its progress, 1 day
      • search tool 3 days.
      • (optional) parallel react agent executor, 2 days
      • integration test 3 days.
      • eval test HF QA datast. 3 days.
  8. RobinQu commented on May 27, 2024

    @RobinQu
    OwnerAuthor

    OpenAI officials parameter for RAG: https://platform.openai.com/docs/assistants/tools/file-search/how-it-works

    By default, the file_search tool uses the following settings:
    Chunk size: 800 tokens
    Chunk overlap: 400 tokens
    Embedding model: text-embedding-3-large at 256 dimensions
    Maximum number of chunks added to context: 20 (could be fewer)

    Supported file formats: https://platform.openai.com/docs/assistants/tools/file-search/supported-files

    Image

  9. moved this from In Progress to Done in First GA Release of instinct.cppon Jun 14, 2024
  10. changed the title [-]Adanvanced RAG implementations[/-] [+]Modular RAG implementations: reranker, multi-retrievers[/+] on Jun 14, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions