尽管RAG技术在 LLMs 中具有潜力,但在文本分块方面常常被忽视。文本分块的质量直接影响知识密集型任务的表现。本文提出Meta-Chunking概念,这是一种介于句子和段落之间的文本分割技术,旨在通过逻辑感知来提高文本分割的效率;设计了两种基于LLMs的分块策略:边际采样分块(Margin Sampling Chunking)和困惑度分块(Perplexity Chunking)。此外,还提出了结合Meta-Chunking与动态合并的策略,在不牺牲逻辑连贯性的前提下,实现细粒度与粗粒度文本分块之间的平衡。
基本概念
元分块(Meta-Chunking)的概念,用于解决RAG系统中文本分块的问题。具体来说,元分块是指在段落内的一组句子,这些句子之间具有深度的语言逻辑联系。为了实现元分块,论文设计了两种基于LLMs的策略:边际采样分块(Margin Sampling Chunking)和困惑度分块(Perplexity Chunking)。具体来说:
-
Meta-Chunking: Meta-Chunking是指在段落内,由具有深度语言逻辑连接的句子集合构成的文本分块。这种分块方式旨在增强文本分割过程中的逻辑连贯性。
-
边际采样分块: 通过LLMs进行二元分类,判断连续句子是否需要分块,基于边际采样的概率差异做出决策。具体公式如下:
其中,表示分块判断的二元决策,表示形成指令,包含单个或多个句子。
通过模型M获得的概率,可以计算出两个选项的概率差。通过与阈值对比,得出是否分块的结论。
-
困惑度分块: 通过分析每个句子的困惑度(PPL)特征来精确识别文本分块边界。具体公式如下:
困惑度分块公式如下:
其中 表示句子 中的总词数; 表示句子 中的第 个词; 表示句子 中所有在 之前的词; 表示基于前 个词和所有在 之前的词计算得到的第 个词的困惑度
通过分析的分布特征,特别是识别最小值点,来确定潜在的分块边界。
- 动态合并策略: 为了应对不同文本的复杂性,论文提出了一种结合元分块和动态合并的策略,以实现细粒度和粗粒度文本分块的平衡。具体步骤包括:设置初始阈值并进行元分块操作,然后根据用户指定的分块长度迭代合并相邻的元分块,直到总长度满足或接近要求。
实验设计
-
数据集: 实验在四个基准数据集上进行,包括中文和英文文本,涵盖简短到长篇文档。数据集包括2WikiMultihopQA、Qasper、MultiFieldQA、MultiHop-RAG等。
-
实验设置:
-
使用Qwen2-1.5B、Internlm2-1.8B、Baichuan2-7B和Qwen2-7B等模型进行元分块。
-
探索了较小模型(如Pythia-0.16B、Pythia-0.41B和Qwen2-0.5B)的适用性和性能。
-
对于较长的文本,采用KV缓存机制在计算PPL时保持句子间的逻辑连贯性,同时避免GPU内存溢出。
-
实验在NVIDIA H800上进行文本分割,在NVIDIA GeForce RTX 3090上进行评估。
-
评估指标: 实验使用了七个关键指标进行评估,包括答案的正确性(如BLEU系列指标、ROUGE-L、BERTScore)、事实性(如F1值)和检索文本的召回率(如Hits@系列指标、MAP@10、MRR@10)。这些指标全面评估了元分块方法在不同任务和文本类型上的性能。
结果与分析
- 主要结果: 与基线方法相比,Meta-Chunking在多个数据集上显著提高了单跳和多跳问答任务的性能。例如,在2WikiMultihopQA数据集上,Meta-Chunking比相似性分块提高了1.32的F1值,而时间消耗仅为45.8%。
- 效率与准确性权衡: 边际采样分块方法在使用整个文本块时可以提高性能,但也会增加数据处理时间。相比之下,困惑度分块方法不仅在保持或接近边际采样分块方法的性能水平方面表现出色,还显著提高了处理效率。
- 小模型的性能: 尽管小模型在处理效率上有优势,但在跨语言适应性和性能上存在局限。中等大小的模型(如1.5B参数级别的模型)在性能和效率之间表现出更平衡的性能。
-
重叠分块策略的影响: PPL分块策略在多跳问答场景中表现尤为突出,特别是在单跳任务中,PPL分块策略在所有模型上均实现了2%-3%的性能提升。
-
分块方法的比较: PPL分块方法在BLEU系列指标和ROUGE-L上显著优于传统分块方法,表明其提高了生成文本的准确性和流畅性。
- 重排策略的影响: PPL分块与PPL重排结合的方法在所有指标上均表现最佳,特别是在Hits@8指标上,PPL重排方法在原始分块基础上提高了1.42%,而在PPL分块基础上提高了3.59%。
结论
本文提出了Meta-Chunking概念及其实现策略,显著提高了文本分块的效率和准确性。通过结合Meta-Chunking与动态合并策略,能够灵活应对多样化的分块需求。实验结果表明,Meta-Chunking在多个数据集上优于传统的分块方法,并且在性能、时间和计算成本之间取得了更好的平衡。
如何学习大模型 AI ?
由于新岗位的生产效率,要优于被取代岗位的生产效率,所以实际上整个社会的生产效率是提升的。
但是具体到个人,只能说是:
“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。
这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。
我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。
我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。
第一阶段(10天):初阶应用
该阶段让大家对大模型 AI有一个最前沿的认识,对大模型 AI 的理解超过 95% 的人,可以在相关讨论时发表高级、不跟风、又接地气的见解,别人只会和 AI 聊天,而你能调教 AI,并能用代码将大模型和业务衔接。
- 大模型 AI 能干什么?
- 大模型是怎样获得「智能」的?
- 用好 AI 的核心心法
- 大模型应用业务架构
- 大模型应用技术架构
- 代码示例:向 GPT-3.5 灌入新知识
- 提示工程的意义和核心思想
- Prompt 典型构成
- 指令调优方法论
- 思维链和思维树
- Prompt 攻击和防范
- …
第二阶段(30天):高阶应用
该阶段我们正式进入大模型 AI 进阶实战学习,学会构造私有知识库,扩展 AI 的能力。快速开发一个完整的基于 agent 对话机器人。掌握功能最强的大模型开发框架,抓住最新的技术进展,适合 Python 和 JavaScript 程序员。
- 为什么要做 RAG
- 搭建一个简单的 ChatPDF
- 检索的基础概念
- 什么是向量表示(Embeddings)
- 向量数据库与向量检索
- 基于向量检索的 RAG
- 搭建 RAG 系统的扩展知识
- 混合检索与 RAG-Fusion 简介
- 向量模型本地部署
- …
第三阶段(30天):模型训练
恭喜你,如果学到这里,你基本可以找到一份大模型 AI相关的工作,自己也能训练 GPT 了!通过微调,训练自己的垂直大模型,能独立训练开源多模态大模型,掌握更多技术方案。
到此为止,大概2个月的时间。你已经成为了一名“AI小子”。那么你还想往下探索吗?
- 为什么要做 RAG
- 什么是模型
- 什么是模型训练
- 求解器 & 损失函数简介
- 小实验2:手写一个简单的神经网络并训练它
- 什么是训练/预训练/微调/轻量化微调
- Transformer结构简介
- 轻量化微调
- 实验数据集的构建
- …
第四阶段(20天):商业闭环
对全球大模型从性能、吞吐量、成本等方面有一定的认知,可以在云端和本地等多种环境下部署大模型,找到适合自己的项目/创业方向,做一名被 AI 武装的产品经理。
- 硬件选型
- 带你了解全球大模型
- 使用国产大模型服务
- 搭建 OpenAI 代理
- 热身:基于阿里云 PAI 部署 Stable Diffusion
- 在本地计算机运行大模型
- 大模型的私有化部署
- 基于 vLLM 部署大模型
- 案例:如何优雅地在阿里云私有部署开源大模型
- 部署一套开源 LLM 项目
- 内容安全
- 互联网信息服务算法备案
- …
学习是一个过程,只要学习就会有挑战。天道酬勤,你越努力,就会成为越优秀的自己。
如果你能在15天内完成所有的任务,那你堪称天才。然而,如果你能完成 60-70% 的内容,你就已经开始具备成为一名大模型 AI 的正确特征了。