arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2207.08147cs.LG
收藏

多任务和迁移学习在联邦学习中的应用

Multi-Task and Transfer Learning for Federated Learning Applications

Texas A&M University(德州农工大学);Texas A&M University at Qatar(德州农工大学卡塔尔分校)

Cihat Keçeci, Mohammad Shaqfeh, Hayat Mbayed, Erchin Serpedin

摘要:

联邦学习使许多应用程序受益于大量潜在数据持有客户端的分布式和私有数据集。然而,不同的客户端通常有自己的特定目标,从数据中学习的任务。因此,使用元学习工具(如多任务学习和迁移学习)支持联邦学习将有助于扩大联邦学习的潜在应用,让不同但相关任务的客户共享任务不可知的模型,然后每个客户可以进一步更新和定制特定任务。在联邦多任务学习问题中,训练的深度神经网络模型应该针对每个客户端的各自目标进行微调,同时共享一些参数以提高泛化能力。我们提出训练一个深度神经网络模型,其中更广义的层更接近输入,更个性化的层更接近输出。我们通过引入层类型来实现这一点,例如预训练层、通用层、特定任务层和个人层。我们提供了模拟结果,以突出特定的场景中,基于元学习的联邦学习被证明是有用的。

英文摘要:

Federated learning enables many applications benefiting distributed and private datasets of a large number of potential data-holding clients. However, different clients usually have their own particular objectives in terms of the tasks to be learned from the data. So, supporting federated learning with meta-learning tools such as multi-task learning and transfer learning will help enlarge the set of potential applications of federated learning by letting clients of different but related tasks share task-agnostic models that can be then further updated and tailored by each individual client for its particular task. In a federated multi-task learning problem, the trained deep neural network model should be fine-tuned for the respective objective of each client while sharing some parameters for more generalizability. We propose to train a deep neural network model with more generalized layers closer to the input and more personalized layers to the output. We achieve that by introducing layer types such as pre-trained, common, task-specific, and personal layers. We provide simulation results to highlight particular scenarios in which meta-learning-based federated learning proves to be useful.

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

↑