3 个月前

循环独立机制

循环独立机制

摘要

学习能够反映环境动态特性的模块化结构,有助于提升模型的泛化能力,并增强对仅影响少数底层因果因素的环境变化的鲁棒性。为此,我们提出了一种新的循环架构——循环独立机制(Recurrent Independent Mechanisms, RIMs)。在该架构中,多个循环单元组以近似独立的转移动态运行,仅通过注意力机制的瓶颈进行有限的通信,并且仅在对其最具相关性的时刻才进行更新。我们证明,这种设计促使各RIM模块产生功能专化,从而显著提升模型在训练与评估阶段某些变化因素存在系统性差异的任务上的泛化性能。

代码仓库

fuyuan-li/tensorflow-RIMs
tf
GitHub 中提及
anirudh9119/RIMs
pytorch
GitHub 中提及

基准测试

基准方法指标
atari-games-on-atari-2600-beam-riderRIMs-PPO
Score: 5320
atari-games-on-atari-2600-up-and-downRIMs-PPO
Score: 390000
atari-games-on-atari-2600-zaxxonRIMs-PPO
Score: 15000

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
循环独立机制 | 论文 | HyperAI超神经