KeyFrame內部研究專用

AI Agent的RAG与长期记忆是如何实现的?

白白说大模型·7月21日週二·14 min中文

三句話摘要

向量搜索如何讓 AI 在千維空間中快速檢索相似信息,成為大模型記憶與理解的底層基礎。 多維向量空間與向量檢索演算法是現代 AI 理解世界、召回記憶、無幻覺回答的統一底座。 多維空間的語義表示:計算機無法直接認知文字、圖片,需透過 embedding 模型把任何內容轉換為超高維向量。向量間的距離越近,語義相似度越高,這是所有向量檢索的基礎。

重點整理

重點
  • 1

    多維空間的語義表示:計算機無法直接認知文字、圖片,需透過 embedding 模型把任何內容轉換為超高維向量。向量間的距離越近,語義相似度越高,這是所有向量檢索的基礎。

  • 2

    HNSW 分層導航:採用金字塔式架構(高層高速通道、底層街道),不需逐一比對所有資料,而是順著鄰居網路跳躍,複雜度從線性降至對數級,實現幾毫秒內的極速檢索。

  • 3

    IVF + PQ 雙重壓縮:IVF 先透過聚類把向量分類到不同文件夾,將計算量砍掉 90%;PQ 再把單一向量切段編碼,體積壓縮 8 倍以上,解決超大規模資料的記憶體溢出問題。

  • 4

    非對稱距離計算:查詢向量保持原始高精度,資料庫向量保持壓縮狀態,用預算法表進行快速查表比對,避免高成本的浮點運算。

實用技巧與重點

乾貨
  • 三大核心演算法:HNSW (Hierarchical Navigable Small World)、IVF (Inverted File Index)、PQ (Product Quantization)
  • 壓縮效果:PQ 可將 16 位元組向量壓縮至 2 位元組(體積縮小 8 倍)
  • 搜尋效率:從暴力搜尋的線性時間降至 HNSW 的對數級;IVF 可砍掉 90% 計算量
  • 應用系統:Milvus、Qdrant 等向量資料庫,提供企業級的分散式管理和混合查詢
  • 應用場景:RAG(檢索增強生成)、AI Agent 長期記憶、多模態語義檢索

結論

結論

多維向量空間與向量檢索演算法是現代 AI 理解世界、召回記憶、無幻覺回答的統一底座。

完整解析

詳細

人類透過眼睛、耳朵、觸覺認識三維世界,但計算機只懂數字。要讓機器理解蘋果和橘子的區別,需要先用特徵向量化。譬如用「重量、顏色、表皮粗糙度」三個維度描述,兩個水果在三維空間中就能被明確拉開距離。實務上,大模型的 embedding 層會提取數百至數千個隱性特徵維度,這些無法用語言直觀解釋,但在多維空間中的相對幾何位置決定了語義相似性。距離越近,相似度越高。

面對上億筆向量資料,暴力搜尋需要分鐘級耗時,完全不可行。HNSW 演算法巧妙地借鑒「六度分隔」理論,把空間中距離近的向量用邊連接成鄰居網路。檢索時不逐一比對,而是隨機選個起點,順著網路連線跳向目標方向。關鍵創新是分層設計:最上層像高速公路(節點少、大幅移動)快速定位大區域,中層是省道,最底層是街道包含全部資料。讀寫分離也很重要:寫入時建立連線改變圖結構,查詢時只讀不寫,確保效率。

然而 HNSW 的複雜網路和指標必須全部常駐記憶體,資料量達億級時伺服器記憶體會爆掉。為此引入 IVF 和 PQ 兩大壓縮技術。IVF(倒排文件)的邏輯像商場導購牌:先用 K-Means 將雜亂向量聚類成若干文件夾,每個文件夾有個代表中心。查詢時先找距離最近的文件夾,再在該文件夾內局部計算距離,計算量可砍 90% 以上。PQ(乘積量化)則把一個長向量切成多段,各段分別找聚類中心,用密碼本編號代替原值。一個 16 位元組向量可壓至 2 位元組,體積銳減 8 倍。非對稱距離計算進一步優化:查詢向量保持高精度,資料向量保持壓縮,用預算法表進行查表加法,避免昂貴的浮點運算。

這些底層演算法沒有完美方案,都是工程權衡。HNSW 快但受記憶體限制,IVF 省空間但邊界位置易漏解,PQ 壓縮率高但必然喪失部分精度。向量檢索的核心法則是允許極小誤差,換取百倍效率提升。

在應用層,Milvus、Qdrant 等向量資料庫把底層演算法封裝成完整系統,提供資料管理、混合篩選、分散式高可用。RAG(檢索增強生成)把用戶提問向量化,在資料庫中找最相關段落,併入 prompt 注入大模型,讓模型基於真實資料精準回答,消除幻覺。AI Agent 的長期記憶也依賴同樣機制:歷史對話不實時轉換為向量寫入資料庫,新對話時快速喚醒最相關的記憶片段,實現連貫的多輪交互。

關鍵時刻

Pipeline v2

帶時間戳的重點,會在逐字稿層級分析上線後產生。目前請先透過原始影片觀看。

事實查核

Pipeline v2

說法查證是下一次管線升級的一部分。KeyFrame 只會顯示它真正能驗證的內容。

更多「AI 技術」的內容

🔬“We have foundation models for language, not for physics” — Anima Anandkumar, Bren Professor of Computing
編輯精選
83 min
AI 技術英文PODCAST8月26日

🔬“We have foundation models for language, not for physics” — Anima Anandkumar, Bren Professor of Computing

Latent Space

  • 加速傳統物理模擬的典範轉移:氣象科學家原本認為 AI 無法匹敵數十年的物理建模工作,但傅里葉神經算子不僅達到同等精度,還快了一萬多倍。原本需要超級計算機的計算現在用消費級 GPU 就能完成,這改變了整個領域的思維方式。
  • 傅里葉域的非局部現象捕捉:傅里葉域能有效表示非局部現象(如大氣河流跨越千里的影響),且計算複雜度為準線性,遠優於完全連接的全局模型。這特別適合流體動力學、量子化學等自然現象中普遍存在的非局部相互作用。
  • 多解析度連續函數表示:神經算子將輸入輸出視為連續函數而非固定維度向量,可在推論時以任意解析度查詢,並能在更高解析度上疊加物理約束或額外數據,克服了固定解析度神經網路的限制。
Between Two Nerds: Attribution is dead, long live attribution
32 min
AI 技術英文PODCAST8月25日

Between Two Nerds: Attribution is dead, long live attribution

Risky Business

  • 工具成本的破壞性下降 — 傳統上,攻擊者必須重複使用昂貴自製的惡意軟體或工具組,因為開發和維護成本極高。這種成本結構使得安全研究人員可以通過工具特徵和程式碼簽名來追蹤攻擊者。LLM 自動化了代碼生成與維護工作流,使得攻擊者可以輕易為每個目標生成新工具,或改用通用系統工具,導致傳統的工具特徵分析失效。
  • 所有取證證據都在攻擊者掌控之中 — 無論是使用的 IP 位址、惡意軟體類型或戰術流程,這些都是攻擊者的主動選擇。即使看似是隨機巧合,攻擊者仍有能力在事前決定留下什麼痕跡。因此,所有可恢復的取證證據本質上都是攻擊者願意暴露的信息。
  • LLM 削弱工具簽名但保留高階行為特徵 — LLM 經過公開駭客技術訓練,使不同使用者產生相似的攻擊模式。然而,勒索軟體集團、國家級行為者的受害者選擇、贖金要求方式或目標模式等高階特徵仍具有識別價值。例如,鎖定加密交易所的攻擊幾乎只能指向朝鮮。
Why the Next AI Breakthrough May Come from Physics with Max Welling - #774
55 min
AI 技術英文PODCAST8月25日

Why the Next AI Breakthrough May Come from Physics with Max Welling - #774

TWIML AI

  • 多層篩選的材料設計流程:先搜尋文獻資料庫找現有材料,若無合適的就用生成模型產生數十萬個候選分子,用機器學習力場進行分子動力學模擬篩選,再進行實驗驗證。這套流程相比傳統量子力學計算能加速效率數個數量級。
  • 生成AI與熱力學的數學等價性:資訊論是兩個領域的共同基礎,生成模型的擴散過程與非平衡統計力學描述資訊損失的過程在數學上完全對應,許多開發出來的方法工具在兩領域都有精確對應的形式。
  • 基礎模型的遷移學習策略:先在廣泛材料資料集上訓練基礎力場表示,再針對特定材料類別進行蒸餾微調,既能保持計算效率也能獲得專一性。