RAG系统中的文本切片难题：15种Chunking策略全面解析，轻松搞定RAG！

不秃头de程序猿

于 2025-07-17 14:43:56 发布

阅读量313

点赞数 11

CC 4.0 BY-SA版权

文章标签： ai 大模型程序员转行学习产品经理人工智能

本文链接：https://blog.youkuaiyun.com/Android_XG/article/details/149421689

1、RAG系统也能“切块”？15种Chunking技巧让你的检索生成更聪明！

你知道吗？在构建一个强大的RAG（Retrieval-Augmented Generation）系统时，决定其“聪明程度”的，可能不是模型本身，而是——你怎么“切块”你的文档。

在NLP领域，RAG系统已经成为处理复杂问答、文档摘要、知识库检索等任务的利器。但面对动辄上万字的文档，如何在不丢失上下文的前提下，把它们“切”成模型能消化的“小块”，就成了关键。

今天这篇文章，我们就来系统聊聊：15种Chunking技巧，帮你打造一个既快又准的RAG系统。无论你是做问答系统、文档检索，还是构建企业知识库，这篇文章都值得你收藏。

01｜什么是Chunking？为什么它这么重要？

在RAG系统中，Chunking（切块）指的是：把大文档拆分成小块，以便模型更好地理解和检索信息。

你可能会问：为什么不能直接把整篇文档丢给模型？原因很简单：

大模型有token限制（比如GPT-4最多支持8K tokens）；
文档太长，模型容易“看漏”关键信息；
不切块，检索系统很难精准定位答案。

所以，Chunking不是简单的“切”，而是要在“保留上下文”和“适配模型能力”之间找到平衡。

02｜Chunking的三大核心考量

在正式介绍15种技巧之前，我们先来理解Chunking的三个关键因素：

1. 块的大小（Chunk Size）

太大：容易超token限制，检索慢；
太小：上下文丢失，生成质量差；
建议：根据模型token上限，控制在100~500 tokens之间。

2. 上下文保留（Context Preservation）

切块不能“断句断意”，否则模型会“看不懂”；
使用滑动窗口、语义切块等方式，能有效保留上下文。

3. 多模态处理（Modality Handling）

文档中可能包含表格、图片、代码块；
不同内容类型需要不同的切块策略。

03｜15种Chunking技巧全解析（附代码）

接下来，我们进入正题：15种Chunking技巧，每种都配有使用场景、优缺点和代码示例，建议收藏！

1. 固定大小切块（Fixed-Size Chunking）

原理：按固定词数或token数切分。

适用场景：结构简单的小文档。

优点：实现简单，速度快。

缺点：可能切断句子，丢失语义。

def fixed_size_chunk(text, max_words=100):
    words = text.split()
    return [' '.join(words[i:i + max_words]) for i in range(0, len(words), max_words)]

2. 句子切块（Sentence-Based Chunking）

原理：按句子边界切分。

适用场景：需要保留语义完整性的文档。

优点：语义清晰，上下文连贯。

缺点：句子长度不一，chunk大小不稳定。

import spacy
nlp = spacy.load("en_core_web_sm")

def sentence_chunk(text):
    doc = nlp(text)
    return [sent.text for sent in doc.sents]

3. 段落切块（Paragraph-Based Chunking）

原理：按段落切分。

适用场景：结构清晰的文档，如论文、报告。

优点：自然分段，语义完整。

缺点：段落长度不一，可能超token限制。

def paragraph_chunk(text):
    return text.split('\n\n')

4. 语义切块（Semantic Chunking）

原理：基于语义相似度进行切块。

适用场景：技术文档、复杂文本。

优点：上下文保留好。

缺点：实现复杂，需依赖模型。

def semantic_chunk(text, max_len=200):
    doc = nlp(text)
    chunks = []
    current_chunk = []
    for sent in doc.sents:
        current_chunk.append(sent.text)
        if len(' '.join(current_chunk)) > max_len:
            chunks.append(' '.join(current_chunk))
            current_chunk = []
    if current_chunk:
        chunks.append(' '.join(current_chunk))
    return chunks

5. 模态感知切块（Modality-Specific Chunking）

原理：文本、表格、图片分别处理。

适用场景：PDF、技术手册等混合内容文档。

优点：保留多种模态信息。

缺点：实现复杂。

def modality_chunk(text, images=None, tables=None):
    text_chunks = paragraph_chunk(text)
    return {'text_chunks': text_chunks, 'images': images, 'tables': tables}

6. 滑动窗口切块（Sliding Window Chunking）

原理：相邻chunk之间有重叠。

适用场景：法律、学术文档。

优点：上下文连贯。

缺点：内容重复，处理量大。

