arXiv:2409.02877cs.AIcs.CL2024-09被引 29

将大模型拆解为可动态组合的功能模块,提升效率与灵活性。

Configurable Foundation Models: Building LLMs from a Modular Perspective

  • 用'积木块'比喻功能模块,实现按需组装
  • 发现前馈层神经元具功能分化特征
  • 适合研究高效大模型与可扩展系统者

大模型的快速发展带来了计算效率与持续扩展性的挑战,因其需要海量参数,使得在资源受限设备上应用及适应多样需求愈发困难。受人脑模块化启发,当前趋势是将大模型分解为多个功能模块,支持部分模块推理和动态组合以应对复杂任务,如专家混合(mixture-of-experts)。为突出模块化方法的内在高效性与可组合性,本文提出用“brick”(积木块)代表每个功能模块,并将这种结构命名为可配置基础模型。本文系统梳理了可配置基础模型的构建、使用与局限。首先将模块形式化为预训练中涌现的“新兴积木”——功能神经元分区,以及通过后训练定制的“定制积木”。基于多种功能积木,提出四种面向积木的操作:检索与路由、合并、更新、增长。这些操作使模型能根据指令动态配置,处理复杂任务。我们在主流大模型上进行实证分析,发现前馈网络(FFN)层中的神经元呈现模块化模式,具有功能特异性。最后,本文指出了若干开放问题与未来方向。总体而言,本工作为现有大模型研究提供了新的模块化视角,旨在激发更高效、可扩展的基础模型设计。

原文摘要 · Abstract (English)

Advancements in LLMs have recently unveiled challenges tied to computational efficiency and continual scalability due to their requirements of huge parameters, making the applications and evolution of these models on devices with limited computation resources and scenarios requiring various abilities increasingly cumbersome. Inspired by modularity within the human brain, there is a growing tendency to decompose LLMs into numerous functional modules, allowing for inference with part of modules and dynamic assembly of modules to tackle complex tasks, such as mixture-of-experts. To highlight the inherent efficiency and composability of the modular approach, we coin the term brick to represent each functional module, designating the modularized structure as configurable foundation models. In this paper, we offer a comprehensive overview and investigation of the construction, utilization, and limitation of configurable foundation models. We first formalize modules into emergent bricks - functional neuron partitions that emerge during the pre-training phase, and customized bricks - bricks constructed via additional post-training to improve the capabilities and knowledge of LLMs. Based on diverse functional bricks, we further present four brick-oriented operations: retrieval and routing, merging, updating, and growing. These operations allow for dynamic configuration of LLMs based on instructions to handle complex tasks. To verify our perspective, we conduct an empirical analysis on widely-used LLMs. We find that the FFN layers follow modular patterns with functional specialization of neurons and functional neuron partitions. Finally, we highlight several open issues and directions for future research. Overall, this paper aims to offer a fresh modular perspective on existing LLM research and inspire the future creation of more efficient and scalable foundational models.

模块化大模型可配置架构设计

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