arXiv:2601.00549cs.ITcs.AI2026-01被引 1

提出CoCo-Fed框架,解决边缘联邦学习的内存与通信瓶颈。

CoCo-Fed: A Unified Framework for Memory- and Communication-Efficient Federated Learning at the Wireless Edge

  • 本地通过梯度低秩投影降低内存占用,不增加推理开销。
  • 全局采用正交子空间叠加,单次传输减少带宽消耗超70%。
  • 理论证明在非独立同分布下仍收敛,适合无线传感场景。

在开放无线接入网(O-RAN)架构中部署大规模神经网络是实现原生边缘智能的关键,但面临两大瓶颈:资源受限的gNB进行本地训练所需的巨大内存开销,以及高维模型更新在带宽受限回传链路上的饱和问题。为此,我们提出CoCo-Fed——一种基于压缩与组合的联邦学习统一框架,兼顾本地内存效率与全局通信压缩。本地层面,通过双维度降维处理梯度,将优化器适配至低秩结构,无需引入额外推理参数或延迟。全局层面,设计基于正交子空间叠加的传输协议,将各层更新投影并叠加为每个gNB的单一整合矩阵,显著降低回传流量。我们建立了严格的理论基础,证明了该框架即使在无监督学习条件下(适用于无线感知任务)仍可收敛。在到达角估计任务上的大量仿真表明,CoCo-Fed在内存与通信效率上均显著优于现有基线,且在非独立同分布设置下保持稳健收敛。

原文摘要 · Abstract (English)

The deployment of large-scale neural networks within the Open Radio Access Network (O-RAN) architecture is pivotal for enabling native edge intelligence. However, this paradigm faces two critical bottlenecks: the prohibitive memory footprint required for local training on resource-constrained gNBs, and the saturation of bandwidth-limited backhaul links during the global aggregation of high-dimensional model updates. To address these challenges, we propose CoCo-Fed, a novel Compression and Combination-based Federated learning framework that unifies local memory efficiency and global communication reduction. Locally, CoCo-Fed breaks the memory wall by performing a double-dimension down-projection of gradients, adapting the optimizer to operate on low-rank structures without introducing additional inference parameters/latency. Globally, we introduce a transmission protocol based on orthogonal subspace superposition, where layer-wise updates are projected and superimposed into a single consolidated matrix per gNB, drastically reducing the backhaul traffic. Beyond empirical designs, we establish a rigorous theoretical foundation, proving the convergence of CoCo-Fed even under unsupervised learning conditions suitable for wireless sensing tasks. Extensive simulations on an angle-of-arrival estimation task demonstrate that CoCo-Fed significantly outperforms state-of-the-art baselines in both memory and communication efficiency while maintaining robust convergence under non-IID settings.

联邦学习边缘计算通信效率低秩压缩

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