Table of Contents
Fetching ...

Preference Elicitation for Offline Reinforcement Learning

Alizée Pace, Bernhard Schölkopf, Gunnar Rätsch, Giorgia Ramponi

TL;DR

This work tackles offline reinforcement learning when reward signals are unavailable by introducing offline preference elicitation via simulated rollouts (Sim-OPRL). It integrates model-based dynamics learning with pessimistic robust objectives for out-of-distribution states and an optimistic, information-seeking strategy for preference queries, yielding theoretical guarantees on sample complexity tied to offline data coverage of the optimal policy. The authors formalize the problem, provide bounds for both sampling-from-offline data (OPRL) and simulated-rollout strategies, and present a practical implementation with ensembles and uncertainty penalties. Empirically, Sim-OPRL surpasses offline preference baselines across diverse benchmarks, achieving better performance with fewer human queries. The approach has meaningful practical impact for safety-critical domains like healthcare, where environment interaction is restricted but expert input can be simulated or queried in a controlled setting.

Abstract

Applying reinforcement learning (RL) to real-world problems is often made challenging by the inability to interact with the environment and the difficulty of designing reward functions. Offline RL addresses the first challenge by considering access to an offline dataset of environment interactions labeled by the reward function. In contrast, Preference-based RL does not assume access to the reward function and learns it from preferences, but typically requires an online interaction with the environment. We bridge the gap between these frameworks by exploring efficient methods for acquiring preference feedback in a fully offline setup. We propose Sim-OPRL, an offline preference-based reinforcement learning algorithm, which leverages a learned environment model to elicit preference feedback on simulated rollouts. Drawing on insights from both the offline RL and the preference-based RL literature, our algorithm employs a pessimistic approach for out-of-distribution data, and an optimistic approach for acquiring informative preferences about the optimal policy. We provide theoretical guarantees regarding the sample complexity of our approach, dependent on how well the offline data covers the optimal policy. Finally, we demonstrate the empirical performance of Sim-OPRL in various environments.

Preference Elicitation for Offline Reinforcement Learning

TL;DR

This work tackles offline reinforcement learning when reward signals are unavailable by introducing offline preference elicitation via simulated rollouts (Sim-OPRL). It integrates model-based dynamics learning with pessimistic robust objectives for out-of-distribution states and an optimistic, information-seeking strategy for preference queries, yielding theoretical guarantees on sample complexity tied to offline data coverage of the optimal policy. The authors formalize the problem, provide bounds for both sampling-from-offline data (OPRL) and simulated-rollout strategies, and present a practical implementation with ensembles and uncertainty penalties. Empirically, Sim-OPRL surpasses offline preference baselines across diverse benchmarks, achieving better performance with fewer human queries. The approach has meaningful practical impact for safety-critical domains like healthcare, where environment interaction is restricted but expert input can be simulated or queried in a controlled setting.

Abstract

Applying reinforcement learning (RL) to real-world problems is often made challenging by the inability to interact with the environment and the difficulty of designing reward functions. Offline RL addresses the first challenge by considering access to an offline dataset of environment interactions labeled by the reward function. In contrast, Preference-based RL does not assume access to the reward function and learns it from preferences, but typically requires an online interaction with the environment. We bridge the gap between these frameworks by exploring efficient methods for acquiring preference feedback in a fully offline setup. We propose Sim-OPRL, an offline preference-based reinforcement learning algorithm, which leverages a learned environment model to elicit preference feedback on simulated rollouts. Drawing on insights from both the offline RL and the preference-based RL literature, our algorithm employs a pessimistic approach for out-of-distribution data, and an optimistic approach for acquiring informative preferences about the optimal policy. We provide theoretical guarantees regarding the sample complexity of our approach, dependent on how well the offline data covers the optimal policy. Finally, we demonstrate the empirical performance of Sim-OPRL in various environments.
Paper Structure (44 sections, 13 theorems, 59 equations, 8 figures, 5 tables, 3 algorithms)

This paper contains 44 sections, 13 theorems, 59 equations, 8 figures, 5 tables, 3 algorithms.

Key Result

Theorem 5.1

For any $\delta \in (0,1]$, let $\beta_T= c^{\textrm{\tiny MLE}}_T \log(H \mathcal{N}_{\mathcal{F}_T}(1/N_o)/\delta)/N_o$ and $\beta_R ={c^{\textrm{\tiny MLE}}_R\log(\mathcal{N}_{\mathcal{F}_R}(1/N_p)/\delta)}/{N_p}$, where $N_o= H|\mathcal{D}_{\textrm{offline}}|$ is the number of observed transiti where $\alpha=1$ for uniform sampling or $\alpha\leq1$ for uncertainty sampling, $C_R$ is a concent

Figures (8)

  • Figure 1: Environment returns under different preference elicitation strategies. Mean and 95% confidence interval over 6 experiments. Environment returns are normalized between 0 and 100. Only OPRL and Sim-OPRL are fully offline.
  • Figure 2: Algorithm ablations (StarMDP).
  • Figure 3: Preference sample complexity $N_p$ as function of the properties of the observational data, to reach a suboptimality gap of $\epsilon=20$ over normalized environment returns (Star MDP). Mean and 95% confidence intervals over 6 experiments. $\times$ marks when the target suboptimality could not be achieved.
  • Figure 4: Tabular MDP. The environment starts in state $s_0$ and has horizon $H=1$. Transition probabilities from state $s_0$ are given for the two binary actions $a_0, a_1$ (which send the agent to the other state with complementary probability).
  • Figure 5: Star MDP. Transition probabilities are 0.9 for all solid arrows. Omitted actions or complementary transitions keep the state unchanged.
  • ...and 3 more figures

Theorems & Definitions (27)

  • Definition 3.1: Optimality Criterion of Offline Preference Elicitation
  • Definition 3.2: $\epsilon$-bracketing number
  • Definition 3.3: Transition concentrability coefficient, zhan2023provable
  • Theorem 5.1
  • Theorem 6.1
  • Lemma A.1: MLE Guarantee, Lemma 1 in zhan2023provable
  • proof
  • Lemma A.2: TV-distance to MLE
  • proof
  • Lemma A.3: Telescoping Lemma
  • ...and 17 more