從零構(gòu)建 RAG 框架的一些思考

最近剛從太初元碁舉辦的「Teco-RAG 大模型開(kāi)發(fā)任務(wù)挑戰(zhàn)賽」出來(lái),很遺憾沒(méi)有獲獎(jiǎng)。上圖是前 5 名的初賽成績(jī)锭硼,下圖是前 30 名的初賽成績(jī),我排 28 名蜕劝,60.6 分檀头,優(yōu)秀獎(jiǎng)的獲獎(jiǎng)線是 64.4 分,可以看出前 3 名的分?jǐn)?shù)是有斷層的岖沛,分?jǐn)?shù)集中在 60-69 的區(qū)間內(nèi):

這個(gè)比賽還是很有實(shí)踐意義的暑始,因?yàn)樗年P(guān)鍵點(diǎn)在于使用的是去年發(fā)布的 Qwen-7B-Chat,沒(méi)錯(cuò)婴削,不是 Qwen廊镜,也不是 Qwen2.5,而是去年發(fā)布的 Qwen唉俗!這個(gè)比賽的目的是為了探索通過(guò)工程化的方案使用低成本的開(kāi)源模型開(kāi)發(fā) RAG嗤朴,它的基座模型的能力相對(duì)于現(xiàn)在最新的模型差距較大,需要用工程化方案進(jìn)行約束虫溜。

而我做出的優(yōu)化只有兩點(diǎn):

  1. 修改 text_splitter 默認(rèn)參數(shù):chunk_size=640雹姊,chunk_overlap=64,small_chunk_size=192衡楞;
  2. 增加 BM25 檢索召回吱雏,使用 jieba 的搜索引擎模式進(jìn)行分詞;

這兩個(gè)優(yōu)化從最開(kāi)始的 40 分提高到了 60 分瘾境,不過(guò)當(dāng)時(shí)由于對(duì)混合多路檢索理解還不夠歧杏,在后續(xù)的學(xué)習(xí)中,我發(fā)現(xiàn)稀疏檢索 BM25 的檢索數(shù)量(即 top_k)應(yīng)該設(shè)置更大一些寄雀,與稠密檢索(Embedding 向量檢索)之間的比例應(yīng)該是 3: 1,我當(dāng)時(shí)在 BM25 設(shè)置的 top_k 是 5 條陨献,在驗(yàn)證很多答案的時(shí)候盒犹,發(fā)現(xiàn)沒(méi)有召回正確的文檔片段。

由于 RAG 是明年我的重點(diǎn)工作方向之一,所以我需要重新思考急膀。今年轉(zhuǎn)入大模型應(yīng)用開(kāi)發(fā)沮协,主要是 Agent、RAG 和 Fine-tuning 三個(gè)大方向卓嫂,現(xiàn)在對(duì) RAG 的理解還不夠深刻慷暂,因此準(zhǔn)備從零構(gòu)建一個(gè)完整的 RAG 框架,也就是 Build RAG from scratch step by step晨雳。

標(biāo)準(zhǔn)的 RAG 流程還是比較簡(jiǎn)單的:

之前粗略地看過(guò)一篇《四萬(wàn)字長(zhǎng)文巨獻(xiàn)行瑞,一文讀透RAG》的文章,當(dāng)然細(xì)節(jié)太多了餐禁,也沒(méi)有看完血久。RAG 應(yīng)用有一句話:「一周出 demo,半年用不好」帮非,整個(gè)流程實(shí)際上有很多可以優(yōu)化的地方:

這里面我在學(xué)習(xí)的時(shí)候氧吐,忽略了一個(gè)很關(guān)鍵的環(huán)節(jié),評(píng)估(Evaluation):不能在提交方案的時(shí)候才進(jìn)行評(píng)估末盔,事實(shí)上在每個(gè)新增功能和優(yōu)化策略的時(shí)候筑舅,都應(yīng)該進(jìn)行評(píng)估!在 RAG 開(kāi)發(fā)的過(guò)程中陨舱,需要經(jīng)常跑測(cè)試翠拣,得到類似如下圖片:

根據(jù) Anthropic 的 Introducing Contextual Retrieval 這篇文章,實(shí)際上很多優(yōu)化策略是無(wú)效的隅忿,如果沒(méi)有一個(gè)合理的評(píng)測(cè)流程心剥,會(huì)陷入大量的無(wú)用功。

