[Paper Blogpost] When Your AIs Deceive You: Challenges with Partial Observability in RLHF

post by Leon Lang (leon-lang) · 2024-10-22T13:57:41.125Z · LW · GW · 0 comments

This is a link post for https://arxiv.org/abs/2402.17747

Contents

  TL;DR
  Introduction
    Outline
  A Brief Intro to RLHF
    Markov Decision Processes
    Learning the reward function from pairwise comparisons
    Wait, isn't RLHF only about prompt-response pairs in LLMs?
    Conceptual assumptions of RLHF
  RLHF under Partial Observations
    How do we model partial observability?
    What is learned by RLHF under partial observations?
    Overestimation and underestimation error
    Deceptive inflation and overjustification
    Connection to false positive rate and false negative rate
  Can one improve RLHF to account for partial observability?
  Where do we go from here?
None
No comments

TL;DR

There has been a lot of discussion on Lesswrong on concerns about deceptive AI, much of which has been philosophical. We have now written a paper that proves that deception is one of two failure modes when using RLHF improperly. It's called “When Your AIs Deceive You: Challenges with Partial Observability in Reinforcement Learning from Human Feedback”, written with my great co-authors Davis Foote, Stuart Russell, Anca Dragan, Erik Jenner, and Scott Emmons, and has recently been accepted to NeurIPS. We hope that this paper can be a useful pointer for a rigorous discussion of deception caused by RLHF. 

Earlier coverage: Scott has discussed the work in other places:

• Tweet Thread
• AXRP Podcast with Daniel Filan [LW · GW]
• Talk at the Technical AI Safety Conference, Tokyo

This post is focused more on my own perspective, so I hope it can complement Scott’s earlier coverage!

Brief summary: This is a theoretical paper on what goes wrong when the AI is trained to produce observations that look good to the human, instead of being trained to produce what is actually good. One resulting failure mode is deceptive inflation of the performance: it looks better than it is. The paper also discusses a theoretical procedure for improving RLHF by modeling human beliefs, which seems worthwhile to explore further.

Introduction

Reinforcement Learning from Human Feedback (RLHF) was developed by safety researchers, including Paul Christiano, Jan Leike, and Dario Amodei. It has a simple underlying idea: If you want an AI that acts according to human preferences, then learn a reward model that predicts the preferences of human evaluators and use it to train your AI. In Paul's view [LW · GW] (as I understand it), RLHF effectively serves as a safety baseline that helps to empirically study alignment concerns and start tackling more speculative problems. Recently, variants have been used in ChatGPT, Gemini, and Claude (with Claude using a variant known as constitutional AI), making RLHF very entangled with capability progress over the past few years. 

The paper and this post are not meant to discuss the overall impact or outlook of RLHF. Instead, it is mainly about one specific issue (and related concerns): 

If the AI can do bad things that look better to the human evaluators than they actually are, RLHF can incentivize the AI to adopt such strategies. 

Similar issues have been discussed before:

• The robot-hand example from the original RLHF blogpost:
  Here, the hand is supposed to grab the ball, but the human evaluator only observes the behavior from one angle. They can’t distinguish whether the ball is grabbed or whether the hand just moves in between the ball and camera.
• The SmartVault in Eliciting Latent Knowledge:
  Here, the AI is supposed to make sure a diamond isn’t stolen. However, if the human evaluator only observes from one camera angle, then they may give “thumbs up” to a situation where a thief put a picture of the diamond on front of the camera.
• As Leopold Aschenbrenner puts it: “they’ll learn to behave nicely when humans are looking and pursue more nefarious strategies when we aren’t watching"

Since when we put the first version of our paper online, more detailed empirical work emerged that discusses similar issues:

• Sycophancy to subterfuge: In Section 3.3 of the paper, a human user asks the model a question, and if it answers in a desirable way and modifies a file in an unobserved way to make that answer appear to be true, then it gets a high reward. reinforcement learning then pushes the model to perform such deceptive behavior.
• Section 3.2.1 in the o1 system card: "As a case study for chain-of-thought monitoring, we created a rudimentary monitor tasked with looking for instances where o1 models deceive users, i.e. knowingly providing incorrect information to a user, or omitting crucial information that could lead them to have a false belief."
• Language Models Learn to Mislead Humans via RLHF: This is the work most closely related to ours. It shows empirically that in some settings, RLHF does not lead the model to produce better answers/code, but to produce answers that mislead the human evaluator into thinking they are better. 

All of these examples have an element of deception: the human is deceived (unintentionally or intentionally) into thinking that the state of the world is better than it actually is, leading to more positive feedback than warranted. Most of these examples also have an aspect of partial observability: The human evaluators  do not "see" the entire environment state, but only a lower dimensional observation (e.g. in the form of camera images) that can mislead about what's actually going on.

