← Search

Pengjie Gu

11 accepted papers

2025

Cradle: Empowering Foundation Agents towards General Computer Control

ICML 2025poster

Despite their success in specific scenarios, existing foundation agents still struggle to generalize across various virtual scenarios, mainly due to the dramatically different encapsulations of environments with manually designed observation and action spaces. To handle this issue, we propose the Ge…

2025

Improving Reward Models with Proximal Policy Exploration for Preference-Based Reinforcement Learning

NeurIPS 2025poster

Reinforcement learning (RL) heavily depends on well-designed reward functions, which are often biased and difficult to design for complex behaviors. Preference-based RL (PbRL) addresses this by learning reward models from human feedback, but its practicality is constrained by a critical dilemma: wh…

Cited by 0SourceScholar
2025

MTRec: Learning to Align with User Preferences via Mental Reward Models

NeurIPS 2025poster

Recommendation models are predominantly trained using implicit user feedback, since explicit feedback is often costly to obtain. However, implicit feedback, such as clicks, does not always reflect users' real preferences. For example, a user might click on a news article because of its attractive he…

Cited by 0SourceScholar
2024

Improving Unsupervised Hierarchical Representation with Reinforcement Learning

CVPR 2024poster

Learning representations to capture the very fundamental understanding of the world is a key challenge in machine learning. The hierarchical structure of explanatory factors hidden in data is such a general representation and could be potentially achieved with a hierarchical VAE. However training a…

2024

PoRank: A Practical Framework for Learning to Rank Policies

IJCAI 2024poster

In many real-world scenarios, we need to select from a set of candidate policies before online deployment. Although existing Off-policy evaluation (OPE) methods can be used to estimate the online performance, they suffer from high variance. Fortunately, we care only about the ranking of the candidat…

2024

Resisting Stochastic Risks in Diffusion Planners with the Trajectory Aggregation Tree

ICML 2024spotlight

Diffusion planners have shown promise in handling long-horizon and sparse-reward tasks due to the non-autoregressive plan generation. However, their inherent stochastic risk of generating infeasible trajectories presents significant challenges to their reliability and stability. We introduce a novel…

2024

Solving Homogeneous and Heterogeneous Cooperative Tasks with Greedy Sequential Execution

ICLR 2024spotlight

Cooperative multi-agent reinforcement learning (MARL) is extensively used for solving complex cooperative tasks, and value decomposition methods are a prevalent approach for this domain. However, these methods have not been successful in addressing both homogeneous and heterogeneous tasks simultaneo…

Cited by 2SourcePDFScholar
2023

Controlling Type Confounding in Ad Hoc Teamwork with Instance-wise Teammate Feedback Rectification

ICML 2023poster

Ad hoc teamwork requires an agent to cooperate with unknown teammates without prior coordination. Many works propose to abstract teammate instances into high-level representation of types and then pre-train the best response for each type. However, most of them do not consider the distribution of te…

Cited by 2SourcePDFScholar
2023

Offline RL with Discrete Proxy Representations for Generalizability in POMDPs

NeurIPS 2023poster

Offline Reinforcement Learning (RL) has demonstrated promising results in various applications by learning policies from previously collected datasets, reducing the need for online exploration and interactions. However, real-world scenarios usually involve partial observability, which brings crucial…

Cited by 0SourcePDFScholar
2022

Learning Pseudometric-based Action Representations for Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning is a promising approach for practical applications since it does not require interactions with real-world environments. However, existing offline RL methods only work well in environments with continuous or small discrete action spaces. In environments with large and d…

Cited by 20SourcePDFScholar