针对部分可见的联邦学习,提出时空注意力强化学习框架
Federated Client Selection under Partial Visibility: A POMDP Approach with Spatio-Temporal Attention

- 将客户端选择建模为部分可观测马尔可夫决策过程
- 在异构和部分可见场景下优于现有基线方法
- 适合大规模边缘计算中的动态客户端管理
联邦学习依赖有效的客户端选择以缓解数据异构带来的性能下降。现有方法通常假设服务器在每轮通信中可访问所有客户端,但在大规模或边缘部署场景下,受通信、移动性或可用性限制,服务器仅能访问部分客户端,导致每轮仅可观测子集。本文将部分可见条件下的客户端选择建模为部分可观测马尔可夫决策过程(POMDP),提出基于时空注意力的强化学习框架。通过融合历史全局模型与客户端身份嵌入,该方法同时捕捉训练的时序上下文与客户端的持久特征。多数据集实验表明,在异构和部分可见设置下,本方法显著优于现有基线,验证了其在实际联邦学习系统不完全观测挑战中的有效性。
原文摘要 · Abstract (English)
Federated learning relies on effective client selection to alleviate the performance degradation caused by data heterogeneity. Most existing methods assume full visibility of all clients at each communication round. However, in large-scale or edge-based deployments, the server can only access a subset of clients due to communication, mobility, or availability constraints, resulting in partial visibility where only a subset of clients is observable for aggregation in each communication round. In this paper, we formulate federated client selection under partial visibility as a Partially Observable Markov Decision Process (POMDP) and propose a Spatial-Temporal attention-based reinforcement learning framework. By integrating historical global models and client identity embeddings, the proposed method captures both the temporal contexts of training and the persistent characteristics of clients. Experimental results across multiple datasets demonstrate that our approach achieves superior performance compared to existing baselines in heterogeneous and partially visible settings, validating its effectiveness in addressing the challenges of incomplete observations in practical federated learning systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。