联邦学习中同时解决未知类别和未知领域问题的联合方法
Federated Joint Learning for Domain and Class Generalization
- 分组训练类通用模型,避免决策边界混淆
- 基于领域相似性聚合结果,提升跨域泛化能力
- 适合需要强泛化的联邦视觉语言任务
由于视觉-语言模型(如CLIP)参数量大、预训练成本高,高效微调变得至关重要。现有方法通常仅关注未见类别或未见领域中的一个,缺乏联合处理框架。本文提出联邦联合学习领域与类别泛化方法FedDCG,通过领域分组策略,在每组内训练类通用网络以避免决策边界混淆。推理时根据领域相似性聚合类通用结果,融合类别与领域泛化知识。具体采用可学习网络增强类泛化能力,并引入解耦机制分离通用与领域特定知识,提升对未见领域的适应性。在多个数据集上的大量实验表明,FedDCG在准确率和鲁棒性上均优于现有最优基线。
原文摘要 · Abstract (English)
Efficient fine-tuning of visual-language models like CLIP has become crucial due to their large-scale parameter size and extensive pretraining requirements. Existing methods typically address either the issue of unseen classes or unseen domains in isolation, without considering a joint framework for both. In this paper, we propose \textbf{Fed}erated Joint Learning for \textbf{D}omain and \textbf{C}lass \textbf{G}eneralization, termed \textbf{FedDCG}, a novel approach that addresses both class and domain generalization in federated learning settings. Our method introduces a domain grouping strategy where class-generalized networks are trained within each group to prevent decision boundary confusion. During inference, we aggregate class-generalized results based on domain similarity, effectively integrating knowledge from both class and domain generalization. Specifically, a learnable network is employed to enhance class generalization capabilities, and a decoupling mechanism separates general and domain-specific knowledge, improving generalization to unseen domains. Extensive experiments across various datasets show that \textbf{FedDCG} outperforms state-of-the-art baselines in terms of accuracy and robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。