Online Compatible Reward Identification from Preference Feedback
Abstract
In reinforcement learning, human preference feedback is emerging as a viable alternative to expert-designed reward functions, which can be difficult to engineer in real-world problems. However, despite the growing importance of preference feedback, how to effectively elicit preferences remains a fundamental open problem. This work focuses on the compatible reward identification task. The aim is to derive, starting from preference feedback, a reward function compatible with the observed preferences and accurate across the entire state-action space, ensuring higher transferability, safety, and interpretability. Indeed, the most common reinforcement learning from human feedback objective is to learn the optimal policy, requiring accuracy only in the portion of the state-action space that the agent visits. However, this goal cannot provide the same guarantees as compatible reward identification. First, we discuss commonalities and differences between the two goals. Then, we consider deterministic preferences, deriving the minimum number of interactions needed to identify the set of compatible rewards, and showing that using fewer queries may lead to arbitrarily large suboptimality. Finally, we focus on stochastic preferences generated via the Bradley-Terry (BT) model. We introduce the concepts of query basis and its index, relating them to the problem complexity. Upon this, we discuss the connection between the index of a basis and the BT model, as well as the limitations that the model induces in this setting. Additionally, we devise an algorithm to identify a nearly-optimal query basis with polynomial human query complexity.
Lay Summary
Teaching AI using human preference feedback is emerging as a powerful alternative to hand-crafting complex reward formulas. However, standard methods only optimize the AI's behavior for the specific paths it explores, failing to capture a robust understanding of human intent across the entire task space. We tackle the task of "compatible reward identification" by introducing a new algorithm that strategically queries human labelers to map out their exact preference rules using the minimum number of questions. Our theoretical analysis shows, somewhat counter-intuitively, that the most efficient way to learn human intent is by asking people to choose between highly similar options rather than wildly different ones. This framework guarantees that the recovered human intent is accurate everywhere, ensuring that trained AI agents are transferable to new environments and fundamentally safer to deploy.