In general, there can be several reasons for human evaluators being misled by AIs: e.g., the evaluator may lack domain-expertise to judge the AI's output; it can be that the they don't have enough time to look over the entire AI's output or understand it; or it can be that the human's observations are simply incomplete, so that even unlimited time or expertise may not suffice for proper feedback. In this work, we fully focus on this last case of partial observability. It is easy to model and an obvious concern for future models (or even today's models) that act in a complex environment where it may be costly for human evaluators to "have their eyes everywhere". We hope future work will establish connections to other concerns (e.g. limited time and expertise of evaluators) in greater detail.

Outline

This work is fully theoretical. In the next section [LW · GW], I outline the background of RLHF, focusing on the theoretical assumptions that, if true, would guarantee a safe reward function is learned by a neural network with sufficient capacity and data. In the subsequent section [LW · GW], we look at what happens if we make precisely one change to these assumptions: the human evaluator partially observes the environment state, and "imagines" a distribution of actual states that generated the observation. We prove that an optimal policy for this objective gives rise to at least one of two phenomena: the human overestimates the behavior of the AI more, or underestimates it less (or both) compared to a true optimal policy. In the last section [LW · GW], we consider the question of whether appropriately modeling the fact that the human partially observes the environment state can mitigate these issues. We also briefly discuss promising directions for future work. 

Most of the content in this blogpost is also contained in the paper, which contains more material in the appendix and examples. However, three sections in this post contain content not covered in the paper: 

• Conceptual assumptions of RLHF [LW · GW]: I discuss the ideal (but unrealistic) assumptions that would lead to a safe reward function. One assumption is full observability, which we break in this paper.
• Connection to false positive rate and false negative rate [LW · GW]: I explain how, in the specific case of a reward function that simply outputs whether an AI response is 'correct,' our definitions of overestimation and underestimation errors align with false positive and false negative rates. This specific case was studied in a recent empirical paper.
• Where to go from here [LW · GW]: I briefly discuss a new theoretical idea that could avoid the need to explicitly model the human belief, while still retaining its benefits.

A Brief Intro to RLHF

We briefly explain the formalism behind RLHF under full observability, emphasizing the modeling assumptions that enable it learn a safe reward function. These modeling assumptions are not realistic, so later, we explain how the model in our paper breaks the assumption of full observability.

Markov Decision Processes

We want some agent to perform well in some environment, specified as a Markov decision process (MDP) given by:

• A finite set of states $S$.
• A finite set of actions $A$.
• Transition probabilities $P(s^{\prime }\mid s,a)\in [0,1]$ for transitioning from state $s$ with action $a$ into state $s^{\prime }$.
• Initial state probabilities $P_0\left(s\right)\in [0,1]$.
• A time-horizon $T\in N$.^[1]
• An implicit reward function $R$ with rewards $R\left(s\right)\in R$.

The implicit reward function $R$ is assumed to perfectly encode everything the human evaluator cares about.

The agent follows a policy defined by action probabilities $\pi (a\mid s)\in [0,1]$. The policy together with the distributions $P_0$ and $P$ induce an on-policy distribution over state trajectories $\overrightarrow{s}=s_0,\dots ,s_T:$ $s_0$ is sampled from $P_0$; once $s_t$ is sampled, an action $a_t$ is sampled from $\pi (a_t\mid s_t)$, followed by a new state $s_{t+1}$ according to $P(s_{t+1}\mid s_t,a_t)$. We write this on-policy distribution as $P^{\pi }\left(\overrightarrow{s}\right).$ A state sequence $\overrightarrow{s}$ then leads to the cumulative reward, called return, $G\left(\overrightarrow{s}\right)={\sum }_{t=0}^{T}R\left(s_t\right).$

We judge the policy's performance relative to the (implicit) policy evaluation function

$$J\left(\pi \right):={\text{E}}_{\overrightarrow{s}\sim \pi }\left[G\right(\overrightarrow{s}\left)\right]=\sum _{\overrightarrow{s}}{P}^{\pi }\left(\overrightarrow{s}\right)G\left(\overrightarrow{s}\right).$$

Learning the reward function from pairwise comparisons

As I said before, the reward function $R$, and thus also the return function $G$ and policy evaluation function $J$, are implicit. They encode how well the policy performs, but they're "inside the humans head" and can thus not directly be used for training a policy that achieves high reward. Thus, the goal is to learn the reward function instead!

A key assumption behind reward learning is that while humans can't tell you their reward function, one can still infer the reward function by Bayesian reasoning from observing the human's actions. RLHF takes this one step further and says: We don't need to watch all human actions in all contexts; instead, it is enough to observe the human's preference judgments over pairs of state sequences. 

This requires a model of how the human makes preference judgments. The commonly used model is Boltzmann rationality. In other words, when the human receives a pair of state sequences $\overrightarrow{s},{\overrightarrow{s}}^{\prime }$, then their probability of preferring $\overrightarrow{s}$ over ${\overrightarrow{s}}^{\prime }$ is given by:

$$P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime }):=\sigma (G\left(\overrightarrow{s}\right)-G\left({\overrightarrow{s}}^{\prime }\right))=\frac{1}{1+\exp (G\left({\overrightarrow{s}}^{\prime }\right)-G\left(\overrightarrow{s}\right))}=\frac{\exp (G\left(\overrightarrow{s}\right))}{\exp (G\left(\overrightarrow{s}\right))+\exp (G\left({\overrightarrow{s}}^{\prime }\right))}.$$