def sliding_window_chunk(text, chunk_size=100, overlap=20):
    tokens = text.split()
    chunks = []
    for i in range(0, len(tokens), chunk_size - overlap):
        chunk = ' '.join(tokens[i:i + chunk_size])
        chunks.append(chunk)
    return chunks

7. 层级切块（Hierarchical Chunking）

原理：按章节、段落、子段落分层切块。

适用场景：结构化文档，如论文、合同。

优点：保留文档结构。

缺点：实现复杂。

def hierarchical_chunk(text, section_keywords):
    sections = []
    current_section = []
    for line in text.splitlines():
        if any(keyword in line for keyword in section_keywords):
            if current_section:
                sections.append("\n".join(current_section))
            current_section = [line]
        else:
            current_section.append(line)
    if current_section:
        sections.append("\n".join(current_section))
    return sections

8. 内容感知切块（Content-Aware Chunking）

原理：根据内容特征动态调整切块策略。

适用场景：电子书、技术文档。

优点：灵活适应不同内容。

缺点：逻辑复杂。

def content_aware_chunk(text):
    chunks = []
    current_chunk = []
    for line in text.splitlines():
        if line.startswith(('##', '###', 'Introduction', 'Conclusion')):
            if current_chunk:
                chunks.append('\n'.join(current_chunk))
            current_chunk = [line]
        else:
            current_chunk.append(line)
    if current_chunk:
        chunks.append('\n'.join(current_chunk))
    return chunks

9. 表格感知切块（Table-Aware Chunking）

原理：将表格独立切块。

适用场景：财务报表、技术文档。

优点：保留表格结构。

缺点：格式可能丢失。

import pandas as pd

def table_aware_chunk(table):
    return table.to_markdown()

10. Token级切块（Token-Based Chunking）

原理：按token数切块，适配Transformer模型。

适用场景：GPT、BERT等模型。

优点：适配模型限制。

缺点：可能切断句子。

from transformers import GPT2Tokenizer

tokenizer = GPT2Tokenizer.from_pretrained("gpt2")

def token_based_chunk(text, max_tokens=200):
    tokens = tokenizer(text)["input_ids"]
    chunks = [tokens[i:i + max_tokens] for i in range(0, len(tokens), max_tokens)]
    return [tokenizer.decode(chunk) for chunk in chunks]

11. 实体感知切块（Entity-Based Chunking）

原理：基于NER识别实体进行切块。

适用场景：简历、合同、法律文档。

优点：保留实体信息。

缺点：需训练NER模型。

def entity_based_chunk(text):
    doc = nlp(text)
    return [ent.text for ent in doc.ents]

12. 主题切块（Topic-Based Chunking）

原理：使用LDA等主题模型进行切块。

适用场景：新闻、研究论文等多主题文档。

优点：按主题聚合信息。

缺点：需额外建模。

from sklearn.feature_extraction.text import CountVectorizer
from sklearn.decomposition import LatentDirichletAllocation

def topic_based_chunk(text, num_topics=3):
    sentences = text.split('. ')
    vectorizer = CountVectorizer()
    sentence_vectors = vectorizer.fit_transform(sentences)
    lda = LatentDirichletAllocation(n_components=num_topics, random_state=42)
    lda.fit(sentence_vectors)
    # 省略主题分配逻辑
    return sentences

13. 页面切块（Page-Based Chunking）

原理：按PDF页面切块。

适用场景：PDF文档。

优点：实现简单。

缺点：可能断句。

def page_based_chunk(pages):
    return pages

14. 关键词切块（Keyword-Based Chunking）

原理：按关键词切分。

适用场景：结构清晰的文档。

优点：符合文档结构。

缺点：需预定义关键词。

def keyword_based_chunk(text, keywords):
    chunks = []
    current_chunk = []
    for line in text.splitlines():
        if any(keyword in line for keyword in keywords):
            if current_chunk:
                chunks.append('\n'.join(current_chunk))
            current_chunk = [line]
        else:
            current_chunk.append(line)
    if current_chunk:
        chunks.append('\n'.join(current_chunk))
    return chunks

15. 混合切块（Hybrid Chunking）

原理：结合多种策略。

适用场景：复杂文档。

优点：灵活强大。

缺点：实现复杂。

def hybrid_chunk(text):
    paragraphs = paragraph_chunk(text)
    hybrid_chunks = []
    for paragraph in paragraphs:
        hybrid_chunks += sentence_chunk(paragraph)
    return hybrid_chunks

04｜不同场景下如何选择Chunking策略？

场景类型	推荐策略
FAQ、客服系统	句子切块、关键词切块
学术论文	层级切块、语义切块
技术文档	表格感知切块、内容感知切块
多模态文档	模态感知切块、混合切块
法律文档	滑动窗口切块、实体感知切块