Продвинутые методы разбивки

Retrieval Augmented Generation (RAG) с LangChain

Meri Nova

Machine Learning Engineer

Ограничения текущих стратегий разбивки

 

  1. 🤦 Разбивка наивна (без учёта контекста)

    • Игнорирует окружающий текст

  2. 🖇 Разбивка по символам, а не по токенам

    • Токены обрабатываются моделями
    • Риск превысить контекстное окно

 

→ SemanticChunker

 

→ TokenTextSplitter

Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

Символьный разделитель текста разбивает текст на фрагменты по количеству символов.

Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

Токенный разделитель текста разбивает текст на фрагменты по количеству токенов.

Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

Токены выделены, чтобы показать их соответствие значениям chunk_size и chunk_overlap.

Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

import tiktoken
from langchain_text_splitters import TokenTextSplitter
example_string = "Mary had a little lamb, it's fleece was white as snow."

encoding = tiktoken.encoding_for_model('gpt-4o-mini')
splitter = TokenTextSplitter(encoding_name=encoding.name,
                             chunk_size=10,
                             chunk_overlap=2)


chunks = splitter.split_text(example_string)

for i, chunk in enumerate(chunks):
    print(f"Chunk {i+1}:\n{chunk}\n")
Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

Chunk 1:
Mary had a little lamb, it's fleece

Chunk 2:
 fleece was white as snow.
Retrieval Augmented Generation (RAG) с LangChain

Разбивка по токенам

for i, chunk in enumerate(chunks):
    print(f"Chunk {i+1}:\nNo. tokens: {len(encoding.encode(chunk))}\n{chunk}\n")

Chunk 1:
No. tokens: 10
Mary had a little lamb, it's fleece was

Chunk 2:
No. tokens: 6
 fleece was white as snow.
Retrieval Augmented Generation (RAG) с LangChain

Семантическая разбивка

Абзац с предложением о RAG-приложениях и предложением о собаках.

Retrieval Augmented Generation (RAG) с LangChain

Семантическая разбивка

Абзац разбит по символам или токенам, из-за чего контекст потерян.

Retrieval Augmented Generation (RAG) с LangChain

Семантическая разбивка

Семантический разделитель разбил абзац в точке смены темы с RAG на собак.

Retrieval Augmented Generation (RAG) с LangChain

Семантическая разбивка

from langchain_openai import OpenAIEmbeddings
from langchain_experimental.text_splitter import SemanticChunker


embeddings = OpenAIEmbeddings(api_key="...", model='text-embedding-3-small')


semantic_splitter = SemanticChunker(
    embeddings=embeddings,

    breakpoint_threshold_type="gradient",
    breakpoint_threshold_amount=0.8

)
1 https://api.python.langchain.com/en/latest/text_splitter/langchain_experimental.text_splitter. SemanticChunker.html
Retrieval Augmented Generation (RAG) с LangChain

Семантическая разбивка

chunks = semantic_splitter.split_documents(data)
print(chunks[0])

page_content='Retrieval-Augmented Generation for\nKnowledge-Intensive NLP Tasks\ Patrick Lewis,
Ethan Perez,\nAleksandra Piktus, Fabio Petroni, Vladimir Karpukhin, Naman Goyal, Heinrich
Küttler,\nMike Lewis, Wen-tau Yih, Tim Rocktäschel, Sebastian Riedel, Douwe Kiela\nFacebook AI
Research; University College London;New York University;\[email protected]\nAbstract\nLarge
pre-trained language models have been shown to store factual knowledge\nin their parameters,
and achieve state-of-the-art results when fine-tuned on down-\nstream NLP tasks. However, their
ability to access and precisely manipulate knowl-\nedge is still limited, and hence on
knowledge-intensive tasks, their performance\nlags behind task-specific architectures.'
metadata={'source': 'rag_paper.pdf', 'page': 0}
Retrieval Augmented Generation (RAG) с LangChain

Давайте потренируемся!

Retrieval Augmented Generation (RAG) с LangChain

Preparing Video For Download...