arXiv:2510.26182cs.CL2025-10

MossNet用专家混合机制模拟多头注意力,提升状态空间模型性能。

MossNet: Mixture of State-Space Experts is a Multi-Head Attention

  • 在MLP和时间混合核中引入专家混合,实现多头注意力模拟。
  • 同等规模下,语言建模和下游任务表现优于Transformer与SSM模型。
  • 适合追求高效高精度的大型语言模型研发者使用。

大语言模型(LLMs)显著推动了自然语言处理中的生成应用。当前模型架构趋势聚焦于高效的Transformer变体或状态空间/门控循环模型(SSMs, GRMs)。然而,现有基于SSM/GRM的方法通常仅模拟单一注意力头,可能限制其表达能力。本文提出MossNet,一种新颖的“状态空间专家混合”架构,可模拟线性多头注意力(MHA)。MossNet在通道混合MLP块和时间混合SSM核中均采用专家混合(MoE)实现多个“注意力头”。大量实验表明,MossNet在语言建模与下游任务中表现优于同等规模的Transformer与SSM架构。更大版本的MossNet在数万亿标记上训练后,进一步验证其可扩展性与优越性能。实机测试显示,其在三星Galaxy S24 Ultra与Nvidia A100 GPU上的运行速度和资源消耗优于同类基线。结果表明,MossNet是高效、高性能循环式大模型架构的有力新方向。

原文摘要 · Abstract (English)

Large language models (LLMs) have significantly advanced generative applications in natural language processing (NLP). Recent trends in model architectures revolve around efficient variants of transformers or state-space/gated-recurrent models (SSMs, GRMs). However, prevailing SSM/GRM-based methods often emulate only a single attention head, potentially limiting their expressiveness. In this work, we propose MossNet, a novel mixture-of-state-space-experts architecture that emulates a linear multi-head attention (MHA). MossNet leverages a mixture-of-experts (MoE) implementation not only in channel-mixing multi-layered perceptron (MLP) blocks but also in the time-mixing SSM kernels to realize multiple "attention heads." Extensive experiments on language modeling and downstream evaluations show that MossNet outperforms both transformer- and SSM-based architectures of similar model size and data budgets. Larger variants of MossNet, trained on trillions of tokens, further confirm its scalability and superior performance. In addition, real-device profiling on a Samsung Galaxy S24 Ultra and an Nvidia A100 GPU demonstrate favorable runtime speed and resource usage compared to similarly sized baselines. Our results suggest that MossNet is a compelling new direction for efficient, high-performing recurrent LLM architectures.

大模型状态空间专家混合注意力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。