iT邦幫忙

2026 iThome 鐵人賽

DAY 9
0
AI Engineering

企業知識檢索與 RAG 工程:標註、微調、混合檢索、Rerank 與可重現評測系列 第 9 篇

Day 09|Dense Retrieval Baseline:Embedding 真的懂你的 Domain 嗎

  • 分享至 

  • xImage
  •  

完成「BM25 Baseline:先尊重一個很難被淘汰的對手」後,下一個問題是「Dense Retrieval Baseline:Embedding 真的懂你的 Domain 嗎」能否重跑、否定並留下失敗原因。這就是今天的範圍。

今天要回答的問題

以 multilingual embedding 建立向量 baseline,固定 index 與 retrieval config,量測品質與延遲。

工程邊界

Dense Retrieval 的問題不是「懂不懂語意」這麼簡單

Embedding 把 query 與 document 映射到向量空間,近似搜尋再找出相近候選。Domain shift、專有詞、縮寫與語言混用都可能讓 base embedding 的幾何關係不符合你的任務。

因此 baseline 要固定 normalization、distance、index 與 pooling,並保留 query-level nearest neighbors。看錯誤案例比只看向量分數更容易知道是否需要 fine-tune。

一個可以直接重現的起點

Dense baseline 的重點是固定條件

模型、normalize 方法、chunk 單位、index metric、Top-K 與查詢前處理都會影響結果。第一次 baseline 不求最好,求的是「任何人都能重跑出同一個數字」。同時記錄 retrieval latency,避免後面只追品質不看成本。

固定 Dense Retrieval 的關鍵參數

這裡以「固定 embedding、pooling、normalization 與 index 跑 dense baseline」為比較目標,只保留 baseline、candidate 與逐項差值;evaluate 再接到實際評測程式。

METRICS = ('quality metrics', 'latency', 'index size', 'nearest neighbors')
baseline = evaluate("baseline", metrics=METRICS)
candidate = evaluate("candidate", metrics=METRICS)

for metric in METRICS:
    delta = candidate[metric] - baseline[metric]
    print(f"{metric}: {delta:+.4f}")

驗證與落地方式

驗證可從「固定 embedding、pooling、normalization 與 index 跑 dense baseline」開始。比較前先凍結 corpus、query set、qrels、資料切分與評測程式,確保實驗只改動目標元件。若 index、候選數或前處理同時改變,最後的分數差異就無法歸因,容易把資料變動誤認成模型進步。

這篇優先比較:quality metrics、latency、index size、nearest neighbors。除了整體平均,也要保留逐題結果與 query 類型,分辨改善集中在哪些情境、哪些案例反而退步。品質指標還要和 latency、記憶體、索引大小或單次成本一起閱讀,才能判斷提升是否值得帶進 production。

常見誤判

  • 只公布最高分,沒有保留 baseline、負面結果與信賴區間,難以判斷改善是否穩定。
  • 在相同資料上反覆調參又回頭評分,test set 會逐漸變成隱性的 training set。
  • 只看 retrieval 或 generation 的單層指標,忽略錯誤可能沿 pipeline 傳遞,也可能被後段暫時掩蓋。

上線前檢查

每次實驗都應能追回資料版本、模型、index、config 與評測程式版本。上線前再以未參與調參的案例做一次回歸測試,並保留最差案例供 error analysis。若品質提升不足以抵銷延遲、成本或維護複雜度,維持較簡單的 baseline 通常更可靠。

如何閱讀結果

結果應先看逐題變化,再看整體分數。若平均值上升,卻只來自少數容易題,或關鍵 query 類型持續退步,仍不適合直接替換 baseline。每個明顯升降的案例都應回到候選文件、排名與輸入特徵,確認變化符合原先假設。

離線評測通過後,還要檢查線上條件:新文件加入後是否需要重建 index、cache 是否造成舊結果、查詢分布是否漂移。只有資料生命週期也能被管理,實驗中的品質提升才可能持續。

結果判讀

判讀不只看最大或最漂亮的數字。這一天真正要回答的是:檢查 domain shift,而不是只看 cosine 分數。

如果核心指標改善,但錯誤率、尾端延遲、資源成本或恢復能力變差,這代表取捨,而不是無條件進步。相反地,沒有改善也不是無效結果;至少能排除一條看似合理、實際上不值得增加複雜度的路。

今天的結論

今天的工程判斷是:檢查 domain shift,而不是只看 cosine 分數。無論結果支持或否定原本假設,都必須保留完整條件,才能和下一天的實驗串在一起。

下一篇將處理:BM25 為什麼可能打贏 Vector:第一輪 Error Analysis。


參考資料


上一篇
Day 08|BM25 Baseline:先尊重一個很難被淘汰的對手
下一篇
Day 10|BM25 為什麼可能打贏 Vector:第一輪 Error Analysis
系列文
企業知識檢索與 RAG 工程:標註、微調、混合檢索、Rerank 與可重現評測 共 12 篇
圖片
  熱門推薦
圖片
{{ item.channelVendor }} | {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言