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

DynaMiCS:動態混合優化實現性能約束下的大語言模型多領域微調

文章摘要

DynaMiCS是一種動態混合優化器,將多領域微調轉化為約束優化問題,通過短領域探測估計跨領域影響矩陣,動態調整數據混合比例,在提升目標領域性能的同時確保約束領域(如通用知識、指令遵循、安全性)性能不下降,無需參考模型或手動調整權重。

DynaMiCS:動態混合優化實現性能約束下的大語言模型多領域微調
報告錯誤

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

查看更正說明
直接讀正文

在大型語言模型(LLM)的多領域微調中,一個核心挑戰是如何在提升目標領域(如特定任務或領域知識)性能的同時,保持模型在約束領域(如通用知識、指令遵循或安全性評估)上的能力。現有的數據混合策略通常依賴於固定啓發式規則或自適應算法,但這些方法無法明確保證約束領域的能力不會退化。針對這一問題,來自Apple和ETH Zurich的研究團隊提出了DynaMiCS(Dynamic Mixture with Constraints),一種動態混合優化器,將多領域微調重新建模為帶約束的優化問題。

DynaMiCS的核心創新在於每個更新步驟中執行短領域探測運行。具體而言,模型在少量樣本上對每個微調數據集進行快速訓練,並評估其對各評估領域的影響,從而估計一個局部跨領域效應的斜率矩陣。該矩陣量化了在每個數據集上訓練如何影響目標領域和約束領域的損失。基於這些估計,DynaMiCS通過概率單純形上的約束優化計算下一組混合權重,優化目標是在提升目標領域性能的同時,確保約束領域的損失不會超過預設的參考水平(例如,初始模型的損失)。

實驗在多種多領域微調場景中進行,涉及不同數量的目標領域和約束領域。結果表明,DynaMiCS在目標領域改進和約束滿足率上均顯著優於固定混合基線方法。例如,在同時優化數學推理和代碼生成(目標領域)並維持通用知識和安全性(約束領域)的任務中,DynaMiCS不僅取得了更高的目標領域準確率,還成功避免了約束領域的性能下降。此外,DynaMiCS的計算成本更低,因為它無需依賴參考模型、逐樣本評分或手動調節混合權重,僅通過短探測運行即可動態調整混合比例。

該方法的提出為大規模語言模型的精細微調提供了新的思路,使開發者能夠在沒有複雜人工干預的情況下,在多個性能指標之間取得平衡。研究團隊還計劃探索DynaMiCS在更大規模模型和更多領域上的應用,以及將其擴展到持續學習場景中。

展開要點與分析

文章情報

工程師進階

要點

  • DynaMiCS將多領域微調建模為約束優化問題,顯式控制約束領域性能不下降。
  • 通過短領域探測運行估計局部跨領域效應的斜率矩陣,捕捉各微調數據集對各評估領域的影響。
  • 在概率單純形上優化混合權重,以最小化目標領域損失並維持約束領域損失低於參考水平。
  • 相比固定混合基線,DynaMiCS在多個微調場景中實現了更強的目標領域改進和更高的約束滿足率,且計算成本更低。

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