针对大模型联邦学习资源不均问题,提出高效协同训练新方法。
HeteroTune: Efficient Federated Learning for Large Heterogeneous Models
- 采用可灵活融合的适配器架构,支持异构模型高效聚合。
- 通信开销降低99.5%,峰值内存减少约50%,性能提升4.61%。
- 适合计算与存储资源差异大的分布式场景,如移动端模型微调。
尽管大型预训练模型在各类AI任务中表现卓越,但在隐私敏感且分布式的环境中部署仍面临挑战。联邦学习(FL)通过无需共享数据的去中心化微调提供可行方案,但现实应用受制于客户端算力与内存资源的异质性。为此,我们提出HeteroTune,一种面向大规模异构模型、在有限通信与计算预算下的新型联邦微调范式。其核心为一种新架构DeMA(Dense Mixture of Adapters),可在保持模型完整表征能力的同时,实现异构模型间的灵活高效聚合,并促进跨模型知识融合。此外,我们引入轻量级的CMGA(Cross-Model Gradient Alignment)机制,在聚合过程中对齐不同客户端模型的梯度方向,缓解更新冲突,提升训练稳定性与收敛一致性。理论分析与实证结果表明,HeteroTune在多种任务与模型架构下均达到领先性能与效率。例如,在LLaMA模型上,通信开销减少99.5%,峰值内存降低约50%,性能提升4.61%。
原文摘要 · Abstract (English)
While large pre-trained models have achieved impressive performance across AI tasks, their deployment in privacy-sensitive and distributed environments remains challenging. Federated learning (FL) offers a viable solution by enabling decentralized fine-tuning without data sharing, but real-world applications face significant obstacles due to heterogeneous client resources in compute and memory. To address this, we propose HeteroTune, a novel federated fine-tuning paradigm for large, heterogeneous models operating under limited communication and computation budgets. The core of our method lies in a novel architecture, DeMA (Dense Mixture of Adapters), which enables flexible and efficient aggregation of heterogeneous models by preserving their full representational capacity while facilitating seamless cross-model knowledge fusion. We further introduce CMGA (Cross-Model Gradient Alignment), a lightweight yet effective mechanism that enhances training stability by harmonizing gradient directions across heterogeneous client models during aggregation, mitigating update conflicts and promoting more consistent convergence in federated settings. We provide both theoretical analysis and empirical evidence showing that HeteroTune achieves state-of-the-art performance and efficiency across diverse tasks and model architectures. For example, on LLaMA models, it reduces communication overhead by 99.5%, cuts peak memory usage by ~50%, and improves performance by 4.61%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。