arXiv:2503.14553cs.CVcs.LG2025-03中稿 · publication in IEE…被引 9

用视觉嵌入重新定义联邦学习中的非独立同分布数据,更真实反映任务差异。

Redefining non-IID Data in Federated Learning for Computer Vision Tasks: Migrating from Labels to Embeddings for Task-Specific Data Distributions

  • 基于预训练模型提取视觉嵌入,按任务特性构建数据异质性
  • 在7个视觉任务上,使FedAvg损失最高提升60%
  • 为多任务视觉联邦学习提供新基准,适合研究数据分布问题的学者

联邦学习(FL)已成为分布式机器学习的重要范式,但其性能在客户端数据分布非独立同分布(non-IID)时会显著下降。现有研究主要通过标签分布偏移模拟数据异质性,但该方法在计算机视觉多任务场景下存在局限。本文提出从标签转向嵌入的视角,利用预训练深度神经网络提取任务特定的数据嵌入,定义基于嵌入的新型数据异质性。方法包括基于嵌入聚类并用狄利克雷分布分配数据至客户端。在7个代表性视觉任务上的实验表明,该异质性设置使FedAvg的损失最高上升约60%,更真实揭示了数据异质性带来的性能退化。同时引入新基准评估指标,揭示多个待探索的研究方向。代码已开源。

原文摘要 · Abstract (English)

Federated Learning (FL) has emerged as one of the prominent paradigms for distributed machine learning (ML). However, it is well-established that its performance can degrade significantly under non-IID (non-independent and identically distributed) data distributions across clients. To study this effect, the existing works predominantly emulate data heterogeneity by imposing label distribution skew across clients. In this paper, we show that label distribution skew fails to fully capture the data heterogeneity in computer vision tasks beyond classification, exposing an overlooked gap in the literature. Motivated by this, by utilizing pre-trained deep neural networks to extract task-specific data embeddings, we define task-specific data heterogeneity through the lens of each vision task and introduce a new level of data heterogeneity called embedding-based data heterogeneity. Our methodology involves clustering data points based on embeddings and distributing them among clients using the Dirichlet distribution. Through extensive experiments, we evaluate the performance of different FL methods under our revamped notion of data heterogeneity, introducing new benchmark performance measures to the literature. For instance, across seven representative computer vision tasks, our embedding-based heterogeneity formulation leads to up to around 60% increase in the observed loss under FedAvg, indicating that it more accurately exposes the performance degradation caused by data heterogeneity. We further unveil a series of open research directions that can be pursued. (Code: https://github.com/KasraBorazjani/task-perspective-het.git)

联邦学习非IID数据视觉嵌入多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。