arXiv:2605.05896cs.LGcs.AI2026-05

基于验证损失提升选优质客户端,加速物联网联邦学习收敛。

VARS-FL: Validation-Aligned Client Selection for Non-IID Federated Learning in IoT Systems

论文配图:VARS-FL: Validation-Aligned Client Selection for Non-IID Federated Learning in IoT Systems
图 1 · 摘自论文原文
  • 用客户端更新带来的验证损失下降衡量贡献,更贴近全局目标。
  • 在15类非独立同分布物联网入侵检测任务中,准确率提升36%以上。
  • 无需改动训练流程,适合资源受限的工业物联网场景。

联邦学习系统通常采用无状态客户端选择策略,每轮独立决策,忽略客户端历史贡献。在非独立同分布数据下,这导致收敛慢、训练不稳定,尤其当依赖局部代理指标(如训练损失)时,其与全局优化目标存在偏差。这一问题在物联网与工业物联网环境中尤为突出,因设备间数据高度异构,观测到不同的流量模式。本文提出VARS-FL(验证对齐声誉评分联邦学习),通过客户端更新引起的服务器端验证损失降低来量化其贡献。每轮信号聚合为包含滑动窗口近期贡献平均值和对数缩放参与度的声誉分数,实现稳健的探索-利用平衡。该方法不改变本地训练或聚合过程,完全兼容标准FedAvg。在包含100个客户端的Edge-IIoTset数据集上,针对15类非独立同分布物联网入侵检测任务进行多种子评估,对比FedAvg、Oort和Power-of-Choice,VARS-FL始终提升准确率、F1-Macro和损失,并加快收敛(达到80%准确率所需轮数减少最多达36%)。结果表明,验证对齐且历史感知的客户端选择能显著提升异构物联网环境下联邦学习的可靠性与效率。

原文摘要 · Abstract (English)

Federated learning (FL) systems typically employ stateless client selection, treating each communication round independently and ignoring accumulated evidence of client contribution quality. Under non-IID data, this leads to slow convergence and unstable training, particularly when selection relies on local proxies (e.g., training loss) that are misaligned with the global optimization objective. These challenges are especially pronounced in Internet of Things (IoT) and Industrial IoT (IIoT) environments, where data is highly heterogeneous and distributed across devices observing different traffic patterns. In this paper, we propose VARS-FL (Validation-Aligned Reputation Scoring for Federated Learning), a client selection framework that quantifies each client's contribution using the reduction in server-side validation loss induced by its update. These per-round signals are aggregated into a Reputation score that combines a sliding-window average of recent contributions with a logarithmically scaled participation term, enabling robust exploration-exploitation selection. VARS-FL requires no changes to local training or aggregation and remains fully compatible with standard FedAvg. We evaluate VARS-FL on a 15-class non-IID IoT intrusion detection task using the Edge-IIoTset dataset, with 100 clients across multiple seeds, and compare it against FedAvg, Oort, and Power-of-Choice. VARS-FL consistently improves accuracy, F1-Macro, and loss, while accelerating convergence (up to 36% fewer rounds to reach 80% accuracy). These results demonstrate that validation-aligned, history-aware client selection provides a more reliable and efficient training process for federated learning in heterogeneous IoT environments.

联邦学习物联网客户端选择非独立同分布

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