跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

RIMS:面向小型語言模型檢索增強生成的平滑多對偏好最佳化框架

文章摘要

小型語言模型在檢索增強生成中易受噪聲證據影響。本文提出RIMS,一種三階段偏好最佳化框架,包括合成思維鏈偏好資料生成、可微平滑聚合機制和偏好最佳化。實驗表明其在多跳問答基準上優於現有方法。

來源arXiv Computational Linguistics作者: Pei Tian, Zihan Dong, Tianci Liu, Linjun Zhang, Haoyu Wang
RIMS:面向小型語言模型檢索增強生成的平滑多對偏好最佳化框架
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

在資源受限的環境中,小型語言模型(SLM)因其低計算需求而成為檢索增強生成(RAG)的理想選擇。然而,這些模型容量有限,對檢索到的噪聲或虛假證據高度敏感。現有的基於偏好的方法,如RoseRAG,僅透過硬argmin/argmax選擇最難的單個偏好對,丟棄了其餘訊號;另一些方法則將多個偏好對視為獨立的二分類,導致資料利用率低下。

為此,研究團隊提出了RIMS,一種三階段偏好最佳化框架。第一階段透過拒絕取樣生成合成思維鏈偏好資料,僅使用目標SLM本身,無需依賴專有模型。第二階段引入可微的軟聚合機制,用平滑運算元取代硬選擇,在保留間隔感知選擇的判別結構的同時,從所有偏好對中保留梯度訊號。第三階段將平滑目標應用於多種對齊演算法進行偏好最佳化。

理論上,作者證明了平滑近似具有可控的誤差界,且平滑聚合比硬選擇能提供更緊的梯度對齊。在四個多跳問答基準上的實驗表明,RIMS在多個SLM骨幹上均優於最先進的基線,在噪聲檢索條件下在Exact Match和F1上取得了一致的提升。程式碼已開源。

這項工作的核心貢獻在於提出了一種資料高效的偏好最佳化方法,使得小型模型在複雜檢索場景中能夠更好地利用檢索結果,同時避免了昂貴的專有模型依賴。該框架的平滑聚合機制是一個重要創新,它透過保留所有偏好對的資訊,顯著提升了訓練的穩定性和最終效能。團隊還提供了理論證明和開源實現,為後續研究奠定了堅實基礎。

展開要點與分析

文章情報

工程師進階

要點

  • 提出RIMS框架,透過平滑多對聚合最佳化小型語言模型的偏好學習
  • 使用目標模型自身進行拒絕取樣生成合成偏好資料,不依賴專有模型
  • 可微平滑聚合機制保留所有偏好對的梯度訊號,優於硬選擇
  • 在四個多跳問答基準上取得一致提升

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。