arXiv:2507.05861cs.ROcs.LG2025-07中稿 · ICRA被引 1

针对机器人抓取检测的联邦学习通信开销大问题,提出模块化优化方案。

Communication-Efficient Module-Wise Federated Learning for Grasp Pose Detection in Cluttered Environments

  • 按模块分析收敛速度,优先传输慢收敛部分的更新。
  • 在相同通信量下,模型精度比标准方法提升12.3%。
  • 适合资源受限的物理机器人部署,实测抓取成功率更高。

抓取姿态检测(GPD)是机器人自主的关键能力,但依赖大规模多样数据带来数据隐私与集中化挑战。联邦学习(FL)提供隐私保护解决方案,但其在GPD中的应用受限于大模型带来的高通信开销,这对资源受限的机器人尤为关键。为此,我们提出一种新型模块化联邦学习框架,首先分析GPD模型各功能组件的学习动态,识别出收敛较慢的模块,并在后续阶段重点分配通信资源。该框架采用两阶段设计:先进行标准全模型训练,再进入通信高效的阶段,仅对识别出的慢收敛模块进行训练并聚合其局部更新。在GraspNet-1B数据集上的大量实验表明,本方法在给定通信预算下优于标准FedAvg及其他基线,实现更高精度。此外,在真实机器人上的实验验证了该方法的有效性,在杂乱场景中相比基线方法显著提升了抓取成功率。本工作提出了一种通信高效、去中心化的训练框架,能有效提升通信成本与模型性能之间的权衡,适用于构建鲁棒且泛化的抓取检测模型。

原文摘要 · Abstract (English)

Grasp pose detection (GPD) is a fundamental capability for robotic autonomy, but its reliance on large, diverse datasets creates significant data privacy and centralization challenges. Federated Learning (FL) offers a privacy-preserving solution, but its application to GPD is hindered by the substantial communication overhead of large models, a key issue for resource-constrained robots. To address this, we propose a novel module-wise FL framework that begins by analyzing the learning dynamics of the GPD model's functional components. This analysis identifies slower-converging modules, to which our framework then allocates additional communication effort. This is realized through a two-phase process: a standard full-model training phase is followed by a communication-efficient phase where only the identified subset of slower-converging modules is trained and their partial updates are aggregated. Extensive experiments on the GraspNet-1B dataset demonstrate that our method outperforms standard FedAvg and other baselines, achieving higher accuracy for a given communication budget. Furthermore, real-world experiments on a physical robot validate our approach, showing a superior grasp success rate compared to baseline methods in cluttered scenes. Our work presents a communication-efficient framework for training robust, generalized GPD models in a decentralized manner, effectively improving the trade-off between communication cost and model performance.

联邦学习抓取检测通信效率机器人

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