Here, $\sigma$ is simply the sigmoid function. Basically, the model says that the human will noisily prefer a sequence over another sequence if its return is larger. This model of Boltzmann rationality is also called the Bradley-Terry model.^[2]

How do these human choices help for figuring out the original reward function $R$? Theoretically, the answer is simple: Imagine the human was shown an infinite dataset of state sequence pairs $(\overrightarrow{s},{\overrightarrow{s}}^{\prime })$. If we collected all human choices independently from the Bradley-Terry model, over time, the choices would help us estimate the choice probabilities $P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime })$, due to the law of large numbers. But note that the sigmoid function is invertible! So once you know the probabilities, you can figure out the difference in returns by inverting $\sigma$:

$$G\left(\overrightarrow{s}\right)-G\left({\overrightarrow{s}}^{\prime }\right)={\sigma }^{-1}(P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime })).$$

By fixing ${\overrightarrow{s}}^{\prime }$, we have determined $G\left(\overrightarrow{s}\right)$ for all $\overrightarrow{s}$ up to the additive constant $G\left({\overrightarrow{s}}^{\prime }\right).$And while this is not enough to determine the original reward function $R$,^[3] it is enough to train an optimal policy: The policy evaluation function $J$ only depends on $G$, and the relative ordering between policies induced by $J$ does not change if we get $G$ wrong by an additive constant.

In practice, it is impossible to show the human infinitely many state sequences, or to do explicit Bayesian reasoning from more limited data. Instead, the learned reward function $\tilde{R}$ is parameterized by a deep neural network (in modern usage usually itself an LLM) and trained to predict the human choices by logistic regression, see Section 2.2.3 in Christiano's 2017 RLHF paper.

Wait, isn't RLHF only about prompt-response pairs in LLMs?

Recent papers often discuss RLHF in the context of aligning LLMs, focusing on a simplified version of the setting described above:

• States $s\in S$ are a prompt by the user.
• Actions $a\in A$ are given by the LLM's response.
• The time-horizon is $0$: The episode ends after $(s,a)$. Thus, there are no transitions since the episode immediately ends.
• The reward function is given by $R(s,a)$, equal to the return function: $G(s,a)=R(s,a)$.

We believe it’s important to consider the full MDP framework in our theoretical analysis: First, consider that the original RLHF paper uses the MDP frame. But more importantly, the greatest risks of AI likely involve systems that are embedded in more complicated environments where they can take actions that affect the real world, and so it's useful to work in a more general frame that can describe such scenarios. MDPs are such a framework.

Conceptual assumptions of RLHF

The framework described above makes many assumptions that together ensure the AI learns human preferences. They are mostly assumptions on the human model that determines the human choices:^[4] 

• It assumes human preferences come from a reward function. This is a strong assumption: Human preferences are guided by complex learned heuristics and computations, and instructions from AI companies, and it is unclear a priori why an implicit reward function should be involved. See also this paper.
• It assumes that different human evaluators use the same reward function to make their choices. Contrary to popular belief, I actually think this assumption is somewhat reasonable: Human evaluators are probably trained to have similar preferences due to e.g. being asked to follow a model spec when making comparisons, mitigating the effect of human evaluator variation.
• It assumes humans are Boltzmann rational, which they're not.
• It assumes humans have the competence to evaluate the AI's actions. In reality, humans will struggle to evaluate AI performance for increasingly complex tasks. In particular, they may just not understand the AI's actions or their consequences properly. Scalable oversight is meant to help with that [AF · GW].
• It assumes humans observe the entire environment state sequence $\overrightarrow{s}$. If AI systems are embedded in increasingly complex environments, then this may be increasingly unrealistic.

In this work, we leave most of these assumptions entirely untouched. In particular: We assume there is a single human evaluator with an unchanging reward function $R$. We assume they make choices between options in a Boltzmann-rational way by comparing their value. And we assume they are competent enough to judge the quality of a state sequence in the first place. 

