arXiv:2512.14531cs.CL2025-12被引 1

通过动态路由让大模型用同一组参数实现宽深灵活复用,节省内存还提升性能。

VersatileFFN: Achieving Parameter Efficiency in LLMs via Adaptive Wide-and-Deep Reuse

  • 设计双路径结构:宽向复用生成专家混合,深向迭代处理复杂文本。
  • 在多个基准上测试,小模型性能接近大模型,内存占用更低。
  • 适合资源受限场景,如移动端部署或快速迭代模型。

大型语言模型虽性能卓越,但内存开销巨大。现有压缩方法如剪枝和量化仅降低参数量,无法提升模型表达能力。本文提出VersatileFFN,一种新型前馈网络,在固定参数预算下灵活复用宽度与深度维度。受认知双过程理论启发,该结构包含两个自适应路径:宽度复用路径从单一共享的FFN中生成子专家混合,模拟稀疏专家路由而不增加参数;深度复用路径递归应用同一FFN,对复杂令牌进行深度处理。难度感知门控动态分配路径,将简单样本走高效宽度路径,复杂样本进入深度迭代优化。关键在于两条路径共享参数,新增能力仅来自计算而非内存。在多种基准和模型规模上验证有效,代码已开源。

原文摘要 · Abstract (English)

The rapid scaling of Large Language Models (LLMs) has achieved remarkable performance, but it also leads to prohibitive memory costs. Existing parameter-efficient approaches such as pruning and quantization mainly compress pretrained models without enhancing architectural capacity, thereby hitting the representational ceiling of the base model. In this work, we propose VersatileFFN, a novel feed-forward network (FFN) that enables flexible reuse of parameters in both width and depth dimensions within a fixed parameter budget. Inspired by the dual-process theory of cognition, VersatileFFN comprises two adaptive pathways: a width-versatile path that generates a mixture of sub-experts from a single shared FFN, mimicking sparse expert routing without increasing parameters, and a depth-versatile path that recursively applies the same FFN to emulate deeper processing for complex tokens. A difficulty-aware gating dynamically balances the two pathways, steering "easy" tokens through the efficient width-wise route and allocating deeper iterative refinement to "hard" tokens. Crucially, both pathways reuse the same parameters, so all additional capacity comes from computation rather than memory. Experiments across diverse benchmarks and model scales demonstrate the effectiveness of the method. The code is available at https://github.com/huawei-noah/noah-research/tree/master/VersatileFFN.

参数效率模型压缩前馈网络动态路由

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