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

MIITA:面向小語言模型持續學習的記憶體驅動推理時適配框架

文章摘要

MIITA是一種專為資源受限環境下小語言模型(SLM)設計的持續學習框架,透過儲存緊湊的修正方向原型並在推理時基於語義和不確定性線索檢索,實現無破壞性的知識複用,有效緩解災難性遺忘。實驗表明,在固定記憶體預算下,MIITA能持續提升最終效能並減少遺忘。

來源arXiv AI作者: Dong Li, Yanchi Liu, Xujiang Zhao, Wei Cheng, Zhengzhang Chen, Xintao Wu, Zhong Chen, Chen Zhao, Haifeng Chen
MIITA:面向小語言模型持續學習的記憶體驅動推理時適配框架
回報錯誤

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

查看更正說明
直接讀正文

MIITA(Memory-Induced Inference-Time Adaptation)是一項由Dong Li等人提出的創新框架,旨在解決小語言模型(SLM)在資源受限環境下進行持續學習時面臨的災難性遺忘問題。持續學習對於SLM適應不斷變化的應用場景至關重要,但直接更新其有限的引數往往導致之前學到的知識被覆蓋。儘管基於記憶的方法透過將知識儲存與引數更新解耦來緩解這一問題,但現有針對大語言模型(LLM)的方法通常需要大量儲存空間和強大的上下文推理能力,這超出了SLM的能力範圍。

MIITA透過一種全新的思路應對這一挑戰。它設計了一種緊湊的修正方向原型,併為每個原型附加語義錨點,從而以極低的儲存成本編碼重要的監督經驗。在推理階段,MIITA利用語義相似性和不確定性線索從記憶庫中檢索最相關的原型。檢索到的修正方向透過門控機制對模型的隱藏狀態進行臨時調整,從而在不修改骨幹網路引數、不增加提示、也不進行測試時反向傳播的情況下,實現過去知識的安全複用。

理論分析表明,MIITA的設計與一階損失減少、不確定性引導的檢索以及方向覆蓋密切相關,這有助於系統性地保留舊任務的知識。實驗涵蓋了多種持續學習場景,包括任務增量學習、類別增量學習和領域增量學習。在多個自然語言處理基準測試中,MIITA表現出一致的效能提升。例如,在GLUE子任務上,相比於直接微調和傳統的記憶重放方法,MIITA在保持舊任務準確率的同時,顯著提高了新任務的適應速度。其核心優勢在於無需改變模型架構,僅透過記憶原型和推理時適配即可實現知識遷移。此外,MIITA的記憶體佔用遠低於基於完整樣本重放的方法,使其非常適合在儲存受限的裝置上執行。研究團隊還開源了相關程式碼,以便社群進一步驗證和應用。這項研究不僅為小模型的持續學習奠定了理論依據,也為實際部署提供了可行的技術路徑。

展開要點與分析

文章情報

工程師進階

要點

  • 針對小語言模型持續學習中的災難性遺忘問題提出MIITA框架
  • 透過語義錨點儲存經驗為修正方向原型,推理時基於語義和不確定性檢索
  • 採用門控臨時隱藏狀態適配,無需更新骨幹網路、擴充套件提示或測試時反向傳播
  • 實驗驗證在固定記憶體預算下有效提升效能並緩解遺忘

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