arXiv:2601.21420cs.LG2026-01被引 1

让大模型按需分配算力,动态压缩相似词元提升效率

ConceptMoE: Adaptive Token-to-Concept Compression for Implicit Compute Allocation

  • 通过语义相似度合并词元为概念,实现隐式算力分配
  • 在保持算力不变下,语言和多模态任务平均提分超2.3点
  • 压缩比达2时,预填充速度提升175%,适合长文本场景

大语言模型对所有词元均等分配计算资源,忽视部分序列可预测而另一些需深度推理。本文提出ConceptMoE,通过可学习的分块模块动态识别词元间相似性,将语义相近的词元合并为概念表征,在进入高成本概念模型前按目标压缩率 $R$ 压缩序列。关键在于,其MoE架构允许精确重分配节省的算力以匹配基线激活FLOPs(不含注意力计算)与总参数量,从而隔离出真实架构优势。在此条件下,ConceptMoE在语言与视觉-语言任务中持续优于标准MoE:语言预训练+0.9点,长上下文理解+2.3点,多模态基准+0.6点;持续训练中采用层循环时提升达+5.5点,体现实际应用价值。此外,注意力计算最多减少 $R^2 imes$,KV缓存减少 $R imes$。当 $R=2$ 时,实测预填充加速达175%,解码加速最高117%。极小的结构改动即可无缝集成至现有MoE,证明自适应概念级处理能根本性提升大模型的有效性与效率。

原文摘要 · Abstract (English)

Large language models allocate uniform computation across all tokens, ignoring that some sequences are trivially predictable while others require deep reasoning. We introduce ConceptMoE, which dynamically merges semantically similar tokens into concept representations, performing implicit token-level compute allocation. A learnable chunk module identifies optimal boundaries by measuring inter-token similarity, compressing sequences by a target ratio $R$ before they enter the compute-intensive concept model. Crucially, the MoE architecture enables controlled evaluation: we reallocate saved computation to match baseline activated FLOPs (excluding attention map computation) and total parameters, isolating genuine architectural benefits. Under these conditions, ConceptMoE consistently outperforms standard MoE across language and vision-language tasks, achieving +0.9 points on language pretraining, +2.3 points on long context understanding, and +0.6 points on multimodal benchmarks. When converting pretrained MoE during continual training with layer looping, gains reach +5.5 points, demonstrating practical applicability. Beyond performance, ConceptMoE reduces attention computation by up to $R^2\times$ and KV cache by $R\times$. At $R=2$, empirical measurements show prefill speedups reaching 175\% and decoding speedups up to 117\% on long sequences. The minimal architectural modifications enable straightforward integration into existing MoE, demonstrating that adaptive concept-level processing fundamentally improves both effectiveness and efficiency of large language models.

MoE算力分配高效推理概念压缩

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