這里我找到了 Ragas背桐, 一個(gè)專門針對(duì)于 LLM 應(yīng)用效果評(píng)估的庫(kù)优烧,根據(jù)文檔 Testset Generation for RAG 和論文 Judging LLM-as-a-Judge with MT-Bench and Chatbot Arena,以及「Teco-RAG 大模型開(kāi)發(fā)任務(wù)挑戰(zhàn)賽」的參賽經(jīng)歷链峭,我決定接下來(lái)的評(píng)估策略如下:

使用最新的大模型(比如 GPT-4o畦娄、Gemini-2.0-Flash、GLM-4-Plus 等)針對(duì)不同的文檔(金融弊仪、計(jì)算機(jī)等)生成評(píng)估測(cè)試集熙卡,使用較早版本的大模型(比如 Qwen-7B-Chat、Llama-2-7B-Chat励饵、GLM-4-Air 等)進(jìn)行 RAG 策略調(diào)試驳癌。

以上是我目前對(duì) RAG 框架的思考,我正在基于 RapidAI 的 Knowledge-QA-LLM 項(xiàng)目進(jìn)行開(kāi)發(fā)役听,歡迎加入 ROADMAP 討論:https://github.com/RapidAI/Knowledge-QA-LLM/pull/24颓鲜。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末表窘,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子甜滨,更是在濱河造成了極大的恐慌乐严,老刑警劉巖,帶你破解...
    沈念sama閱讀 221,576評(píng)論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件衣摩,死亡現(xiàn)場(chǎng)離奇詭異昂验,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)艾扮,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,515評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門既琴,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人栏渺,你說(shuō)我怎么就攤上這事呛梆。” “怎么了磕诊?”我有些...
    開(kāi)封第一講書人閱讀 168,017評(píng)論 0 360
  • 文/不壞的土叔 我叫張陵填物,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我霎终,道長(zhǎng)滞磺,這世上最難降的妖魔是什么? 我笑而不...
    開(kāi)封第一講書人閱讀 59,626評(píng)論 1 296
  • 正文 為了忘掉前任莱褒,我火速辦了婚禮击困,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘广凸。我一直安慰自己阅茶,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,625評(píng)論 6 397
  • 文/花漫 我一把揭開(kāi)白布谅海。 她就那樣靜靜地躺著脸哀,像睡著了一般。 火紅的嫁衣襯著肌膚如雪扭吁。 梳的紋絲不亂的頭發(fā)上撞蜂,一...
    開(kāi)封第一講書人閱讀 52,255評(píng)論 1 308
  • 那天,我揣著相機(jī)與錄音侥袜,去河邊找鬼蝌诡。 笑死,一個(gè)胖子當(dāng)著我的面吹牛枫吧,可吹牛的內(nèi)容都是我干的浦旱。 我是一名探鬼主播,決...
    沈念sama閱讀 40,825評(píng)論 3 421
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼九杂,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼颁湖!你這毒婦竟也來(lái)了代兵?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書人閱讀 39,729評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤爷狈,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后裳擎,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體涎永,經(jīng)...
    沈念sama閱讀 46,271評(píng)論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,363評(píng)論 3 340
  • 正文 我和宋清朗相戀三年鹿响,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了羡微。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,498評(píng)論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡惶我,死狀恐怖妈倔,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情绸贡,我是刑警寧澤盯蝴,帶...
    沈念sama閱讀 36,183評(píng)論 5 350
  • 正文 年R本政府宣布,位于F島的核電站听怕,受9級(jí)特大地震影響捧挺,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜尿瞭,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,867評(píng)論 3 333
  • 文/蒙蒙 一闽烙、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧声搁,春花似錦黑竞、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 32,338評(píng)論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至充石,卻和暖如春莫换,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背骤铃。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 33,458評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工拉岁, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人惰爬。 一個(gè)月前我還...
    沈念sama閱讀 48,906評(píng)論 3 376
  • 正文 我出身青樓喊暖,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親撕瞧。 傳聞我的和親對(duì)象是個(gè)殘疾皇子陵叽,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,507評(píng)論 2 359

推薦閱讀更多精彩內(nèi)容