arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2404.01232cs.CLcs.CV

基于多模态原型的开放词汇联邦学习

Open-Vocabulary Federated Learning with Multimodal Prototyping

Huimin Zeng, Zhenrui Yue, Dong Wang

AI总结:

针对联邦学习中新用户提出任意未见类别查询的开放词汇难题,本文提出Fed-MP框架,利用轻量客户端残差自适应聚合VLM并结合多模态原型机制,提升模型对未知类别的泛化能力。

摘要:

现有的联邦学习(FL)研究通常假设训练标签空间与测试标签空间完全相同。然而,在真实应用中,这一假设过于理想化而难以成立。新用户可能提出涉及未见类别数据的查询,而这类开放词汇查询会直接导致此类FL系统失效。因此,本工作明确聚焦于FL中尚未得到充分研究的开放词汇挑战。也就是说,对于新用户,全局服务器应当理解其涉及任意未知类别的查询。为解决该问题,我们利用预训练视觉-语言模型(VLM)。具体而言,我们提出了一种在FL场景下为VLM量身定制的新型适配框架,名为Federated Multimodal Prototyping(Fed-MP,联邦多模态原型)。Fed-MP基于轻量级客户端残差自适应地聚合本地模型权重,并通过一种新颖的多模态原型机制进行预测。Fed-MP利用从已见类别中学到的知识,增强适配后VLM对未见类别的鲁棒性。我们在多个数据集上的实证评估验证了Fed-MP的有效性。

英文摘要:

Existing federated learning (FL) studies usually assume the training label space and test label space are identical. However, in real-world applications, this assumption is too ideal to be true. A new user could come up with queries that involve data from unseen classes, and such open-vocabulary queries would directly defect such FL systems. Therefore, in this work, we explicitly focus on the under-explored open-vocabulary challenge in FL. That is, for a new user, the global server shall understand her/his query that involves arbitrary unknown classes. To address this problem, we leverage the pre-trained vision-language models (VLMs). In particular, we present a novel adaptation framework tailored for VLMs in the context of FL, named as Federated Multimodal Prototyping (Fed-MP). Fed-MP adaptively aggregates the local model weights based on light-weight client residuals, and makes predictions based on a novel multimodal prototyping mechanism. Fed-MP exploits the knowledge learned from the seen classes, and robustifies the adapted VLM to unseen categories. Our empirical evaluation on various datasets validates the effectiveness of Fed-MP.

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

↑