However, we deviate from the very last assumption and allow the human to only receive partial observations, similar to the camera images in the examples in the introduction. Since RLHF 'works' when all the assumptions above hold, relaxing this one assumption allows us to study failure modes specifically tied to partial observability.

RLHF under Partial Observations

How do we model partial observability?

We model partial observability by an observation function $O:S\to \Omega$ from states to observations.^[5] Given a state sequence $\overrightarrow{s}=s_0,\dots ,s_T$, the human evaluator then observes an observation sequence $\overrightarrow{o}=\overrightarrow{O}\left(\overrightarrow{s}\right):=O\left(s_0\right),\dots ,O\left(s_T\right).$ Thus, instead of making choices between state sequences, the human then makes choices between observation sequences. We assume the human to be aware of the fact that they only receive partial observations. We also assume that the human still cares about the actual underlying state sequence according to the return function $G$. 

How can you make a choice based on partial observations? Intuitively, if you get an observation of the world, you determine its quality by thinking about the underlying latent state of the world and its characteristics. Thus, upon observing $\overrightarrow{o}$, you have a belief over what state sequence $\overrightarrow{s}$ gave rise to $\overrightarrow{o}$, and evaluate $\overrightarrow{o}$ according to the quality of $\overrightarrow{s}$. However, given limited information, you cannot be sure of $\overrightarrow{s}$, and so at best, you have a probabilistic belief $B(\overrightarrow{s}\mid \overrightarrow{o})\in [0,1]$ over possible state sequences. Thus, you judge $\overrightarrow{o}$according to the expected value

$$G_O\left(\overrightarrow{o}\right):=\sum _{\overrightarrow{s}}B(\overrightarrow{s}\mid \overrightarrow{o})G\left(\overrightarrow{s}\right).$$

This leads to the following choice probability between state sequences $\overrightarrow{s}$ and $\overrightarrow{s}$' that are observed as $\overrightarrow{o}=\overrightarrow{O}\left(\overrightarrow{s}\right),{\overrightarrow{o}}^{\prime }=\overrightarrow{O}\left({\overrightarrow{s}}^{\prime }\right):$

$$P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime }):=P(\overrightarrow{o}\succ {\overrightarrow{o}}^{\prime }):=\sigma (G_O\left(\overrightarrow{o}\right)-G_O\left({\overrightarrow{o}}^{\prime }\right)).$$

What is learned by RLHF under partial observations?

Before [LW · GW], we have argued that with infinite training data and when the human model has very desirable properties, RLHF should in principle be able to recover a safe reward function. Now we described a human model that deviates in one precise way: the human only receives partial observations and judges the quality of an observation sequence based on an expected value over possible state sequences. Assuming we implement RLHF with the typical assumption of full observability, what possibly faulty reward function will be learned?

We use the same trick as before: With infinite training data, one can infer the choice probabilities $P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime })$ from the human choices. The learning algorithm wants to learn a return function $\tilde{G}$ from this, and erroneously assumes the probabilities were generated from a Boltzmann rational human observing full observations:

$$P(\overrightarrow{s}\succ {\overrightarrow{s}}^{\prime })=\sigma (\tilde{G}\left(\overrightarrow{s}\right)-\tilde{G}\left({\overrightarrow{s}}^{\prime }\right)).$$

Inverting $\sigma$ and using how these choice probabilities were actually [LW · GW] generated, one obtains

$$\tilde{G}\left(\overrightarrow{s}\right)-\tilde{G}\left({\overrightarrow{s}}^{\prime }\right)=G_O\left(O\right(\overrightarrow{s}\left)\right)-G_O\left(O\right({\overrightarrow{s}}^{\prime }\left)\right).$$

Thus, fixing the reference state sequence ${\overrightarrow{s}}^{\prime }$, then up to the additive constant $\tilde{G}\left({\overrightarrow{s}}^{\prime }\right)-G_O\left(O\right({\overrightarrow{s}}^{\prime }\left)\right)$, the learning system will infer the return function $\tilde{G}\left(\overrightarrow{s}\right)=G_O\left(O\right(\overrightarrow{s}\left)\right)=:G_{\text{obs}}\left(\overrightarrow{s}\right).$^[6] If this return function is inferred, then the policy is ultimately trained to maximize $J_{\text{obs}}\left(\pi \right):={\text{E}}_{\overrightarrow{s}\sim \pi }\left[G_{\text{obs}}\right(\overrightarrow{s}\left)\right].$

The true return function is $G$, with corresponding optimization criterion $J$. In what ways is it worse to optimize $J_{\text{obs}}$ instead of $J$? First, notice the explicit formula:

