arXiv:2411.16003cs.LGcs.AI2024-11被引 6

通过联邦学习降低大模型推理成本,让资源有限者也能参与训练。

eFedLLM: Efficient LLM Inference Based on Federated Learning

  • 用基于Transformer的联邦学习分摊计算与内存负担
  • 结合SVD和内存层次策略,显著提升效率
  • 设计激励机制保障协作安全,适合边缘设备用户

大型语言模型(LLMs)开启了人工智能的新纪元。然而,其庞大的数据规模和参数量对计算与内存资源提出了极高要求,限制了广泛用户和研究者的使用。本文提出一种高效方法,提升LLM推理的运行效率与可负担性。通过基于Transformer的联邦学习(FL)与模型并行分布式训练,将计算负载和内存需求在网络参与者间有效分摊,使资源有限的用户也能协同训练前沿模型。我们还创新性地在FL框架中引入激励机制,奖励有益贡献并过滤恶意行为,保障训练过程的完整性和可靠性。同时,结合内存层次策略与权重矩阵的奇异值分解(SVD),进一步提升计算与内存效率。公式分析与数值计算结果表明,该方法显著优化了资源利用,推动前沿LLM的普及,使更广泛用户既能参与贡献,也能从中受益。

原文摘要 · Abstract (English)

Large Language Models (LLMs) herald a transformative era in artificial intelligence (AI). However, the expansive scale of data and parameters of LLMs requires high-demand computational and memory resources, restricting their accessibility to a broader range of users and researchers. This paper introduces an effective approach that enhances the operational efficiency and affordability of LLM inference. By utilizing transformer-based federated learning (FL) with model-parallel distributed training, our model efficiently distributes the computational loads and memory requirements across a network of participants. This strategy permits users, especially those with limited resources to train state-of-the-art LLMs collaboratively. We also innovate an incentive mechanism within the FL framework, rewarding constructive contributions and filtering out malicious activities, thereby safeguarding the integrity and reliability of the training process. Concurrently, we leverage memory hierarchy strategies and Singular Value Decomposition (SVD) on weight matrices to boost computational and memory efficiencies further. Our results, derived from formulaic analyses and numerical calculations, demonstrate significant optimization of resource use and democratize access to cutting-edge LLMs, ensuring that a wide scale of users can both contribute to and benefit from these advanced models.

联邦学习大模型推理资源优化分布式训练

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