用鱼氏信息指导课程学习,加速大模型联邦微调。
Fisher Information-based Efficient Curriculum Federated Learning with Large Language Models
- 基于鱼氏信息自适应选数据,提升训练效率。
- 动态选择层与稀疏参数更新,提速达98.61%。
- 适合资源受限设备上的大模型高效微调。
联邦学习(FL)是一种可协同训练分散数据的有前景范式,可用于微调大语言模型(LLMs)。尽管LLMs规模巨大,训练数据量显著增加,导致计算和通信成本高昂。数据通常非独立同分布(non-IID),需在各设备上进行自适应处理。虽然低秩适配(LoRA)能大幅减少需更新的参数量,但仍需耗费大量时间传输所有层的低秩参数。本文提出一种基于鱼氏信息的高效课程联邦学习框架(FibecFed),包含两项新方法:自适应联邦课程学习与高效稀疏参数更新。首先,提出基于鱼氏信息的方法,在各设备内自适应采样数据,提升FL微调效果。其次,动态选择合适层进行全局聚合,并对局部更新采用稀疏参数与LoRA,以提高效率。基于10个数据集的大量实验表明,FibecFed在准确率上最高达45.35%,微调速度比17种基线方法快98.61%。
原文摘要 · Abstract (English)
As a promising paradigm to collaboratively train models with decentralized data, Federated Learning (FL) can be exploited to fine-tune Large Language Models (LLMs). While LLMs correspond to huge size, the scale of the training data significantly increases, which leads to tremendous amounts of computation and communication costs. The training data is generally non-Independent and Identically Distributed (non-IID), which requires adaptive data processing within each device. Although Low Rank Adaptation (LoRA) can significantly reduce the scale of parameters to update in the fine-tuning process, it still takes unaffordable time to transfer the low-rank parameters of all the layers in LLMs. In this paper, we propose a Fisher Information-based Efficient Curriculum Federated Learning framework (FibecFed) with two novel methods, i.e., adaptive federated curriculum learning and efficient sparse parameter update. First, we propose a fisher information-based method to adaptively sample data within each device to improve the effectiveness of the FL fine-tuning process. Second, we dynamically select the proper layers for global aggregation and sparse parameters for local update with LoRA so as to improve the efficiency of the FL fine-tuning process. Extensive experimental results based on 10 datasets demonstrate that FibecFed yields excellent performance (up to 45.35% in terms of accuracy) and superb fine-tuning speed (up to 98.61% faster) compared with 17 baseline approaches).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。