arXiv:2509.26524cs.LGcs.AI2025-09被引 2

针对多任务多模态联邦学习中的个性化难题,提出两阶段自适应优化方法。

TAP: Two-Stage Adaptive Personalization of Multi-Task and Multi-Modal Foundation Models in Federated Learning

  • 利用客户端与服务器模型结构差异,分阶段选择性更新参数以减少干扰。
  • 在全局模型稳定后进行知识蒸馏,恢复通用结构提升泛化能力。
  • 适用于数据、任务、模态均异构的联邦学习场景,尤其适合复杂多模态应用。

在联邦学习中,本地模型个性化受到广泛关注,但基础模型的个性化微调仍缺乏深入研究。尤其在客户端间存在数据、任务和模态多重异质性的场景下,现有方法难以有效应对。为此,本文提出两阶段自适应个性化(TAP)方法。第一阶段利用客户端与服务器间不匹配的模型架构,选择性地用全局更新替换个性化参数,显式限制跨任务与跨模态干扰。第二阶段在全局模型训练完成后进行后置知识蒸馏,仅在模型稳定后重新引入可泛化的知识,从而在不牺牲个性化的前提下增强模型泛化能力。本文首次提供了在跨客户端模态-任务对异质性条件下联邦基础模型训练的收敛性分析,并验证了模态-任务对数量对微调效果的影响。大量实验表明,TAP在多种数据集和任务上均显著优于当前最优基线。代码已公开于 https://github.com/lee3296/TAP。

原文摘要 · Abstract (English)

In federated learning (FL), local personalization of models has received significant attention, yet personalized fine-tuning of foundation models remains underexplored. In particular, there is a lack of understanding in the literature on how to personalize foundation models in settings where there exist heterogeneity not only in data, but also in tasks and modalities across the clients. To address this gap, we propose Two-Stage Adaptive Personalization (TAP). In the first stage, TAP leverages mismatched model architectures between clients and the server to selectively replace personalized parameters with global updates, explicitly limiting cross-task and cross-modality interference. In the second stage, TAP conducts post-FL distillation on the global model to recover a beneficial shared structure. By reintroducing generalizable knowledge only after the global model has stabilized, TAP enhances generalization without compromising personalization. In developing our methodology, we introduce the first convergence analysis of federated foundation model training at the server under modality-task pair heterogeneity across clients, and demonstrate the impact of the number of modality-task pairs on model fine-tuning. Through extensive experiments, we demonstrate the effectiveness of TAP across a variety of datasets and tasks in comparison to state-of-the-art baselines. The implementation code is publicly available at https://github.com/lee3296/TAP.

联邦学习多模态个性化基础模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。