arXiv:2410.14852cs.LGcs.CR2024-10NAACL被引 14

在保护隐私的前提下,实现大模型的高效压缩。

FedSpaLLM: Federated Pruning of Large Language Models

  • 通过联邦学习框架本地化剪枝,无需共享数据。
  • 支持不同设备资源,剪枝后模型仍保持高精度。
  • 适合医疗、金融等需隐私保护的场景使用。

大语言模型(LLMs)虽性能卓越,但部署成本高昂。剪枝可减小模型规模,但现有方法依赖公开校准数据,不适用于隐私敏感场景。为此,我们提出首个专为剪枝设计的联邦学习框架FedSpaLLM,使客户端在私有数据上本地剪枝,同时应对系统异构性并保持通信效率。框架引入三项创新:(1) 新型ℓ₀-范数聚合函数,仅对非零权重进行平均,保留关键参数;(2) 自适应掩码扩展技术,满足全局稀疏目标的同时兼容客户端个性化剪枝决策;(3) 层采样策略,降低通信开销并根据客户端资源个性化剪枝过程。大量实验表明,FedSpaLLM在多样化联邦设置中显著提升剪枝性能。

原文摘要 · Abstract (English)

Large Language Models (LLMs) achieve state-of-the-art performance but are challenging to deploy due to their high computational and storage demands. Pruning can reduce model size, yet existing methods assume public access to calibration data, which is impractical for privacy-sensitive applications. To address the challenge of pruning LLMs in privacy-preserving settings, we propose FedSpaLLM, the first federated learning framework designed specifically for pruning LLMs. FedSpaLLM enables clients to prune their models locally based on private data while accounting for system heterogeneity and maintaining communication efficiency. Our framework introduces several key innovations: (1) a novel $\ell_0$-norm aggregation function that ensures only non-zero weights are averaged across clients, preserving important model parameters; (2) an adaptive mask expansion technique that meets global sparsity targets while accommodating client-specific pruning decisions; and (3) a layer sampling strategy that reduces communication overhead and personalizes the pruning process based on client resources. Extensive experiments show that FedSpaLLM improves pruning performance in diverse federated settings.

大模型剪枝联邦学习隐私保护

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