arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.03097cs.LGcs.AI
收藏

联邦集成引导的离线强化学习

Federated Ensemble-Directed Offline Reinforcement Learning

Texas A&M University(德克萨斯农工大学)

Desik Rengarajan, Nitin Ragothaman, Dileep Kalathil, Srinivas Shakkottai

AI总结:

针对联邦离线强化学习中简单结合标准方法性能差的问题,提出基于集成学习提炼客户端智慧的FEDORA算法,在多类环境和真实机器人上均取得更优性能。

摘要:

我们研究联邦离线强化学习(RL)问题:在该场景下,分布式智能体仅能借助由不同未知行为策略生成的小规模预收集数据集,协作学习高质量控制策略。若将标准离线RL方法与标准联邦学习方法简单结合来解决此问题,会得到性能不佳的策略。为此,我们提出联邦集成引导离线强化学习算法(Federated Ensemble-Directed Offline Reinforcement Learning Algorithm, FEDORA),该算法通过集成学习方法提炼各客户端的集体智慧。我们开发了FEDORA代码库,可利用联邦学习平台上的分布式计算资源。实验表明,在多种复杂连续控制环境和真实数据集上,FEDORA的性能显著优于其他方法,包括基于合并数据池的离线RL方法。最后,我们在真实移动机器人上验证了FEDORA的性能。我们的代码和实验视频可在https://github.com/DesikRengarajan/FEDORA获取。

英文摘要:

We consider the problem of federated offline reinforcement learning (RL), a scenario under which distributed learning agents must collaboratively learn a high-quality control policy only using small pre-collected datasets generated according to different unknown behavior policies. Naïvely combining a standard offline RL approach with a standard federated learning approach to solve this problem can lead to poorly performing policies. In response, we develop the Federated Ensemble-Directed Offline Reinforcement Learning Algorithm (FEDORA), which distills the collective wisdom of the clients using an ensemble learning approach. We develop the FEDORA codebase to utilize distributed compute resources on a federated learning platform. We show that FEDORA significantly outperforms other approaches, including offline RL over the combined data pool, in various complex continuous control environments and real-world datasets. Finally, we demonstrate the performance of FEDORA in the real-world on a mobile robot. We provide our code and a video of our experiments at \url{https://github.com/DesikRengarajan/FEDORA}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