在无标签数据流中,让联邦学习客户端自主选择最有价值的数据进行标注。
Learn How to Query from Unlabeled Data Streams in Federated Learning
- 用多智能体强化学习让各客户端协作决策该标记哪些数据。
- 在图像和文本任务上显著提升模型准确率,优于现有方法。
- 适合数据持续流入且标注成本高的真实联邦学习场景。
联邦学习(FL)允许去中心化客户端在保护本地数据隐私的前提下协同学习。现有研究通常假设训练开始时每个客户端已有离线标注数据,但实际中数据常以无标签流式方式到达客户端。由于标注成本高昂,如何在客户端筛选出有信息量的样本进行标注至关重要。然而,在本地选择样本的同时兼顾全局训练目标,是联邦学习特有的挑战。本文将联邦学习中的数据查询过程建模为一种协作式分布式决策问题,提出名为LeaDQ的有效解决方案,该方案利用多智能体强化学习算法。具体而言,在全局信息隐式引导下,LeaDQ能有效学习各客户端的本地策略,引导其选择有助于提升全局模型精度的样本。在图像与文本任务上的大量仿真结果表明,LeaDQ在多种联邦学习场景中均显著提升了模型性能,优于基准算法。
原文摘要 · Abstract (English)
Federated learning (FL) enables collaborative learning among decentralized clients while safeguarding the privacy of their local data. Existing studies on FL typically assume offline labeled data available at each client when the training starts. Nevertheless, the training data in practice often arrive at clients in a streaming fashion without ground-truth labels. Given the expensive annotation cost, it is critical to identify a subset of informative samples for labeling on clients. However, selecting samples locally while accommodating the global training objective presents a challenge unique to FL. In this work, we tackle this conundrum by framing the data querying process in FL as a collaborative decentralized decision-making problem and proposing an effective solution named LeaDQ, which leverages multi-agent reinforcement learning algorithms. In particular, under the implicit guidance from global information, LeaDQ effectively learns the local policies for distributed clients and steers them towards selecting samples that can enhance the global model's accuracy. Extensive simulations on image and text tasks show that LeaDQ advances the model performance in various FL scenarios, outperforming the benchmarking algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。