Mamba写代码真的超越Transformer!原始论文入选顶流新会议

💡 原文中文,约2500字,阅读约需6分钟。
📝

内容提要

Mistral AI和Mamba合作开发了一个名为Codestral Mamba的开源模型,它是基于Mamba2架构的代码生成模型。与Transformer架构不同,Mamba架构允许线性时间推理,并理论上支持无限长度的输入。Codestral Mamba在基准测试中表现优异。Mistral AI还发布了一个名为Mathstral的新数学模型。Mathstral在各个学科中表现出色,并且可以在更多推理时间下取得更好的结果。

🔎

延伸解读

Mamba架构的突破与认可

Mamba架构由Tri Dao和Albert Gu提出,旨在解决Transformer处理长文本时算力消耗巨大的问题。其核心创新包括对输入信息的选择性处理、硬件感知算法和更简单的架构。Mamba初代论文曾被ICLR拒稿,但最近被CoLM2024接收,Mamba2论文也入选ICML 2024,表明该架构正获得学术界的认可。

Codestral Mamba的性能表现

Codestral Mamba在基准测试中总体性能超越CodeGemma-1.1 7B、CodeLlama 7B、DeepSeek v1.5 7B和CodeLlama 34B。不过,DeepSeek v1.5 7B在Spider、HumanEval Java、HumanEval Bash、MBPP等任务上优于Codestral Mamba,显示其性能并非全面领先。

Mathstral的推理时间优化

Mathstral在MATH基准上得分56.6%,MMLU达到63.47%。通过增加推理时间计算,其性能可进一步提升:使用多数投票机制时MATH得分68.37%,应用强效奖励模型时可达74.59%。这表明该模型在更多计算资源下能获得更好的结果。

❓

Q&A

Codestral Mamba是什么模型?

Codestral Mamba是基于Mamba2架构的开源代码生成模型,由Mistral AI和Mamba合作开发。

Mamba架构与Transformer架构有什么不同?

Mamba架构支持线性时间推理,理论上可以处理无限长度的输入,而Transformer架构在处理长文本时算力消耗较大。

Codestral Mamba在基准测试中的表现如何?

Codestral Mamba在基准测试中表现优异,超越了多个现有模型,包括CodeGemma-1.1和CodeLlama。

Mathstral模型的主要应用领域是什么?

Mathstral模型专注于STEM领域,即科学、技术、工程和数学。

Mamba架构的创新点有哪些?

Mamba架构的创新点包括对输入信息的选择性处理、硬件感知的算法和更简单的架构设计。

Mamba2架构相比于初代Mamba有什么改进?

Mamba2架构在状态空间上扩大了8倍,训练速度提高了50%。

🏷️

标签

➡️

继续阅读