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

語言模型能在思考中學習嗎?跨架構的梯度自適應測試

文章摘要

一項新研究系統測量了動態評估在混合線性注意力、擴散語言模型和長上下文基準上的表現,發現梯度更新主要適應風格和詞彙而非檢索,並揭示將事實寫入權重會損害推理能力,而將結論作為行為蒸餾到低秩介面卡可達到上下文上限且不影響GSM8K。

來源arXiv Machine Learning作者: Surya Saka
語言模型能在思考中學習嗎?跨架構的梯度自適應測試
回報錯誤

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

查看更正說明
直接讀正文

語言模型在閱讀文本時能否透過梯度步驟進行學習?這種現象被稱為動態評估,已被證明可以降低模型在文件剩餘部分的損失。然而,此前尚未有研究系統測量其在混合線性注意力架構、擴散語言模型以及標準長上下文基準上的行為。作者Surya Saka在論文中填補了這一空白。

實驗設定如下:每128個token的塊執行一次SGD步驟,僅更新最後四分之一塊的MLP輸出投影,不進行額外訓練,並使用2k token的視窗。對於softmax注意力模型(Qwen3-1.7B、Qwen3-4B、Gemma-4-E4B),動態評估使文件上的下一詞損失降低5-8%,並且優於同一凍結模型使用完整32k token上下文的表現。相比之下,混合Gated-DeltaNet模型(Qwen3.5-4B、Qwen3.8-27B)在任何學習率下增益均不足1%,表明線性注意力塊本身已經執行了這種自適應。

該機制同樣適用於掩碼擴散模型,且效果更顯著:Nemotron-Diffusion-8B上損失降低8%,DiffusionGemma-26B上降低17-20%,但需要降低學習率並對更新進行裁剪。

研究進一步在RULER、BABILong、LongBench-v2、SCOTUS、HotpotQA、MuSiQue、ContractNLI和引文補全等基準上測試,使用兩個模型尺寸和至少100個專案。結果顯示,在完整文件上更新的權重無法恢復因上下文截斷而損失的準確率。這表明動態評估帶來的適應主要是風格和詞彙層面的,而非檢索能力。

此外,研究發現使用下一詞損失寫入事即時,模型儲存的是提示格式而非事實本身:在聊天提示下召回率僅30%,而上下文內召回率為80%,並且無論劑量如何都會導致GSM8K分數下降36-38點。相反,將結論作為行為寫入——透過將模型自身以結論為條件的判斷蒸餾到低秩介面卡中——可以達到上下文內的效能上限(在100個遊戲上為0.83和0.88,對比上下文內的0.84和0.88),同時GSM8K保持不變。該結論可以來自一個透過提出資訊增益問題來尋找結論的詢問迴圈。

展開要點與分析

文章情報

投資人進階

要點

  • Softmax注意力模型透過動態評估可將下一詞損失降低5-8%,並優於使用完整32k上下文的凍結模型。
  • 混合Gated-DeltaNet模型增益不足1%,表明線性注意力塊已自行完成自適應。
  • 在多個長上下文基準上,權重更新無法恢復因上下文截斷損失的準確率,適應的是風格而非檢索。
  • 將結論蒸餾為行為(低秩介面卡)可達到上下文天花板,且GSM8K效能不變。

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