本文来源公众号“量子位”,仅用于学术分享,侵权删,干货满满。
原文链接:新架构Mamba更新二代!作者:别争了,数学上Transformer和SSM是一回事
Transformer挑战者、新架构Mamba,刚刚更新了第二代:
Mamba-2,状态空间扩大8倍,训练速度提高50%!
更重要的是,团队研究发现原来Transformer和状态空间模型(SSM)竟然是近亲??
两大主流序列建模架构,在此统一了。
没错,这篇论文的提出的重磅发现:Transformer中的注意力机制与SSM存在着非常紧密的数学联系。
团队通过提出一个叫结构化状态空间二元性(Structured State Space Duality,SSD)的理论框架,把这两大模型家族统一了起来。
Mamba一代论文年初被ICLR拒稿,当时还