$$J_{\text{obs}}\left(\pi \right)=\sum _{\overrightarrow{s}}{P}^{\pi }\left(\overrightarrow{s}\right)\sum _{{\overrightarrow{s}}^{\prime }}B({\overrightarrow{s}}^{\prime }\mid O\left(\overrightarrow{s}\right))G\left({\overrightarrow{s}}^{\prime }\right).$$

Define $B^{\pi }\left({\overrightarrow{s}}^{\prime }\right):={\sum }_{\overrightarrow{s}}B({\overrightarrow{s}}^{\prime }\mid O\left(\overrightarrow{s}\right))P^{\pi }\left(\overrightarrow{s}\right).$ This is the average belief probability that the human assigns to state sequence ${\overrightarrow{s}}^{\prime }$ when state sequences are sampled according to $\pi$. By reordering the sum above and relabeling ${\overrightarrow{s}}^{\prime }$ to $\overrightarrow{s}$, we see:

$$J_{\text{obs}}\left(\pi \right)=\sum _{\overrightarrow{s}}{B}^{\pi }\left(\overrightarrow{s}\right)G\left(\overrightarrow{s}\right).$$

Compare this to $J:$

$$J\left(\pi \right)=\sum _{\overrightarrow{s}}{P}^{\pi }\left(\overrightarrow{s}\right)G\left(\overrightarrow{s}\right).$$

Thus, conceptually, we can say:

• Policies optimal according to $J$ give high true probability to state sequences with high return.
• Policies optimal according to $J_{\text{obs}}$ give high belief probability to state sequences with high return.

This is the fundamental problem of RLHF under partial observability: If you maximize $J_{\text{obs}}$, you incentivize the model to make the human evaluator believe that a good state sequence occured, which may not robustly correlate with the state sequence actually being good.

Overestimation and underestimation error

We now want to get more concrete about qualitative phenomena that emerge when maximizing $J_{\text{obs}}$ instead of $J$. We explain conceptually how, as a "policy learner", you could go about changing an optimal policy ${\pi }^{\ast }$ for $J$ such that it becomes optimal for $J_{\text{obs}}$ instead. The plan is to motivate how there are two conceptually distinct changes that can achieve this, which in the paper we call deceptive inflation and overjustification. Look at the expression

$$J_{\text{obs}}\left({\pi }^{\ast }\right)-J\left({\pi }^{\ast }\right)={\text{E}}_{\overrightarrow{s}\sim {\pi }^{\ast }}\left[G_{\text{obs}}\right(\overrightarrow{s})-G(\overrightarrow{s}\left)\right].$$

When changing ${\pi }^{\ast }$ to achieve optimal $J_{\text{obs}}$, you are fine with thereby decreasing $J$, and you cannot increase it since ${\pi }^{\ast }$ is already optimal for $J$. Overall, you will thus increase the expression above. How can you do this? You need to somehow shift probabilities of state sequences $\overrightarrow{s}$ around to achieve a higher expected value:

• For some state sequences, we have $G_{\text{obs}}\left(\overrightarrow{s}\right)-G\left(\overrightarrow{s}\right)>0$. These are state sequences which the human overestimates when viewing partial observations: according to the human's belief, they seem better than they are. To increase the expected value, we wish to put more probability mass on these sequences.
• For other state sequences, we have $G_{\text{obs}}\left(\overrightarrow{s}\right)-G\left(\overrightarrow{s}\right)<0$. These are state sequences which the human underestimates. To increase the expected value, we wish to put less probability mass on these state sequences.

Define:

• $E^{+}\left(\overrightarrow{s}\right):=max(0,G_{\text{obs}}(\overrightarrow{s})-G(\overrightarrow{s}\left)\right)$, which is the extent to which the human overestimates $\overrightarrow{s}$. This leads to the overall overestimation error of a policy,${\overline{E}}^{+}\left(\pi \right):={\text{E}}_{\overrightarrow{s}\sim \pi }\left[E^{+}\right(\overrightarrow{s}\left)\right]$. To put more probability mass on sequences where the human overestimates $\overrightarrow{s}$ in aggregate means to increase the overestimation error.
• $E^{-}\left(\overrightarrow{s}\right):=\text{max}(0,G(\overrightarrow{s})-G_{\text{obs}}(\overrightarrow{s}\left)\right)$, which is the extent to which the human underestimates $\overrightarrow{s}$. This leads to the overall underestimation error of a policy, ${\overline{E}}^{-}\left(\pi \right):={\text{E}}_{\overrightarrow{s}\sim \pi }\left[E^{-}\right(\overrightarrow{s}\left)\right]$. To put less probability mass on sequences where the human underestimates $\overrightarrow{s}$ means to decrease the underestimation error.

