arXiv:2601.21461cs.LGcs.AI2026-01被引 3

提出静态路由的大型查找层,用嵌入缓存提升稀疏模型效率。

L$^3$: Large Lookup Layers

  • 用静态词元路由聚合上下文相关嵌入,实现高效稀疏计算。
  • 在26亿活跃参数下,性能优于同等稀疏度的MoE和密集模型。
  • 适合追求高吞吐、低延迟部署的工业级语言模型应用。

现代稀疏语言模型通常通过专家混合(MoE)层实现稀疏性,动态硬路由存在硬件效率低和训练不稳定等问题。相比之下,分词器嵌入表虽天然稀疏,但缺乏上下文信息。本文提出大型查找层(L$^3$),将嵌入表扩展为解码器层,以静态词元路由方式按上下文聚合一组学习过的嵌入,使模型能通过嵌入缓存有效平衡内存与计算。L$^3$包含两个核心部分:(1) 系统友好的架构,支持快速训练与无开销的CPU卸载推理;(2) 基于信息论的嵌入分配算法,有效协调速度与质量。我们在高达26亿活跃参数的Transformer上进行实验,结果表明L$^3$在语言建模和下游任务中显著优于密集模型及同规模MoE。

原文摘要 · Abstract (English)

Modern sparse language models typically achieve sparsity through Mixture-of-Experts (MoE) layers, which dynamically route tokens to dense MLP "experts." However, dynamic hard routing has a number of drawbacks, such as potentially poor hardware efficiency and needing auxiliary losses for stable training. In contrast, the tokenizer embedding table, which is natively sparse, largely avoids these issues by selecting a single embedding per token at the cost of not having contextual information. In this work, we introduce the Large Lookup Layer (L$^3$), which generalizes embedding tables to model decoder layers as a means of further scaling sparsity. L$^3$ layers use static token-based routing to aggregate a set of learned embeddings per token in a context-dependent way, allowing the model to efficiently balance memory and compute by caching information in embeddings. L$^3$ has two main components: (1) a systems-friendly architecture that allows for fast training and CPU-offloaded inference with no overhead, and (2) an information-theoretic embedding allocation algorithm that effectively balances speed and quality. We empirically test L$^3$ by training transformers with up to 2.6B active parameters and find that L$^3$ strongly outperforms both dense models and iso-sparse MoEs in both language modeling and downstream tasks.

稀疏模型嵌入优化MoE改进高效推理

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