中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Prototypical context-aware dynamics generalization for high-dimensional model-based reinforcement learning

文献类型:会议论文

作者Junjie, Wang1,4; Yao, Mu3; Dong, Li2; Qichao,Zhang1,4; Dongbin, Zhao1,4; Yuzheng, Zhuang2; Ping, Luo3; Bin, Wang2; Jianye, Hao2
出版日期2023-05
会议日期2023-5-5
会议地点Kigali City, Rwanda, Africa
英文摘要

The ability to generalize different dynamics is crucial for decision-making in autonomous driving that relies on high-dimensional inputs. The latent world model provides a promising way to learn policies in a compact latent space for tasks with high-dimensional observations, however, its generalization across diverse environments with unseen dynamics remains challenging. Although the recurrent structure utilized in current advances helps to capture local dynamics, modeling only state transitions without an explicit understanding of environmental context limits the generalization ability of the dynamics model. To address this issue, we propose a Prototypical Context-Aware Dynamics (ProtoCAD) model, which captures the local dynamics by temporally consistent latent context and enables dynamics generalization in high-dimensional control tasks. ProtoCAD extracts useful contextual information with the prototypes clustered over the batch, and it benefits model-based reinforcement learning in two ways: 1) A temporally consistent prototypes regularizer is utilized, which encourages the prototype assignments produced for different temporal parts of the same latent trajectory to be temporally consistent instead of comparing the features; 2) A context representation is designed, which combines both projection embedding of latent states and aggregated prototypes and can significantly improve the dynamics generalization ability. Extensive experiments show that ProtoCAD surpasses existing methods in terms of dynamics generalization.

源URL[http://ir.ia.ac.cn/handle/173211/52274]  
专题复杂系统管理与控制国家重点实验室_深度强化学习
作者单位1.State Key Laboratory of Multimodal Artificial Intelligence Systems, Institute of Automation, Chinese Academy of Sciences
2.Huawei Noah’s Ark Lab
3.The University of Hong Kong
4.University of Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Junjie, Wang,Yao, Mu,Dong, Li,et al. Prototypical context-aware dynamics generalization for high-dimensional model-based reinforcement learning[C]. 见:. Kigali City, Rwanda, Africa. 2023-5-5.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。