今天,deepmind开源了一个内部强化学习库trfl,用于在tensorflow中编写强化学习智能体。这个库包含deepmind开发的大量成功的agent的关键算法组件,对于强化学习智能体的再现研究很有用。
今天,deepmind开源了一个新的高效的构建模块库,用于在tensorflow中编写强化学习(rl)智能体。这个库名为trfl(发音为’truffle’),代表了deepmind内部用于大量非常成功的agent的关键算法组件集合,如dqn,ddpg和impala(importance weighted actor learner architecture)。
一个典型的深度强化学习agent由大量的交互组件组成:至少包括环境(environment)和一些表示价值(value)或策略(policy)的深层网络,但它们通常也包括组件,例如环境的学习模型,伪奖励函数或replay系统。
这些组件通常以微妙的方式相互作用(通常没有在论文中详细记录),因此很难在如此庞大的计算图中识别错误。openai最近发表的一篇博客文章通过分析强化学习智能体的一些最流行的开源实现,凸显了这个问题,他们发现10个实现中有6个“具有社区成员发现并被作者确认的小bug”。
解决这个问题的一种方法是通过开源的完整agent实现,帮助研究社区复现论文的结果。例如,我们最近发布了一种高度可扩展的分布式训练架构impala,使用v-trace agent探索在单个智能体上完成多种任务的挑战。
这些大型agent代码库对于再现研究非常有用,但也很难修改和扩展。一种不同的、互补的方法是提供可靠的、经过良好测试的通用构建块实现,可以在各种不同的rl智能体中使用。此外,通过将这些核心组件抽象到单个库中,使用一致的api,可以更轻松地组合来自许多不同论文的创新想法。
trfl库包含实现经典rl算法以及更前沿技术的许多函数。这里提供的损失函数和其他操作是在纯tensorflow中实现的。它们不是完整的算法,而是在构建功能齐全的rl智能体时所需的特定于rl的数学运算实现。
对于value-based的强化学习,我们提供tensorflow ops 用于在离散动作空间中学习,例如td-learning,sarsa, q-learning及其变体,以及用于实现连续控制算法的操作,例如dpg。
trfl库还包含用于学习分配价值函数的ops。这些ops支持批处理,并通过将其输入到tensorflow optimiser来返回可以最小化的损失。一些损失函数在批转换运行(例如sarsa,q-learning......),其他一些损失在多批轨迹上运行(例如q lambda,retrace,......)。
对于基于策略的方法,trfl提供实用程序,可以轻松实现a2c等在线方法,以及支持off-policy纠正技术,如v-trace。trfl还支持连续动作空间中的策略梯度计算。
最后,trfl还提供了unreal使用的辅助伪奖励函数(pseudo-reward functions)的实现,我们发现它可以提高各个领域的数据效率。
这不是一次性的发布。由于trfl库在deepmind内部广泛使用,因此我们将继续维护它,并随着时间的推移添加新功能。我们也渴望得到更广泛的rl社区对这个库的贡献。
trfl库由deepmind研究工程团队创建。
开源获取地址:
https://github.com/deepmind/trfl
开源代码
安装
可以使用以下命令从github直接从pip安装trfl:pip install git+git://github.com/deepmind/trfl.git
trfl同时适用于tensorflow的cpu和gpu版本,但它没有将tensorflow列为一个requirement,因此你需要单独安装tensorflow和tensorflow-probability。
用例
loss是代表损失的张量。对于q-learning,它是预测的q-values和td targets之间的平方差的一半。
额外信息位于q_learning命名元组中,包括q_learning.td_error和q_learning.target。
大多数情况下,你可能只对loss感兴趣:
该模块中的所有损失函数使用上述约定返回损失张量和额外信息。
不同的函数可能有不同的额外字段。有关更多信息,请查看每个函数的文档。
蓝牙HI-FI车载DSP数字音频方案
首款模拟IC方案,解决产品三大阶段可靠性问题
贸泽备货Bosch BMM350磁力计 为3D、虚拟和增强现实、室内导航等领域提供支持
探讨一下分布式锁的设计与实现
华为开始进军6G,6G网速到底有多快?
DeepMind开源了一个内部强化学习库TRFL
选择性波峰焊介绍
诺基亚9什么时候上市?最新消息:诺基亚9工程机、跑分、配置曝光,超强旗舰来袭实力不容小觑!
智能家居不可或缺的 Smart Sensor 解决方案(一)
朝微电子:“以研带产”实现多项半导体器件国产化
长虹等家电厂商“拓路”智慧家庭 竞谋物联“蓝海”
华为Mate40 RS保时捷斩获2020年手机保值榜第一
AD20拼板步骤
虚拟现实火爆的背后 这7个问题需要解决
为何UWB定位方案能快速占领室内定位市场?
智能配电柜与普通配电柜
为什么单片机的I/O口需要驱动?
亚马逊携手 AWS 为企业提供混合云基础架构
5G基站节能及数字化管理解决方案-安科瑞任心怡
1MOREStylish耳机体验 更像是Android平台的AirPods