首页
学习
活动
专区
圈层
工具
发布
技术百科首页 >状态空间模型 >状态空间模型与线性注意力有何理论联系?

状态空间模型与线性注意力有何理论联系?

词条归属:状态空间模型

1. 结构化状态空间对偶(SSD)

2024 年发表的 Mamba-2 提出了结构化状态空间对偶(Structured State Space Duality,SSD)框架,从数学上揭示了 SSM 与注意力机制的深层联系:二者可视为同一结构化矩阵的不同分解形式。

2. SSM 与掩码注意力的等价性

SSD 框架证明,当状态矩阵为标量乘以单位矩阵(所有对角元素相同)时,SSM 递推在数学上等价于一种带 1-半可分因果掩码的掩码自注意力。这意味着同一次计算既可表达为 SSM 循环,也可表达为类注意力的矩阵乘法。

3. 统一视角的意义

这一理论统一表明,SSM 与 Transformer 并非截然对立的两种范式,而是序列模型谱系上的不同取值点。当所有状态值取 1 时,注意力形式退化为标准因果线性注意力。该联系为设计和混合两种架构提供了理论基础。

相关文章
Hinton的GLOM模型与千脑理论有何本质不同?
Geoffrey Hinton在最新发表的一篇论文“如何在神经网络中表示部分-整体层次结构”中提出了一种被称为GLOM的新理论。
AI科技评论
2021-05-19
1.6K0
新架构Mamba更新二代!作者:别争了,数学上Transformer和SSM是一回事
更重要的是,团队研究发现原来Transformer和状态空间模型(SSM)竟然是近亲??‍
量子位
2024-06-04
1.1K0
SSM 能取代 Transformer 搞出更「牛」的大模型吗?
---- 本周为您解读 ③ 个值得细品的 AI & Robotics 业内要事 ----
机器之心
2024-06-17
1.3K0
【源头活水】再战Transformer!Mamba 2来了,新架构训练效率大幅提升!!!
“问渠那得清如许,为有源头活水来”,通过前沿领域知识的学习,从其他研究领域得到启发,对研究问题的本质有更清晰的认识和理解,是自我提高的不竭源泉。为此,我们特别精选论文阅读笔记,开辟“源头活水”专栏,帮助你广泛而深入的阅读科研文献,敬请关注!
马上科普尚尚
2024-06-18
8200
Transformers是SSMs:通过结构化状态空间对偶性的广义模型和高效算法(一)
尽管Transformer一直是深度学习在语言建模中取得成功的主要架构,但最近的研究表明,如Mamba之类的状态空间模型(SSMs)在小到中等规模上能够匹敌或超越Transformer的性能。我们表明,这两类模型实际上是非常相关的,并在一个经过充分研究的结构化半可分离矩阵类的各种分解之间,发展出SSM和注意力变体之间丰富的理论联系框架。我们的状态空间对偶性(SSD)框架使我们能够设计一种新的架构(Mamba-2),其核心层是对Mamba的选择性SSM的改进,速度提高了2-8倍,同时在语言建模方面继续与Transformer保持竞争力。
AI浩
2024-10-22
1.1K0
点击加载更多
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
领券