Table of Contents
Fetching ...

Dueling Posterior Sampling for Preference-Based Reinforcement Learning

Ellen R. Novoseller, Yibing Wei, Yanan Sui, Yisong Yue, Joel W. Burdick

TL;DR

This work addresses learning under human trajectory-level preferences in reinforcement learning by introducing Dueling Posterior Sampling (DPS), a Bayesian, model-based algorithm that jointly infers environment dynamics and a utility function from trajectory preferences. DPS integrates a credit assignment mechanism to translate sparse trajectory-level feedback into state-action rewards, using Bayesian linear regression and Thompson sampling to balance exploration and exploitation. The authors establish an asymptotic Bayesian no-regret guarantee for DPS under a linear-credit model and demonstrate strong empirical performance across several benchmark domains, robust to the choice of credit assignment model. Overall, the paper extends posterior-sampling RL to the preference-based setting with theoretical guarantees and practical effectiveness, offering a principled framework for learning from human preferences in complex environments.

Abstract

In preference-based reinforcement learning (RL), an agent interacts with the environment while receiving preferences instead of absolute feedback. While there is increasing research activity in preference-based RL, the design of formal frameworks that admit tractable theoretical analysis remains an open challenge. Building upon ideas from preference-based bandit learning and posterior sampling in RL, we present DUELING POSTERIOR SAMPLING (DPS), which employs preference-based posterior sampling to learn both the system dynamics and the underlying utility function that governs the preference feedback. As preference feedback is provided on trajectories rather than individual state-action pairs, we develop a Bayesian approach for the credit assignment problem, translating preferences to a posterior distribution over state-action reward models. We prove an asymptotic Bayesian no-regret rate for DPS with a Bayesian linear regression credit assignment model. This is the first regret guarantee for preference-based RL to our knowledge. We also discuss possible avenues for extending the proof methodology to other credit assignment models. Finally, we evaluate the approach empirically, showing competitive performance against existing baselines.

Dueling Posterior Sampling for Preference-Based Reinforcement Learning

TL;DR

This work addresses learning under human trajectory-level preferences in reinforcement learning by introducing Dueling Posterior Sampling (DPS), a Bayesian, model-based algorithm that jointly infers environment dynamics and a utility function from trajectory preferences. DPS integrates a credit assignment mechanism to translate sparse trajectory-level feedback into state-action rewards, using Bayesian linear regression and Thompson sampling to balance exploration and exploitation. The authors establish an asymptotic Bayesian no-regret guarantee for DPS under a linear-credit model and demonstrate strong empirical performance across several benchmark domains, robust to the choice of credit assignment model. Overall, the paper extends posterior-sampling RL to the preference-based setting with theoretical guarantees and practical effectiveness, offering a principled framework for learning from human preferences in complex environments.

Abstract

In preference-based reinforcement learning (RL), an agent interacts with the environment while receiving preferences instead of absolute feedback. While there is increasing research activity in preference-based RL, the design of formal frameworks that admit tractable theoretical analysis remains an open challenge. Building upon ideas from preference-based bandit learning and posterior sampling in RL, we present DUELING POSTERIOR SAMPLING (DPS), which employs preference-based posterior sampling to learn both the system dynamics and the underlying utility function that governs the preference feedback. As preference feedback is provided on trajectories rather than individual state-action pairs, we develop a Bayesian approach for the credit assignment problem, translating preferences to a posterior distribution over state-action reward models. We prove an asymptotic Bayesian no-regret rate for DPS with a Bayesian linear regression credit assignment model. This is the first regret guarantee for preference-based RL to our knowledge. We also discuss possible avenues for extending the proof methodology to other credit assignment models. Finally, we evaluate the approach empirically, showing competitive performance against existing baselines.

Paper Structure

This paper contains 26 sections, 30 theorems, 153 equations, 7 figures, 4 tables, 3 algorithms.

Key Result

Proposition 1

The sampled dynamics converge in distribution to their true values as the DPS iteration increases.

Figures (7)

  • Figure 1: Empirical performance of DPS; each simulated environment is shown under the two least-noisy user preference models evaluated. The plots show DPS with three credit assignment models: Gaussian process regression (GPR), Bayesian linear regression, and a Gaussian process preference model. PSRL is an upper bound that receives numerical rewards, while EPMC is a baseline. Plots display the mean +/- one standard deviation over 100 runs of each algorithm tested. The remaining user noise models are plotted in Appendix \ref{['sec:additional_experiments']}. For RiverSwim and Random MDPs, normalization is with respect to the total reward achieved by the optimal policy. Overall, we see that DPS performs well and is robust to the choice of credit assignment model.
  • Figure 2: Empirical performance of DPS in the RiverSwim environment. Plots display mean +/- one standard deviation over 100 runs of each algorithm tested. Normalization is with respect to the total reward achieved by the optimal policy. Overall, we see that DPS performs well and is robust to the choice of credit assignment model.
  • Figure 3: Empirical performance of DPS in the Random MDP environment. Plots display mean +/- one standard deviation over 100 runs of each algorithm tested. Normalization is with respect to the total reward achieved by the optimal policy. Overall, we see that DPS performs well and is robust to the choice of credit assignment model.
  • Figure 4: Empirical performance of DPS in the Mountain Car environment. Plots display mean +/- one standard deviation over 100 runs of each algorithm tested. Overall, we see that DPS performs well and is robust to the choice of credit assignment model.
  • Figure 5: Empirical performance of DPS in the RiverSwim environment for different hyperparameter combinations. Plots display mean +/- one standard deviation over 30 runs of each algorithm tested with logistic user noise and $c = 0.001$. Overall, we see that DPS is robust to the choice of hyperparameters. The hyperparameter values depicted in each plot are (from left to right): for Bayesian linear regression, $(\sigma, \lambda) = \{(0.5, 0.1), (0.5, 10), (0.1, 0.1), (0.1, 10), (1, 0.1)\}$; for GP regression, $(\sigma_f^2, \sigma_n^2) = \{(0.1, 0.001), (0.1, 0.1), (0.01, 0.001), (0.001, 0.0001), (0.5, 0.1)\}$; for Bayesian logistic regression (special case of the GP preference model), $(\lambda, \alpha) = \{(1, 1), (30, 1), (20, 0.5), (1, 0.5), (30, 0.1)\}$; and additionally for the GP preference model, $c \in \{0.5, 1, 2, 5, 13\}$. See Table \ref{['table:hyperparams_RiverSwim']} for the values of any hyperparameters not specifically mentioned here.
  • ...and 2 more figures

Theorems & Definitions (55)

  • Proposition 1
  • Proposition 2
  • Theorem 1
  • Lemma 12
  • Theorem 2
  • Lemma 17
  • Theorem 3
  • Definition 1: Value function given transition dynamics, rewards, and a policy
  • Definition 2: Optimal deterministic policy given transition dynamics and rewards
  • Definition 3: Eigenvalue notation
  • ...and 45 more