arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.04400cs.LGcs.AIcs.CRcs.MM
收藏

自适应原型知识迁移用于混合模态与异构任务的联邦学习

Adaptive Prototype Knowledge Transfer for Federated Learning with Mixed Modalities and Heterogeneous Tasks

Keke Gai, Mohan Wang, Jing Yu, Dongjue Wang, Qi Wu

AI总结:

针对混合模态与异构任务下联邦学习标签不统一和任务异构问题,提出自适应原型知识迁移框架AproMFL,通过动态原型构建与聚合权重调整,在三个异构数据集上准确率和召回率分别提升0.42%~6.09%和1.6%~3.89%。

摘要:

混合模态的多模态联邦学习(MFL)使单模态和多模态客户端能够协作训练模型,同时确保客户端的隐私。作为局部数据的代表性样本,原型为混合模态的MFL提供了一种资源消耗低且不依赖先验知识的方法。然而,现有的基于原型的MFL方法假设客户端之间具有统一的标签且每个客户端的任务相同,这在混合模态的MFL中是不切实际的。在本工作中,我们提出了一种用于混合模态的自适应基于原型的多模态联邦学习(AproMFL)框架,以解决上述问题。我们的AproMFL通过自适应构建的原型在无需统一标签的情况下迁移知识。客户端根据标签自适应地选择原型构建方法;服务器将客户端原型转换为统一的多模态原型,并对其进行聚类以形成全局原型。为了解决任务异构性中的模型聚合问题,我们开发了一种基于客户端关系图的方案来动态调整聚合权重。此外,我们提出了全局原型知识迁移损失和全局模型知识迁移损失,以实现全局知识向局部知识的迁移。实验结果表明,在三个高度异构的数据集(α=0.1)和两个异构任务上,AproMFL优于四个基线,其准确率和召回率的最佳结果分别比FedIoT(基于FedAvg的MFL)高出0.42%~6.09%和1.6%~3.89%。

英文摘要:

Multimodal Federated Learning (MFL) with mixed modalities enables unimodal and multimodal clients to collaboratively train models while ensuring clients' privacy. As a representative sample of local data, prototypes offer an approach with low resource consumption and no reliance on prior knowledge for MFL with mixed modalities. However, existing prototype-based MFL methods assume unified labels across clients and identical tasks per client, which is impractical in MFL with mixed modalities. In this work, we propose an Adaptive prototype-based Multimodal Federated Learning (AproMFL) framework for mixed modalities to address the aforementioned issues. Our AproMFL transfers knowledge through adaptively-constructed prototypes without unified labels. Clients adaptively select prototype construction methods in line with labels; server converts client prototypes into unified multimodal prototypes and cluster them to form global prototypes. To address model aggregation issues in task heterogeneity, we develop a client relationship graph-based scheme to dynamically adjust aggregation weights. Furthermore, we propose a global prototype knowledge transfer loss and a global model knowledge transfer loss to enable the transfer of global knowledge to local knowledge. Experimental results show that AproMFL outperforms four baselines on three highly heterogeneous datasets ($α=0.1$) and two heterogeneous tasks, with the optimal results in accuracy and recall being 0.42%~6.09% and 1.6%~3.89% higher than those of FedIoT (FedAvg-based MFL), respectively.

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

↑