随着大语言模型(LLM)技术的发展,RAG(Retrieval Augmented Generation)技术得到了广泛探讨和研究,越来越多的高级 RAG 检索方法也随之被人发现,相对于普通的 RAG 检索,高级 RAG 通过更深化的技术细节、更复杂的搜索策略,提供出了更准确、更相关、更丰富的信息检索结果。本文首先讨论这些技术,国产 向量数据库并基于 Milvus 给出一个实现案例。
; [7 v' k) {6 z
* n' Y$ y. S2 \# `4 q; ]8 [7 S1 c$ q" S) d
01.初级 RAG
+ X% n0 ]- ? H5 [2 r4 ^# L5 c J( d5 ]% t' L2 a% V
# H2 t5 l# l) t% s
初级 RAG 的定义! ~! ~, Z, \. Q N
) U0 k# C- d8 b
$ w- f) y/ X* n8 w9 o; V
初级 RAG 研究范式代表了最早的方法论,在 ChatGPT 广泛采用后不久就取得了重要地位。初级 RAG 遵循传统的流程,包括索引创建(Indexing)、检索(Retrieval)和生成(Generation),常常被描绘成一个“检索—读取”框架,其工作流包括三个关键步骤:& a+ y# k2 H! O
4 ~/ Z7 m# ?, K8 E3 s$ M
Y p0 x. c) m. e* R 语料库被划分为离散的块,然后使用编码器模型构建向量索引。6 K6 X. }# X# j, _* R# e
2 d h/ p8 A' @7 C8 i
; h3 U I0 ^: G: h1 b
RAG 根据查询与索引块(Indexed Chunk)的向量相似度识别并对块进行检索。
& ^, O; j. e2 r; J& b
; e$ F- S+ M) l# c3 u
" v8 d1 m0 x: Z" { i$ G 模型根据检索块(Retrieved Chunk)中获取的上下文信息生成答案。) M7 b+ ~6 S5 n& d# t$ [0 K- Z
3 q: W# r3 z& \# [( h. b( [4 L4 A
初级 RAG 的局限性7 E# Y. Y6 Q1 J- ?
9 r; W F. k4 F- c! c8 J6 e4 c4 s/ O9 a, A1 W/ H* X
初级 RAG 在三个关键领域面临着显著挑战:"检索"、"生成"和"增强"。# V- u) | y+ h& ~
' b' \2 ~' W* M9 y; c
6 ~) L$ q# x8 `1 u
初级 RAG 的检索质量存在许多问题,例如低精度和低召回率等。低精度会导致检索到的块无法对齐,以及幻觉等潜在问题。低召回率会导致无法检索到所有的相关块,从而导致LLM的回复不够全面。此外,使用老旧的信息进一步加剧了问题,可能导致不准确的检索结果。
4 R! R0 T! u+ X: t) S$ h; c
5 P1 g% @ i$ J+ u' U5 n5 C
; Z" T- e5 y' w 生成回复质量面临着幻觉挑战,即 LLM 生成的答案并没有基于所提供的上下文,和上下文不相关,或者生成的回复存在着包含有害或歧视内容的潜在风险。9 P6 T* M T# z
+ r+ o( n6 H! E7 X7 M. x1 U& z* ?
. x. R7 ~2 W, J/ X4 d 在增强过程中,初级 RAG 在如何有效地将检索到的段落的上下文与当前生成任务进行整合方面也面临着不小的挑战。低效的整合可能导致输出不连贯或破碎化。冗余和重复也是一个棘手的问题,特别是当多个检索到的段落包含相似信息时,生成的回复中可能会出现重复的内容。
$ u, M$ {: v/ J- p/ q1 ~0 O5 D7 t0 x% i
9 Z: _) r2 i# ]$ [2 a V+ K
02.高级 RAG4 o( I: n0 [+ _& x. Q
6 k5 }7 d( i5 z% g7 C9 `+ _% o3 e2 Q5 b, [
为了解决初级 RAG 的不足,高级 RAG 诞生了,并有针对性地进行了功能增强。首先讨论这些技术,这些技术可被归类为检索前优化、检索中优化和检索后优化。, a( m$ l) z% U, m/ D$ H
) c- g. y3 L5 G1 q- A2 k
" I1 @4 ^' J+ U% t, o! Z( n- k" o: Y
检索前优化4 _/ U. u( @8 Q4 ]+ p
5 x. |$ p7 L. ?0 @. j8 n7 ]/ M& R
* L+ r3 X3 f2 _8 K; G1 _( ? 检索前的优化关注数据索引优化以及查询优化,数据索引优化技术旨在以提高检索效率的方式存储数据:4 i# C6 p. m2 S5 ^3 v8 L ]
3 D4 D% F3 e* o
: N' r% `( g$ {9 v1 Y 滑动窗口:在数据块之间使用重叠,这是最简单的技术之一。+ \, v) q# I9 Y* ?
" }# C& ]; F0 Y: |
: S1 Y" d% T* S* V& k5 v 增强数据粒度:应用数据清理技术,例如移除不相关信息、确认事实准确性、更新过时信息等。
- s4 W. a$ D3 J/ c6 t: U" y
4 i6 r8 T4 P% j) \" c. |4 ]" R: {5 J. M Q0 o+ }. n
添加元数据:如用于过滤的日期、目的或章节信息等。
3 Y9 C0 {, c' v% C2 R) E0 k* m; k" d" y, F4 |9 F
. n" W1 c0 h4 ]3 Y 优化索引结构涉及不同的数据索引策略:如调整块大小或使用多索引策略。本文我们将实现的一种技术是句子窗口检索,它在检索时嵌入单个句子,并在推断时用更大的文本窗口替换它们。
4 {6 _% |7 \3 a$ |, u$ A3 a X
& y+ P9 L4 p0 v6 I' d
, S2 `* g- g* G 检索中优化0 k5 k2 ?) X: d
- V3 w: b, t0 b. _% w
7 k4 C: M. [ D+ S) z- H
检索阶段主要是识别最相关的上下文。通常,检索是基于向量搜索,它计算查询和索引数据之间的语义相似性。因此,大多数检索优化技术都围绕着 embedding 模型:
9 r6 L: W1 D+ v3 u; M
" F* Q2 @; E5 ]( c/ q
! k+ p& p/ m+ c" d- ?5 k' }0 S5 Z 微调 embedding 模型:定制化 embedding 模型到特定领域上下文,特别是针对具有发展性或罕见术语的领域。例如,BAAI/bge-small-en 是一个高性能 embedding 模型,可以进行微调。
7 ~. u1 }9 F/ b1 C
( s T. y; y( C' D" L( T+ Y9 S8 S. W0 a6 c7 T0 \) ^$ I
动态 embedding:适应词语使用中的上下文,不同于使用每个词一个向量的静态embedding。例如,OpenAI 的 embeddings-ada-02 是一个复杂的动态embedding模型,捕捉到上下文理解。除了向量搜索之外,还有其他检索技术,如混合搜索(hybrid search),通常指的是将向量搜索与基于关键词的搜索相结合的概念。如果检索需要精确的关键词匹配,这种检索技术很有益处。
" W# U1 Y# G( J4 L; G. @4 t! t6 k+ ~6 k9 P% ?" s
|