arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.06332cs.LGcs.AI
收藏

联邦微调中低秩适配器的聚合

Aggregating Low Rank Adapters in Federated Fine-tuning

Apheris AI(Apheris AI公司)

Evelyn Trautmann, Ian Hales, Martin F. Volk

AI总结:

针对大模型联邦微调的高成本与通信开销问题,本文提出一种新型低秩适配器聚合方法,在GLUE基准上与现有方法对比评估性能。

摘要:

大语言模型的微调需要高昂的计算与内存资源,因此成本极高。在联邦数据集上训练时,还会额外增加通信开销。为此,参数高效微调方法(PEFT)正变得愈发重要。在此领域,采用低秩适配方法(LoRA)进行微调已取得了十分优异的效果。LoRA方法在联邦学习中的应用,尤其是适配矩阵的聚合,是当前的一个研究方向。本文提出了一种全新的聚合方法,将其与多种现有低秩适配器聚合方法进行对比——这些适配器均在大机器学习模型的联邦微调中训练得到,并在选定的GLUE基准数据集上评估了各类方法的性能。

英文摘要:

Fine-tuning large language models requires high computational and memory resources, and is therefore associated with significant costs. When training on federated datasets, an increased communication effort is also needed. For this reason, parameter-efficient methods (PEFT) are becoming increasingly important. In this context, very good results have already been achieved by fine-tuning with low-rank adaptation methods (LoRA). The application of LoRA methods in Federated Learning, and especially the aggregation of adaptation matrices, is a current research field. In this article, we propose a novel aggregation method and compare it with different existing aggregation methods of low rank adapters trained in a federated fine-tuning of large machine learning models and evaluate their performance with respect to selected GLUE benchmark datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