研究人员开发了DiffuMamba,这是一种利用Mamba骨干网络来提高推理效率的新型扩散语言模型。这种方法解决了基于Transformer的模型存在的二次注意力或KV缓存开销问题,尤其是在长序列处理方面。DiffuMamba及其混合变体DiffuMamba-H在下游任务上的性能与Transformer模型相当,同时实现了显著更高的吞吐量。研究表明,Mamba混合器结合缓存高效的块扩散,为基于扩散的生成系统提供了实现线性扩展序列建模的有前景的途径。 AI
影响 引入了更高效的扩散语言模型骨干网络,可能加速生成任务。
排序理由 该集群包含一篇详细介绍新模型架构及其性能的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →