4 papers
Task Aware Dreamer for Task Generalization in Reinforcement Learning
Chengyang Ying, Xinning Zhou, Zhongkai Hao +4
A long-standing goal of reinforcement learning is to acquire agents that can learn on training tasks and generalize well on unseen tasks that may share a similar dynamic but with d…
Exploratory Diffusion Model for Unsupervised Reinforcement Learning
Chengyang Ying, Huayu Chen, Xinning Zhou +3
Unsupervised reinforcement learning (URL) aims to pre-train agents by exploring diverse states or skills in reward-free environments, facilitating efficient adaptation to downstrea…
Self-Consistent Model-based Adaptation for Visual Reinforcement Learning
Xinning Zhou, Chengyang Ying, Yao Feng +2
Visual reinforcement learning agents typically face serious performance declines in real-world applications caused by visual distractions. Existing methods rely on fine-tuning the…
PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning
Chengyang Ying, Zhongkai Hao, Xinning Zhou +4
Designing generalizable agents capable of adapting to diverse embodiments has achieved significant attention in Reinforcement Learning (RL), which is critical for deploying RL agen…