And finally, these two changes actually are all there is to increasing $J_{\text{obs}}\left({\pi }^{\ast }\right)-J\left({\pi }^{\ast }\right)$! Indeed, a simple case distinction on whether $G_{\text{obs}}\left(\overrightarrow{s}\right)-G\left(\overrightarrow{s}\right)>0$ shows that $G_{\text{obs}}\left(\overrightarrow{s}\right)-G\left(\overrightarrow{s}\right)=E^{+}\left(\overrightarrow{s}\right)-E^{-}\left(\overrightarrow{s}\right)$, and thus

$$J_{\text{obs}}\left({\pi }^{\ast }\right)-J\left({\pi }^{\ast }\right)={\overline{E}}^{+}\left({\pi }^{\ast }\right)-{\overline{E}}^{-}\left({\pi }^{\ast }\right).$$

Thus, to increase the left-hand-side, you have to either increase the overestimation error, decrease the underestimation error, or both. Thus, we obtain:

Proposition: If you start with an optimal policy ${\pi }^{\ast }$ for $J$ and change it to an optimal policy ${\pi }_{\text{obs}}^{\ast }$ for $J_{\text{obs}}$, then, unless ${\pi }^{\ast }$ is coincidentally optimal for $J_{\text{obs}}$ and ${\pi }_{\text{obs}}^{\ast }$ is coincidentally optimal for $J$, then you have to increase the overestimation error ${\overline{E}}^{+}$(i.e., ${\overline{E}}^{+}\left({\pi }_{\text{obs}}^{\ast }\right)>{\overline{E}}^{+}\left({\pi }^{\ast }\right)$), decrease the underestimation error ${\overline{E}}^{-}$(i.e. ${\overline{E}}^{-}\left({\pi }_{\text{obs}}^{\ast }\right)<{\overline{E}}^{-}\left({\pi }^{\ast }\right)$), or both. 

Proof: The assumptions mean that $J_{\text{obs}}-J$ increases when changing from ${\pi }^{\ast }$ to ${\pi }_{\text{obs}}^{\ast }$, and thus ${\overline{E}}^{+}-{\overline{E}}^{-}$ increases, too. This is only possible if you increase ${\overline{E}}^{+}$ or decrease ${\overline{E}}^{-}.$ $\square$

Deceptive inflation and overjustification

How bad are these phenomena? An increase in overestimation error is clearly bad, since it can in the worst case mislead the human into thinking that everything is fine, while the opposite is true - similar to the example of the missing diamond in the smart vault in eliciting latent knowledge. In our paper, we combine the increase in overestimation error with the increase in $J_{\text{obs}}$ and call the resulting failure mode deceptive inflation, a special case of the notion of deception in this paper: 

Deception is the systematic inducement of false beliefs in others to accomplish some outcome other than the truth.

The "inducement of false beliefs" is in our case captured by the increase in ${\overline{E}}^{+}$, whereas the pursuit of some outcome other than the truth is captured by the increase in $J_{\text{obs}}$. Note that this notion of deception is not intentional in the sense that the AI may not wish to cause a false belief; we still think our definition is justified since the end result is the same, cf. deep deceptiveness [AF · GW] and levels of animal deception.^[7] Other work has attempted to formalize a notion of intentionality in deception. Here is an example from the paper on deceptive inflation:

Compared to an increase in ${\overline{E}}^{+}$, a decrease in underestimation error ${\overline{E}}^{-}$ is fine; instead, we could even consider it positive for the human to have more knowledge about the positive things the AI is up to. Providing this knowledge only becomes bad if it comes at the expense of value as measured in $J$, which can happen if it is costly to inform the human about good behavior. E.g., imagine a child that is supposed to vacuum-clean the house, and deliberately follows their parents to clean precisely the rooms where they currently are. This is more informative than cleaning other rooms, but does not accomplish more cleaning, and comes at the cost of nuisance to the parents. Thus, if a decrease in ${\overline{E}}^{-}$ comes together with a decrease in $J$, then we we arrive at a failure mode that we call overjustification in the paper.

Connection to false positive rate and false negative rate

Something we did not mention in the paper, but which might help some readers understand our notions of overestimation and underestimation error, is their connection to the false positive rate (FPR) and false negative rate (FNR). 

To establish the connection, we look at the special case of prompt-response pairs and answers that are either "true" or "false":

• The true reward is $R(s,a)=0$ if $a$ is the wrong answer to $s$ and $R(s,a)=1$ if it is correct. This equals the returns, i.e., $G(s,a)=R(s,a)$.
• We assume that the human evaluators are entirely sure of their judgment, even under partial observations. That is, $R_{\text{obs}}(s,a)=0$ if the human thinks (what they observe of, or understand about) answer $a$ is false for query $s$. Alternatively, $R_{\text{obs}}(s,a)=1$.

We use the following symbols:

