解决联邦学习中持续更新模型时遗忘旧知识的问题
Federated Continual Instruction Tuning
- 用动态知识组织和子空间激活应对持续学习挑战
- 在不同数据分布下显著减少灾难性遗忘,提升性能
- 适合资源有限但需持续优化模型的研究者
大型多模态模型(LMMs)的出色表现依赖海量指令微调数据,但监督微调带来的计算成本和数据收集需求让多数研究者难以承受。联邦学习(FL)有望整合分布式数据与算力以降低联合训练开销。然而,现有方法多假设任务数量固定,而现实场景中客户端持续接触新知识,常因内存限制无法保留旧任务。本文提出联邦持续指令微调(FCIT)基准,包含两个真实场景、四种设置及十二个精心筛选的指令微调数据集。为应对挑战,我们提出动态知识组织机制以有效融合各任务更新,并设计子空间选择性激活机制在推理时分配特定任务输出。大量实验表明,该方法在不同数据异质性和灾难性遗忘条件下均显著提升模型性能。代码与数据集已开源:https://github.com/Ghy0501/FCIT。
原文摘要 · Abstract (English)
A vast amount of instruction tuning data is crucial for the impressive performance of Large Multimodal Models (LMMs), but the associated computational costs and data collection demands during supervised fine-tuning make it impractical for most researchers. Federated learning (FL) has the potential to leverage all distributed data and training resources to reduce the overhead of joint training. However, most existing methods assume a fixed number of tasks, while in real-world scenarios, clients continuously encounter new knowledge and often struggle to retain old tasks due to memory constraints. In this work, we introduce the Federated Continual Instruction Tuning (FCIT) benchmark to model this real-world challenge. Our benchmark includes two realistic scenarios, encompassing four different settings and twelve carefully curated instruction tuning datasets. To address the challenges posed by FCIT, we propose dynamic knowledge organization to effectively integrate updates from different tasks during training and subspace selective activation to allocate task-specific output during inference. Extensive experimental results demonstrate that our proposed method significantly enhances model performance across varying levels of data heterogeneity and catastrophic forgetting. Code and dataset are released at https://github.com/Ghy0501/FCIT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。