用低秩残差专家结构,让大模型微调更高效且灵活。
S'MoRE: Structural Mixture of Residual Experts for Parameter-Efficient LLM Fine-tuning
- 通过分层低秩分解专家权重,构建多层残差结构。
- 相同参数量下,性能优于传统MoE或LoRA,提升指数级灵活性。
- 适合追求高效微调的AI研发人员和资源有限场景。
微调预训练大语言模型面临参数效率与模型容量的双重挑战。现有方法如低秩适配(LoRA)虽高效但灵活性不足,而混合专家(MoE)虽提升容量却导致参数增多且利用率低。为此,我们提出结构化残差专家混合(S'MoRE),将LoRA的效率与MoE的灵活性无缝结合。S'MoRE采用专家权重的分层低秩分解,生成多阶残差,并以多层结构互联。通过将输入令牌路由至残差子树,仅需少量低秩矩阵即可模拟大量专家的能力。我们将残差的跨层传播设计为一种特殊图神经网络(GNN),并证明在相近参数预算下,S'MoRE的结构灵活性相比传统MoE(或混合-LoRA)呈指数级提升。理论分析与实证结果均表明,S'MoRE在微调性能上表现优异,为高效大模型适配提供了变革性方案。代码已开源:https://github.com/ZimpleX/SMoRE-LLM。
原文摘要 · Abstract (English)
Fine-tuning pre-trained large language models (LLMs) presents a dual challenge of balancing parameter efficiency and model capacity. Existing methods like low-rank adaptations (LoRA) are efficient but lack flexibility, while Mixture-of-Experts (MoE) enhance model capacity at the cost of more & under-utilized parameters. To address these limitations, we propose Structural Mixture of Residual Experts (S'MoRE), a novel framework that seamlessly integrates the efficiency of LoRA with the flexibility of MoE. Conceptually, S'MoRE employs hierarchical low-rank decomposition of expert weights, yielding residuals of varying orders interconnected in a multi-layer structure. By routing input tokens through sub-trees of residuals, S'MoRE emulates the capacity of numerous experts by instantiating and assembling just a few low-rank matrices. We craft the inter-layer propagation of S'MoRE's residuals as a special type of Graph Neural Network (GNN), and prove that under similar parameter budget, S'MoRE improves structural flexibility of traditional MoE (or Mixture-of-LoRA) by exponential order. Comprehensive theoretical analysis and empirical results demonstrate that S'MoRE achieves superior fine-tuning performance, offering a transformative approach for efficient LLM adaptation. Our implementation is available at: https://github.com/ZimpleX/SMoRE-LLM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。