[控场AI]
模型Mamba 2

Mamba2

基于状态空间模型(SSM)的高效序列建模架构,将状态维度与输入通道绑定,支持矩阵乘法形式并行计算,具有线性复杂度和常数级推理显存占用,是长上下文场景中 Transformer 的替代或补充方案。

时间轴 (近 90 天)

10月17日

Mamba2 chunk scan kernel 在 SM90 上产生 NaN 的问题,很可能属于由架构差异触发的数值稳定性缺陷。

待验证50%
10月17日

vLLM 发布了 v0.32.0rc3 版本,其中包含修复 Mamba2 chunk scan kernel 在 NVIDIA SM90 架构上产生 NaN 输出的 Bugfix。

待验证50%
10月17日

Mamba2 是一种基于状态空间模型(SSM)的序列建模架构,将 SSM 状态维度与输入通道绑定,并将逐时间步串行递推重写为矩阵乘法形式,可利用 GPU Tensor Core 并行计算。

待验证50%
10月17日

Mamba2 的 chunk scan 机制采用块内并行与块间串行的两阶段策略,使并行度从序列长度级降低到块数量级。

待验证50%
10月16日

Mamba2在Mamba的基础上引入了更结构化的状态空间对偶形式,使递归计算与并行扫描能够更高效地结合,并改善了硬件利用率。

待验证50%
10月16日

Mamba2架构具有线性时间复杂度和恒定大小的状态表示,其状态表示不随序列长度增长。

待验证50%
10月16日

Transformer的KV缓存大小随序列长度线性增长,而Mamba2的递归状态缓存大小与上下文长度无关,只取决于状态维度和层数。

待验证50%
10月16日

Mamba2保持线性时间复杂度和恒定状态大小,因此在长序列场景中具有理论优势。

待验证50%
10月7日

针对 gated-delta、Mamba2 和短卷积等混合架构模型,样本打包得到改进并修正了打包逻辑问题

待验证50%

全部知识事实 (9)

来源文章