跳到主要內容
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框架,通過平滑多對聚合優化小型語言模型的偏好學習
  • 使用目標模型自身進行拒絕採樣生成合成偏好數據,不依賴專有模型
  • 可微平滑聚合機制保留所有偏好對的梯度信號,優於硬選擇
  • 在四個多跳問答基準上取得一致提升

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