arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.13091cs.LG

Auto-Agent-Distiller:通过神经架构搜索实现高效深度强化学习智能体

Auto-Agent-Distiller: Towards Efficient Deep Reinforcement Learning Agents via Neural Architecture Search

Rice University(莱斯大学)

Yonggan Fu, Zhongzhi Yu, Yongan Zhang, Yingyan Celine Lin

AI总结:

针对DRL模型复杂度高、难以实时部署的问题,本文提出首个面向DRL的NAS框架A2D,并通过同时蒸馏教师智能体actor与critic的知识稳定搜索,获得任务特定且兼顾分数与效率的智能体。

摘要:

AlphaGo的惊人表现激发了人们将深度强化学习(DRL)开发用于智能机器人等众多现实应用的爆发式兴趣。然而,DRL通常高得令人难以承受的复杂性,与许多DRL应用所需的实时控制和受限资源相矛盾,限制了由DRL驱动的智能设备的巨大潜力。尽管已有大量工作致力于压缩其他深度学习模型,但现有研究几乎未触及DRL压缩的表层。在本工作中,我们首先发现存在一个能够同时最大化测试分数和效率的最优DRL模型规模,这激发了对任务特定DRL智能体的需求。因此,我们提出了Auto-Agent-Distiller(A2D)框架,据我们所知,这是首个应用于DRL的神经架构搜索(NAS)方法,可为不同任务自动搜索同时优化测试分数和效率的最优DRL智能体。具体而言,我们证明了普通NAS在搜索最优智能体时很容易失败,因为其会在DRL训练稳定性中导致高方差;随后我们开发了一种新颖的蒸馏机制,从教师智能体的actor和critic中蒸馏知识,以稳定搜索过程并提升被搜索智能体的最优性。大量实验和消融研究一致验证了我们的发现以及A2D的优势和普遍适用性,其在测试分数和效率上均优于人工设计的DRL。所有代码将在论文被接收后发布。

英文摘要:

AlphaGo's astonishing performance has ignited an explosive interest in developing deep reinforcement learning (DRL) for numerous real-world applications, such as intelligent robotics. However, the often prohibitive complexity of DRL stands at the odds with the required real-time control and constrained resources in many DRL applications, limiting the great potential of DRL powered intelligent devices. While substantial efforts have been devoted to compressing other deep learning models, existing works barely touch the surface of compressing DRL. In this work, we first identify that there exists an optimal model size of DRL that can maximize both the test scores and efficiency, motivating the need for task-specific DRL agents. We therefore propose an Auto-Agent-Distiller (A2D) framework, which to our best knowledge is the first neural architecture search (NAS) applied to DRL to automatically search for the optimal DRL agents for various tasks that optimize both the test scores and efficiency. Specifically, we demonstrate that vanilla NAS can easily fail in searching for the optimal agents, due to its resulting high variance in DRL training stability, and then develop a novel distillation mechanism to distill the knowledge from both the teacher agent's actor and critic to stabilize the searching process and improve the searched agents' optimality. Extensive experiments and ablation studies consistently validate our findings and the advantages and general applicability of our A2D, outperforming manually designed DRL in both the test scores and efficiency. All the codes will be released upon acceptance.

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

↑