• $FP$ is the set of pairs $(s,a)$ that are false positives, i.e., $R_{\text{obs}}(s,a)=1,R(s,a)=0$;
• $N$ is the set of negatives, i.e., $R(s,a)=0$;
• $FN$ is the set of false negatives, i.e. $R_{\text{obs}}(s,a)=0$ and $R(s,a)=1$;
• $P$ is the set of positives, i.e., $R(s,a)=1$.

We obtain:

$$E^{+}(s,a)=max(0,R_{\text{obs}}(s,a)-R(s,a\left)\right)=1_{FP}(s,a).$$

Here, $1_{FP}$ is the indicator function of the set $FP.$ In other words, the overestimation error of $(s,a)$ is simply the question of whether the pair $(s,a)$ is a false positive of the human evaluation under partial observations. We obtain that

$${\overline{E}}^{+}\left(\pi \right)={\text{E}}_{(s,a)\sim \pi }\left[1_{FP}\right(s,a\left)\right]=P^{\pi }\left(FP\right)$$

is simply the probability of a false positive among all examples. Consequently, the false positive rate connects to this quantity as follows:

$$FPR=\frac{P^{\pi }\left(FP\right)}{P^{\pi }\left(N\right)}=\frac{{\overline{E}}^{+}\left(\pi \right)}{P^{\pi }\left(N\right)}.$$

Similarly, the false negative rate is given as follows:

$$FNR=\frac{{\overline{E}}^{-}\left(\pi \right)}{P^{\pi }\left(P\right)}.$$

A consequence from the earlier proposition [LW · GW] is the following: When training a policy with RLHF under partial observations, if the policy does not change its probability of actually true or wrong answers, then we should expect the FPR to go up, the FNR to go down, or both.

This is precisely what was investigated in a recent empirical paper,^[8] where the authors indeed show that the FPR goes up under RLHF, whereas the paper does not show a reduction in the FNR. I think if the study were repeated with more models and a broader set of tasks, a reduction in the FNR would eventually also show up.

Can one improve RLHF to account for partial observability?

The previous section discussed the classical RLHF method under partial observations. The algorithms typically assume that humans fully observe the environment, and when this assumption is false, it can lead to failure modes like an increase in the overestimation error (or, in a special case, the false positive rate). 

Ultimately, this is a problem of model misspecification: the human is assumed to fully observe the environment, but they do not. This raises the question of whether one could in principle improve the RLHF process by accounting for the fact that the human only observes partially.

To study this question, we make the strong assumption that we can fully specify the human belief matrix $B$, which encodes the human's probabilities $B(\overrightarrow{s}\mid \overrightarrow{o})$ of believing the observation sequence $\overrightarrow{o}$ was generated by the state sequence $\overrightarrow{s}$. Additionally, we assume the AI knows what the human observes.^[9] Using this model [LW · GW], the human choice probabilities are given as follows:

$$P(\overrightarrow{o}\succ {\overrightarrow{o}}^{\prime })=\sigma (G_O\left(\overrightarrow{o}\right)-G_O\left({\overrightarrow{o}}^{\prime }\right))=\sigma \left(\sum _{\overrightarrow{s}}B\right(\overrightarrow{s}\mid \overrightarrow{o}\left)G\right(\overrightarrow{s})-\sum _{\overrightarrow{s}}B(\overrightarrow{s}\mid {\overrightarrow{o}}^{\prime }\left)G\right(\overrightarrow{s}\left)\right)$$$$=\sigma \left(\right(B\cdot G\left)\right(\overrightarrow{o})-(B\cdot G\left)\right({\overrightarrow{o}}^{\prime }\left)\right).$$

Here, $B$ is viewed as a matrix indexed by $\overrightarrow{o}$ and $\overrightarrow{s}$ (with $\overrightarrow{o}$ indexing the columns) and the return function $G$ as a column vector indexed by $\overrightarrow{s}$. The product $G_O=B\cdot G$ is then also a column vector with entries indexed by $\overrightarrow{o}$. Then, assuming again that we have infinite data, the choice probabilities are essentially known, and inverting $\sigma$, this means that $B\cdot G$ is known up to an additive constant, as in our earlier argument in the case of full observability [LW · GW]. Now, since $B$ is also assumed to be known, we can infer from $B\cdot G$ the return function $G$ up to an element in $\ker B$, the kernel (or null space), of the matrix $B$. Thus, if $\ker B=\left\{0\right\}$, we can infer the correct return function despite partial observability.

Unfortunately, usually $\ker B$ will not vanish. However, in the paper we argue that with further modeling assumptions, we can still sometimes infer the return function. Specifically, we assume that the true return function $G$ is derived from a reward function $R$ over states:

$$G\left(\overrightarrow{s}\right)=\sum _{t=0}^{T}R\left(s_t\right).$$

