中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Data-efficient model-based reinforcement learning with trajectory discrimination

文献类型:期刊论文

作者Qu, Tuo3; Duan, Fuqing3; Zhang, Junge1; Zhao, Bo2; Huang, Wenzhen1
刊名COMPLEX & INTELLIGENT SYSTEMS
出版日期2023-10-11
页码10
ISSN号2199-4536
关键词Reinforcement learning Deep learning Continuous control task World model
DOI10.1007/s40747-023-01247-5
通讯作者Duan, Fuqing(fqduan@bnu.edu.cn)
英文摘要Deep reinforcement learning has always been used to solve high-dimensional complex sequential decision problems. However, one of the biggest challenges for reinforcement learning is sample efficiency, especially for high-dimensional complex problems. Model-based reinforcement learning can solve the problem with a learned world model, but the performance is limited by the imperfect world model, so it usually has worse approximate performance than model-free reinforcement learning. In this paper, we propose a novel model-based reinforcement learning algorithm called World Model with Trajectory Discrimination (WMTD). We learn the representation of temporal dynamics information by adding a trajectory discriminator to the world model, and then compute the weight of state value estimation based on the trajectory discriminator to optimize the policy. Specifically, we augment the trajectories to generate negative samples and train a trajectory discriminator that shares the feature extractor with the world model. Experimental results demonstrate that our method improves the sample efficiency and achieves state-of-the-art performance on DeepMind control tasks.
资助项目This work is supported by National Key Research and Development Project Grant, grant/award number: 2018AAA01008-02.[2018AAA01008-02] ; National Key Research and Development Project Grant
WOS研究方向Computer Science
语种英语
出版者SPRINGER HEIDELBERG
WOS记录号WOS:001081013100001
资助机构This work is supported by National Key Research and Development Project Grant, grant/award number: 2018AAA01008-02. ; National Key Research and Development Project Grant
源URL[http://ir.ia.ac.cn/handle/173211/53008]  
专题复杂系统认知与决策实验室
通讯作者Duan, Fuqing
作者单位1.Chinese Acad Sci, Inst Automat, 95 Zhongguancun East Rd, Beijing 100190, Peoples R China
2.Beijing Normal Univ, Sch Syst Sci, 19 Xinjiekou Outer St, Beijing 100875, Peoples R China
3.Beijing Normal Univ, Sch Artificial Intelligence, 19 Xinjiekou Outer St, Beijing 100875, Peoples R China
推荐引用方式
GB/T 7714
Qu, Tuo,Duan, Fuqing,Zhang, Junge,et al. Data-efficient model-based reinforcement learning with trajectory discrimination[J]. COMPLEX & INTELLIGENT SYSTEMS,2023:10.
APA Qu, Tuo,Duan, Fuqing,Zhang, Junge,Zhao, Bo,&Huang, Wenzhen.(2023).Data-efficient model-based reinforcement learning with trajectory discrimination.COMPLEX & INTELLIGENT SYSTEMS,10.
MLA Qu, Tuo,et al."Data-efficient model-based reinforcement learning with trajectory discrimination".COMPLEX & INTELLIGENT SYSTEMS (2023):10.

入库方式: OAI收割

来源:自动化研究所

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

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