针对异构网络优化大模型联邦微调,显著缩短训练时间。
Adaptive Federated LoRA in Heterogeneous Wireless Networks with Independent Sampling
- 基于独立采样推导新收敛界,无需严格梯度有界假设。
- 联合优化LoRA低秩矩阵比例与采样概率,降低墙时收敛时间。
- 适合资源差异大的设备协同训练,尤其适用于无线环境。
联邦LoRA已成为在分布式设备上高效微调大语言模型的有前景技术,通过减少可训练参数量提升效率。然而,现有方法常忽视系统与数据异构性的理论与实践影响,难以优化整体训练效率,特别是在墙时(wall-clock time)方面表现不足。本文提出一种自适应联邦LoRA策略,结合独立客户端采样,以最小化在计算与通信异构性下的联邦微调收敛墙时。我们首先推导出具有任意且独立客户端采样机制的联邦LoRA新收敛界,显著无需严格的有界梯度假设。随后,设计一种自适应带宽分配方案,考虑异构客户端资源与系统带宽约束。基于该理论,构建并求解一个非凸优化问题,联合确定LoRA压缩率与采样概率,目标是最小化墙时收敛时间。进一步提出一种高效且低复杂度算法近似求解。大量实验表明,所提方法在多种模型与数据集上相比当前最优方法显著降低墙时训练时间。
原文摘要 · Abstract (English)
Federated LoRA has emerged as a promising technique for efficiently fine-tuning large language models (LLMs) on distributed devices by reducing the number of trainable parameters. However, existing approaches often inadequately overlook the theoretical and practical implications of system and data heterogeneity, thereby failing to optimize the overall training efficiency, particularly in terms of wall-clock time. In this paper, we propose an adaptive federated LoRA strategy with independent client sampling to minimize the convergence wall-clock time of federated fine-tuning under both computation and communication heterogeneity. We first derive a new convergence bound for federated LoRA with arbitrary and independent client sampling, notably without requiring the stringent bounded gradient assumption. Then, we introduce an adaptive bandwidth allocation scheme that accounts for heterogeneous client resources and system bandwidth constraints. Based on the derived theory, we formulate and solve a non-convex optimization problem to jointly determine the LoRA sketching ratios and sampling probabilities, aiming to minimize wall-clock convergence time. An efficient and low-complexity algorithm is developed to approximate the solution. Finally, extensive experiments demonstrate that our approach significantly reduces wall-clock training time compared to state-of-the-art methods across various models and datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。