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

DualOptim+:橋接共享與分離優化器狀態,提升大語言模型的機器遺忘能力

文章摘要

DualOptim+是一個新型優化框架,通過引入基態和增量態,自適應地橋接共享與分離的優化器狀態,以改善大語言模型中的機器遺忘。實驗表明,該框架在遺忘、安全對齊和多任務學習等任務中取得了更優的權衡。

來源arXiv Machine Learning作者: Xuyang Zhong, Qizhang Li, Yiwen Guo, Chen Liu
DualOptim+:橋接共享與分離優化器狀態,提升大語言模型的機器遺忘能力
報告錯誤

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

查看更正說明
直接讀正文

香港城市大學的研究團隊近日提出了名為DualOptim+的新型優化框架,旨在顯著提升大語言模型(LLM)的機器遺忘(machine unlearning)性能。該研究成果已被機器學習頂級會議ICML 2026接收。

機器遺忘技術旨在讓模型在特定數據集上“忘記”所學信息,同時保持對其他知識的記憶,這在隱私保護、模型修正和去除不良內容等方面具有重要應用。然而,遺忘目標與保留目標之間往往存在衝突,導致現有方法難以取得良好平衡。

DualOptim+通過引入基態(base state)和增量態(delta states)來巧妙化解這一衝突。基態捕捉遺忘和保留目標的共同表示,而增量態則分別保留各自特定的殘差信息。優化器能夠根據遺忘梯度與保留梯度之間的方向衝突程度,自適應地橋接這兩種狀態,從而在優化過程中動態平衡不同目標。

此外,研究團隊還推出了DualOptim+的量化變體——DualOptim+ 8bit,通過8位量化技術大幅減少內存佔用,同時不犧牲模型性能。實驗涵蓋了虛構和真實世界的遺忘任務、安全對齊以及多任務學習等多種場景。結果表明,DualOptim+在多目標優化中始終能夠實現更優的權衡,顯著優於現有方法。

該工作的代碼已在GitHub上開源,為大型語言模型的機器遺忘領域提供了新的優化思路和實用工具。

展開要點與分析

文章情報

工程師進階

要點

  • DualOptim+引入基態捕獲遺忘和保留目標的共同表示,增量態保留特定殘差。
  • 該框架根據梯度方向衝突自適應橋接共享與分離狀態。
  • 提出8位量化變體DualOptim+ 8bit,降低內存開銷。
  • 在多個任務上優於現有方法,代碼已開源。

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