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

MARCH:通過內容路由狀態錨點擴展循環記憶

文章摘要

MARCH通過週期性地緩存循環狀態檢查點作為“狀態錨點”,為固定大小的循環記憶引入可增長的內容尋址記憶庫。該方法保留了高效的循環計算路徑,同時能夠隨上下文長度擴展記憶容量,並在標準預訓練後於常識推理、LongBench和上下文檢索上超越多種線性注意力變體。

來源arXiv Machine Learning作者: Ming Zhang, Kaisen Yang, Shu Yu, Ermo Hua, Ning Ding, Xia Hu, Bowen Zhou, Chaochao Lu, Youbang Sun
MARCH:通過內容路由狀態錨點擴展循環記憶
報告錯誤

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

查看更正說明
直接讀正文

在長上下文建模中,Transformer 與循環模型分別代表了兩種不同的取捨。Transformer 的優勢在於它保留了 token 級別的記憶,這種記憶隨上下文長度線性增長,因此能夠精細地回溯較早的信息,從而在長文本檢索和知識召回任務中表現突出。但這份能力並非沒有代價:訓練時的注意力計算量隨序列長度呈二次方增長,推理階段還需要維護同樣不斷膨脹的鍵值緩存,導致顯存佔用和延遲隨之上升。

循環類架構(如狀態空間模型)則走另一條路線:它們把全部歷史信息壓縮進一個固定大小的隱狀態,解碼時只需要更新這個狀態,計算成本與序列長度基本解耦,因此推理效率很高。然而,固定大小意味着容量有限,早期的關聯信息很容易被後續輸入覆蓋,模型往往只能記住最近的內容,在需要精確回憶的基準上明顯弱於 Transformer。

針對這一矛盾,Ming Zhang 等九位研究者提出了 MARCH(Memory-Anchor Routing across Context History)。其核心思想是給循環模型增加一個可擴展的“外部記憶庫”:模型每隔一定步數就把當前的循環狀態檢查點保存下來,稱為狀態錨點;每個錨點會附帶一個由內容決定的緊湊鍵。這樣,隨着上下文變長,記憶庫中的錨點數量也會增加,歷史信息的解析度與內存開銷之間可以按需調節。

在具體計算時,MARCH 讓每個 token 生成一個錨點查詢,去檢索所有因果上可用的歷史錨點,並把檢索結果以類似注意力的方式與當前狀態聚合。換言之,模型既保留了原有的循環計算主幹,又獲得了一條額外的長期記憶通路,不需要像 Transformer 那樣維護完整的鍵值緩存,也不會被固定大小的狀態所限制。

論文在標準預訓練設置下進行了評估,實驗涵蓋常識推理、LongBench 長上下文基準以及上下文檢索任務。結果顯示,MARCH 在多個線性注意力變體之上取得了一致的性能優勢,説明內容路由的狀態緩存確實能強化循環模型的長期記憶,同時保住其高效的原生計算路徑。此外,錨點間隔的選擇會直接影響記憶的粒度:間隔越小,歷史細節越豐富,但存儲和檢索開銷也會隨之增加;間隔越大,記憶越粗糙,但效率更高。

該論文 2026 年 8 月 12 日提交至 arXiv,編號 2608.12435,目前在 cs.LG 分類下,DOI 為 10.48550/arXiv.2608.12435。論文由 Ming Zhang 等九位作者完成,arXiv 的記錄顯示稿件提交人為 Youbang Sun。需要指出的是,實驗結論基於標準預訓練配置,實際部署中的收益仍取決於具體任務與算力約束。

展開要點與分析

文章情報

工程師進階

要點

  • MARCH通過週期性緩存循環狀態檢查點作為“狀態錨點”,為循環模型提供可增長的記憶庫。
  • 每個狀態錨點配有內容相關的緊湊鍵,當前token通過錨點查詢對歷史錨點執行注意力式聚合。
  • 在標準預訓練後,MARCH在常識推理、LongBench和上下文檢索上優於多種線性注意力變體。

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