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

睡眠框架:讓大語言模型在持續學習中保持知識的新方法

文章摘要

谷歌研究團隊提出“睡眠框架”,通過模擬人類睡眠中的記憶鞏固與夢境階段,使大語言模型能夠在持續學習中不斷吸收新事實並提升推理能力,同時避免災難性遺忘。

睡眠框架:讓大語言模型在持續學習中保持知識的新方法
報告錯誤

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

查看更正說明
直接讀正文

現代大語言模型(LLM)在預訓練和上下文學習方面表現卓越,但一旦部署,它們往往就處於靜態狀態,知識截止日期固定,難以持續獲取新能力或適應新信息。直接使用新數據更新模型看似是顯而易見的解決方案,但持續的參數更新常常引發災難性遺忘——學習新任務會嚴重損害模型在舊任務上的性能。而僅依賴上下文學習,知識保留又受限於即時上下文窗口,會話結束後短期信息便被遺忘。

受人類大腦在休息時處理和鞏固記憶的啓發,谷歌研究團隊引入了一種名為“睡眠框架”的持續學習範式。該框架為LLM賦予連續的週期性生命週期,使其能夠不斷學習,將脆弱的短期上下文鞏固為穩定的長期參數。通過交替進行主動交互和離線記憶鞏固,模型可以持續獲取新事實、保留長達1000萬token的長上下文知識,並提升複雜推理能力。

在睡眠範式下,LLM在兩個主要狀態間運行:主動(清醒)階段,模型接收、處理輸入並與之交互,將臨時信息存儲在快速短期記憶組件中;睡眠階段,模型暫停處理新輸入,將內部計算轉向記憶鞏固和自我提升。儘管“睡眠”“做夢”等術語借自神經科學,但該框架完全在固定模型架構和人類定義的訓練規則內運作。在離線睡眠階段,模型解鎖預分配的多專家(MoE)記憶塊,並使用自生成的練習示例更新數值權重。這種自動化後訓練循環幫助已部署的模型吸收新事實並提高推理準確性,同時保持核心代碼庫和設計不變。

睡眠過程通過兩個階段解決災難性遺忘和持續適應問題:記憶鞏固和夢境。記憶鞏固模仿人類非快速眼動慢波睡眠,通過兩個關鍵技術機制實現:參數擴展與連續記憶層次,以及知識播種(向上蒸餾)。為防止災難性遺忘,模型在睡眠期間動態擴展容量。利用MoE和Hope架構,新解鎖的低秩專家被集成到穩定的低頻記憶塊中,以存儲新知識而不干擾舊參數。高頻層(如上下文記憶)中的脆弱短期記憶通過廣義知識蒸餾(GKD)結合基於強化學習的模仿學習轉移到新擴展的長期參數中。具體包括:同策略蒸餾,擴展後的學生模型從自採樣序列生成的教師分佈中學習token級反饋;學習模仿(LTI),學生模型在完成教師生成文本的前綴時,根據語義和絕對相似度(Levenshtein距離)獲得獎勵。一旦記憶鞏固完成,高頻臨時參數被重置,為未來活躍會話釋放容量。

夢境階段則模仿快速眼動睡眠,實現自主自我提升循環。在合成數據生成過程中,MoE路由器隨機激活非標準專家,迫使模型混合不同領域的知識並發現潛在模式。模型生成合成訓練樣本(“夢境”),並使用基於梯度的數據選擇進行過濾,優先選擇最有可能提升整體模型能力的樣本。然後,模型對排名最高的夢境樣本進行監督微調,並評估生成是否改善下游任務性能。

除了實證結果,睡眠範式還有理論基礎,闡明其兩個階段為何能支持持續學習而不犧牲舊能力。對於記憶鞏固,分析表明知識播種可理解為結構化的低秩壓縮問題。模型不需要將短期痕跡的每個細節複製到參數中,而是識別一組緊湊的方向,以最好地保留模型自身的同策略行為和獎勵所鼓勵的語義質量。同策略蒸餾使鞏固與模型在推理時實際產生的軌跡對齊,減少僅訓練教師生成分佈時可能出現的失配;語義獎勵則改變值得保留的內容,優先考慮保留意義和任務相關行為的方向,而不僅僅是匹配token概率。參數擴展提供了寫入這些選定記憶所需的容量,而不會覆蓋先前鞏固的知識。

對於夢境,理論將隨機專家激活和基於獎勵的樣本選擇描述為改善學習幾何的機制。注入不同的MoE拓寬了合成生成過程中探索的梯度方向集,使模型接觸到比普通清醒數據更豐富的更新集合。基於梯度的過濾則強調攜帶最有價值學習信號的合成經驗,加強信息方向,同時保留底層梯度空間的結構。這些步驟共同產生更平衡的優化景觀:學習較少受少數方向支配,梯度條件更好,模型在精煉能力時能取得更穩定的進展。從這個角度看,夢境不僅僅是自生成數據增強,而是一個有意識地多樣化、選擇和鞏固模型自我改進更新的過程。

在持續翻譯方面,對於未見過的語言(如滿語和卡拉芒語),睡眠框架在多階段評估中保持了近乎零的性能下降,而ICL、Cartridges和監督微調(SFT)等標準基線在順序學習中遭受了嚴重的災難性遺忘。在BABILong基準的長上下文理解中,睡眠方法展示了高達1000萬token的近完美擴展,通過睡眠時鞏固將長上下文信息直接壓縮到參數記憶中,優於GPT-4和Titans等強基線。在SQuAD和持續預訓練(CPT)評估的事實知識融入中,睡眠方法在單任務上達到48.9%,在CPT評估中達到46.2%,超越了包括基礎模型、SFT和自適應語言模型(SEAL)在內的基線方法,成功整合新事實信息而不降低現有模型表示。最後,在包括AIME 24/25和HMMT 25的數學推理基準上,睡眠增強的基礎模型分別達到79.2%、69.0%和46.1%的準確率,優於SFT、組相對策略優化(GRPO)和同策略自蒸餾(OPSD)等基線,表明將記憶鞏固與模仿學習相結合可驅動卓越的推理性能。

總之,睡眠範式為自主、持續學習的AI系統提供了一條路徑。通過超越靜態預訓練和脆弱的上下文窗口,交替進行主動學習和離線休息,使LLM能夠保留短期記憶,將新技能抽象為長期參數結構,並在不發生災難性遺忘的情況下持續自我改進。該研究由Ali Behrouz(谷歌和康奈爾大學)、Farnoosh Hashemi(康奈爾大學)、Adel Javanmard(谷歌和南加州大學)和Vahab Mirrokni共同完成。

展開要點與分析

文章情報

工程師進階

要點

  • 睡眠框架通過交替的主動交互與離線記憶鞏固來持續學習。
  • 記憶鞏固階段使用參數擴展和知識蒸餾將短期記憶轉化為長期參數。
  • 夢境階段通過隨機專家選擇和強化學習實現自主自我提升。
  • 實驗表明該方法在持續翻譯、長上下文理解和推理任務上優於基線。

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