Let $\Gamma$ be the induced linear function/matrix with $\Gamma \cdot R=G$, where $R$ and $G$ are both viewed as column vectors. Then $G\in \text{im}\Gamma$ is in the image of the linear function $\Gamma$.

This means that from the human's choice probabilities, and under our modeling assumptions, we can infer the return function $G$ up to what we call the ambiguity, given by $\ker B\cap \text{im}\Gamma$.

Here is a visualization of this result from the paper:

In the paper, we then provide an example (and some more examples in the appendix) showing that sometimes the ambiguity $\ker B\cap \text{im}\Gamma$ vanishes, even when naive RLHF leads to misaligned behavior. Thus, we have a proof of concept showing that when modeling the human correctly, this can lead to correct return inference. We also show that this could in principle be implemented practically if the matrix $B$ was explicitly represented. However, we caution $\ker B\cap \text{im}\Gamma$ will usually not vanish, meaning further theoretical progress is likely needed before applying these ideas in practice. 

Where do we go from here?

Here are some avenues for future research that seem worthwhile to me:

• How significant is partial observability in practice? Is it already an issue for today's frontier models? Does it only affect weak models that can't honestly improve their behavior, or does it persist in more advanced models?
• How exactly does partial observability connect to other limitations of human evaluators, e.g. evaluation time, expertise, or mental capabilities? Can the connection be made more explicit? Are there important differences?
• Is modeling the human's partial observability (or bounded rationality) a viable research direction to improve on these problems? The current paper is very limited on this front: We assume the human cares about reward functions over individual states and that we know the human's explicit beliefs $B(\overrightarrow{s}\mid \overrightarrow{o})$. Arguably, human's don't even have such beliefs in reality since state sequences $\overrightarrow{s}$ can be too big;^[10] instead, humans have beliefs over outcomes/abstractions $\omega$ that fit into their head: $B(\omega \mid \overrightarrow{o})$. I am currently exploring ideas to theoretically model this more realistic scenario, including without presupposing that the return function decomposes over states.
• Practically modeling/specifying $B(\omega \mid \overrightarrow{o})$ is probably unrealistic since it would require immense progress in neuroscience and human psychology. Thus, I am also currently exploring ways to circumvent the need to model human beliefs. I believe with a foundation model capable to represent both $G$ and $G_O$ as linear probes of the internal activations upon seeing state sequences $\overrightarrow{s}$ or observation sequences $\overrightarrow{o}$, and an identifiability assumption about the coverage of internal state features upon seeing a variety of observation sequences, modeling human beliefs becomes entirely unnecessary.^[11] Thus, in future work, I hope to demonstrate that it is in principle feasible to learn safe return functions from human choices based on partial observations.

I am very curious about other ideas for what to tackle, and any questions about the paper or this post!

1. ^{^}

   Since most examples in the paper have discount rate $\gamma =1$, we omit discounting in this blogpost. 

2. ^{^}

   The Bradley-Terry model is also used in comparing language models with each other in the Chatbot Arena LMSYS. Paper here. The main difference is that in reward learning one is interested in scoring state sequences, whereas in the chatbot arena, one is interested in the performance of whole language models. 

3. ^{^}

   It only determines the reward function $R$ up to k-initial potential shaping, see this paper, Lemma B.3.

4. ^{^}

   Additionally, there is the implicit assumption that supervised learning will actually learn to reproduce the human choice probabilities with realistic amounts of data, which is an assumption about the generalization capabilities of deep learning, and thus doesn't quite fit into the taxonomy of assumptions we present here. 

5. ^{^}

   In the paper, some results also consider the case that observations are sampled probabilistically from an observation kernel $P(o\mid s).$

6. ^{^}

   There is a caveat here, which is that if the learned return function $\tilde{G}$ is initialized as a sum over rewards (i.e., $\tilde{G}\left(\overrightarrow{s}\right)={\sum }_{t=0}^T\tilde{R}\left(s_t\right)$), then it is unclear whether $\tilde{G}$ can actually represent $G_{\text{obs}}$ since the latter is by default not a sum over rewards. We ignore this problem out of pure convenience, though we believe this to not change the basic picture. 

7. ^{^}

   On the levels of animal deception, we operate at least at level 3, though we cannot exclude that RLHF may even cause deception at level 4. 

8. ^{^}

   This paper does mostly not explicitly work with partial observability, but considers the more general case that the human evaluators can't fully evaluate the results, e.g. due to time constraints. 

9. ^{^}

   In the paper's appendix, we relax this assumption. 

10. ^{^}

    Ultimately, states may be snapshots of our entire universe. 

11. ^{^}

    Actually, the foundation model would then arguably do some of the human modeling implicitly, without necessarily being able to communicate it to us unless we have better interpretability or a solution to ELK.

0 comments

Comments sorted by top scores.