跳到正文
NVIDIA Technical Blog:Agentic AI / Generative AI· Michelle Horton·· 2 小時前AI 評分19

ReplaySSM:重新思考混合模型的狀態儲存,以加快解碼

ReplaySSM: Rethinking State Storage for Faster Hybrid-Model Decoding

AI 導讀

ReplaySSM 聚焦重新思考混合模型的狀態儲存,以加快解碼。混合模型會交錯使用 SSM 層與少量注意力層;NVIDIA Nemotron 3、Qwen3.5 至 Qwen3.8、Kimi-K3 和 GLM-5.3-Flash 均採用此架構,而 SSM 層以固定大小的狀態取代會增長的 KV cache。

正文

A cube assembled from smaller dark blocks, with glowing cyan, pink, yellow, and green sections against a black background.Hybrid models interleave state-space model (SSM) layers with a few attention layers. NVIDIA Nemotron 3, Qwen3.5 through Qwen3.8, Kimi-K3, and GLM-5.3-Flash are...A cube assembled from smaller dark blocks, with glowing cyan, pink, yellow, and green sections against a black background.

Hybrid models interleave state-space model (SSM) layers with a few attention layers. NVIDIA Nemotron 3, Qwen3.5 through Qwen3.8, Kimi-K3, and GLM-5.3-Flash are all built this way, with Mamba-2, Gated DeltaNet (GDN), or Kimi Delta Attention (KDA) as the SSM layer. SSM layers are designed to make decoding fast: they use a fixed-size state in place of a key-value (KV) cache that grows with…

Source

來源:NVIDIA Technical Blog:Agentic AI / Generative AI · developer.nvidia.com