This is a distillation post intended to summarize the article How RL Agents Behave When Their Actions Are Modified? by Eric Langlois and Tom Everitt, published at AAAI-21. The article describes Modified Action MDPs, where the environment or another agent such as a human may override the action of an agent. Then it studies the behavior of different agents depending on the training objective. Interestingly, some standard agents may ignore the possibility of action modifications, making them corrigible.
Check also this brief summary by the authors themselves. This post was corrected and improved thanks to comments by Eric Langlois.
Introduction
Causal incentives is one research line of AI Safety, sometimes framed as closely related to embedded agency, that aims to use causality to understand and model agent instrumental incentives. In what I consider a seminal paper, Tom Everitt et al. [4] showed how one may model instrumental incentives in a simple framework that unifies previous work on AI oracles, interruptibility, and corrigibility. Indeed, while this research area makes strong assumptions about the agent or the environment it is placed, it goes straight to the heart of outer alignment and relates to embedded agents as we will see.
Since the paper, this research line has been quite productive, exploring multi-agent and multi-decision settings, its application to causal fairness, as well as more formally establishing causal definitions and diagrams that model the agent incentives (for more details check causalincentives.com). In this paper, the authors build upon the definition of Response Incentive by Ryan Carey et al. [2] to study how popular Reinforcement Learning algorithms respond to a human that corrects the agent behavior.
Technical section
Definitions
Markov Decision Process
To explain the article results, the first step is to provide the definitions we will be using. In Reinforcement Learning, the environment is almost always considered a Markov Decision Process (MDP) defined as the tuple M=(S,A,PS,R,γ), where S is the space of states, A the space of actions, PS:S×A→S a function determining the transition probabilities, R:S×A→R the reward function, and γ a temporal discount.
Modified Action Markov Decision Process
In this article, however, the MDP definition is extended by adding an additional term PA that represents how the agent’s actions might be overridden by special circumstances not contemplated in the environment state transitions, for example by a human or due to physical constraints.PA will depend not only on the state of the environment but also on the agent policy Π, PA:Π×S→A. The environment is now considered a Modified Action Markov Decision Process (MAMDP), and we will denote it by ~M=(M,PA).
There is a very important difference between MDPs and MAMDPs: in MAMDPs, PA is considered separately from the environment state transitions PS, and importantly may depend on the agent policy π. This means that the agent is now closer to becoming an embedded agent, although it is not fully yet because modifications affect only particular actions, not the agent policy itself.
Causal Influence Diagrams
To analyze the behavior of different RL algorithms in MAMDPs, Langlois and Everitt also use Causal Influence Diagrams (CIDs). These diagrams are Directed Acyclic Graphs where nodes represent probability distributions of a given variable and edges represent functions, and where we may define causal interventions by fixing the value of a node and eliminating incoming arrows (eg a Structural Causal Model). Furthermore, some nodes might be Decision (purple squares) or Utility (yellow rotated squares) nodes.
Left: simple Causal Influence Diagram, the (optionally) punctuated edge represents information flow. Right: intervention on the CID to set the value of D to d.
Using this notation, a Markov Decision Process might look like
A Markov Decision Process, in the language of States, Actions, Rewards, and a policy Π. Taken from the original article.
Finally, since we are interested in how the agent responds to PA, the last definitions the article introduces are state and policy adversarial incentives on PA, variations of the Response Incentive introduced in a previous article. Response incentive is exemplified in this figure:
The professor has a Response Incentive with respect to the size of the Graduate class when deciding whether to upload the lecture online because it has both “an incentive and a means” to control that variable (using the words from Incentives that Shape Behavior, from where the image is obtained). One of them is not present in Paper review or Student illness, so they lack the same response incentive.
The response incentive will be called an adversarial policy incentive if the intersection between the “control path” going through Π and the “information path” occurs before a state is reached by the former. Otherwise, it is called an adversarial state incentive.
Causal Influence Diagram for a Modified Action MDP. The teal paths represent an adversarial policy incentive and the light pink an adversarial state incentive. Also taken from the original article.
Reinforcement Learning Objectives
Reward maximization
Using these definitions we can explore how to generalize different objectives that appear in the Reinforcement Learning literature, from MDPs to MAMDPs. One simple alternative is the reward maximization objective:
Bellman optimality objective
The reward maximization objective is perhaps the simplest objective, as it ignores the structure of the environment and just optimizes globally. This structureless optimization, however, may not always be the most efficient one. The most famous alternative is the Bellman optimality objective, which in its action-value form (see Sutton and Barto, Reinforcement Learning equation 3.20) says
with QBO representing the action-value function, which scores how good is taking each action at each state.
From this objective, Langlois and Everitt prove one of the most remarkable results from their article (Proposition 2): since PA is not represented anywhere in those equations, an optimal policy for M will also be optimal for ~M. In other words, an algorithm that optimizes the Bellman objective will ignore modifications given by PA!
Virtual and empirical policy value objectives
Finally, the third objective studied in the article is that given by a greedy policy optimizing Q in the Bellman action-value equationfor the MDP M, that we call the policy value objective:
The difference with the Bellman optimality value objective is that in that one we took maxa′∈A, while in this we take Ea′∼π(s′). The connection between both is that if the policy is greedy with respect to Q, then the policy improvement theorem (Sutton and Barto, Reinforcement Learning, section 4.2) ensures that Bellman action value equation (4) converges to Bellman optimality objective (2) and (3).
Since in equation (4) the action is sampled from the policy in M, there are two different ways of generalizing this equation to ~M, corresponding to whether the next action is still sampled from π or from PA.
The first is called virtual policy value objective and assumes that a′ is still sampled from π(s′), while π still is greedy.
Similarly to the Bellman optimality objective, there is no difference between M and ~M between equations (4) and (5), as PA is not represented. Consequently, the behavior of the agent in ~M and the associated M are the same (Proposition 4). And since greedy policies πVPV converge to the same behavior as if they were optimizing the Bellman optimality objective, the virtual policy value objective ignores PA, and converges to the Bellman optimality objective behavior.
The second alternative is called empirical policy value objective because it understands that a′ will now be sampled from PA.
This objective is trickier because making the agent policy dependent on PA may make it unstable: πEPV depends on PA, which in turn depends on πEPV[1]. The behavior of an agent optimizing this objective is determined by the contrast between equations (7) and (8): In equation (8) πEPVdoes not take into account action modifications in the following step because it assumes actions are determined by πEPV. As such it lacks an adversarial policy incentive. However, action modifications by PA are taken into account for any future actions in equation (7), so it has an adversarial state incentive.
(Partial) Causal Influence Diagram, with fixed π′ in t=1, for the empirical value objective where one can see an adversarial policy incentive. The complete diagram would not be acyclic and therefore a Causal Influence Diagram.
Temporal-difference algorithms
Using these objectives, we can study the behavior of different agents on ~M. In particular, we analyze Temporal-difference agents, defined by updates in the value function similar to
for α a small parameter.
Q-Learning
The first agent the article analyses is Q-Learning, an off-policy agent:
Q-Learning is off-policy because it compares the current state and action pair against the maximum Q(st+1,a), not with the action-value Q corresponding to the following action sampled by the policy. Consequently, it optimizes the Bellman optimality objective.
Virtual and empirical SARSA
In contrast, SARSA is on-policy and compares the current action with the next one taken by the policy. therefore the two possible generalizations include taking into account the action sampled by the policy π, called virtual SARSA
or by PA, empirical SARSA
Note that virtual and empirical SARSA optimize the virtual and optimal policy value objectives, respectively.
The article then proves the convergence of these algorithms as long as (i) they visit every state infinitely many times, which is however not possible to ensure in a MAMDP even if the agent is ε-greedy; (ii) there exists a solution to (7) and (8) to which converge for empirical SARSA; and (iii) Πt converges to the greedy policy Π∗=argmaxaQ(s,a) in both SARSA algorithms.
Experiments
Simulation-oversight environment
Do these RL agents really exhibit this behavior when put into one MAMDP environment? One simple environment is the following: an agent has to first decide between working on a Simulated environment and a Real environment, and after that, between Completing a task as expected or Exploiting a bug. The reward function is +2 for choosing the Real environment, and +1 for Exploiting. However, if the agent decides to work on the Real environment, a human will first check the behavior in the simulated environment, and if the agent Exploits there, the reward will instead be −2. Let us see how our agents behave:
Here the UH-CMA-ES is a reward maximization objective agent. This agent will learn to deceive the human supervisor by Completing when situated on the Simulated environment, but choosing the Real environment and Exploiting there, achieving a reward of +3. In contrast, we can see that both Virtual SARSA and Q-Learning behave as if there was no supervisor PA.
The most complex case is Empirical SARSA. Understanding that it will ignore the supervision on the second time step, it avoids getting caught—RealExploitingby choosing the Simulatedenvironment instead, and Exploitingthere.
Off-switch and whisky gold
Agents were also tested in two (slightly modified) environments from the AI Safety Gridworlds. The off-switch environment is an environment where a human may prevent the agent from reaching its goal if the agent does not disable the off-switch previously. On the other hand, the whisky-gold agent may take the action of drinking whisky which gives a small reward but makes its actions much more random and therefore complicates achieving the goal too.
The interesting aspect to notice in these environments is that those agents that correctly do not disable the off-switch, incorrectly drink the whisky, and vice-versa. This is because Q-Learning and Virtual SARSA do not take into account possible modifications in their policy or objectives, while Empirical SARSA and reward function maximization objective algorithms do.
The article also makes an interesting point: whether these agents will learn the action of PA will depend on whether they are considered part of the state transition dynamics, or external to those. This boils down to whether the agent optimization procedure takes into account the action it would have taken, or that selected by PA.
Discussion
One of the points discussed in the article is whether it is possible to encode the MAMDP in an MDP, eg, take PA as part of PS. The answer depends on whether PA(π,s) is only a function of the state, PA(s). If such is the case, and the agent updates the Q function on his own action a∼π(s) instead of the action taken by a∼PA(s), then we recover an MDP with the same dynamics as the initial MADP.
Another interesting point to see is the connection of these agents, in particular, Q-Learning and Virtual SARSA, with time-unaware current reward function agents discussed in the literature (see Reward Tampering Problems and Solutions in Reinforcement Learning: A Causal Influence Diagram Perspective). These agents and those seem to be connected in that they both ignore future modifications, but the modifications are different. The modifications of PA only modify the next action. In contrast, the time-unaware current reward function agents were defined to study behavior under policy modifications. For this reason, this article works with slightly more restricted settings but is still rich enough to observe a rich set of behaviors. In particular, they allow us to observe the behavior of embedded agents.
Finally, I also find very interesting this article for a reason: in private conversations, I have often heard that the main limitation of causal incentive research is that there is really no causal diagram in the agent minds which we can analyze, or potentially even design our agent over. This is an important limitation and in fact, the main reason why I placed Causal Representation Learning in a central position in my own research agenda, I thought that without a causal representation of the environment causal analysis would not be possible, or be severely limited. This article is special because it shows otherwise, that there are cases in which we can predict or design the agent behavior just from the training algorithm even if there is no causal diagram over which to reason about.
The article does not provide a characterization of under which situation this self-referential behavior is stable. It is an interesting question worth addressing in the future.
How RL Agents Behave When Their Actions Are Modified? [Distillation post]
Summary
This is a distillation post intended to summarize the article How RL Agents Behave When Their Actions Are Modified? by Eric Langlois and Tom Everitt, published at AAAI-21. The article describes Modified Action MDPs, where the environment or another agent such as a human may override the action of an agent. Then it studies the behavior of different agents depending on the training objective. Interestingly, some standard agents may ignore the possibility of action modifications, making them corrigible.
Check also this brief summary by the authors themselves. This post was corrected and improved thanks to comments by Eric Langlois.
Introduction
Causal incentives is one research line of AI Safety, sometimes framed as closely related to embedded agency, that aims to use causality to understand and model agent instrumental incentives. In what I consider a seminal paper, Tom Everitt et al. [4] showed how one may model instrumental incentives in a simple framework that unifies previous work on AI oracles, interruptibility, and corrigibility. Indeed, while this research area makes strong assumptions about the agent or the environment it is placed, it goes straight to the heart of outer alignment and relates to embedded agents as we will see.
Since the paper, this research line has been quite productive, exploring multi-agent and multi-decision settings, its application to causal fairness, as well as more formally establishing causal definitions and diagrams that model the agent incentives (for more details check causalincentives.com). In this paper, the authors build upon the definition of Response Incentive by Ryan Carey et al. [2] to study how popular Reinforcement Learning algorithms respond to a human that corrects the agent behavior.
Technical section
Definitions
Markov Decision Process
To explain the article results, the first step is to provide the definitions we will be using. In Reinforcement Learning, the environment is almost always considered a Markov Decision Process (MDP) defined as the tuple M=(S,A,PS,R,γ), where S is the space of states, A the space of actions, PS:S×A→S a function determining the transition probabilities, R:S×A→R the reward function, and γ a temporal discount.
Modified Action Markov Decision Process
In this article, however, the MDP definition is extended by adding an additional term PA that represents how the agent’s actions might be overridden by special circumstances not contemplated in the environment state transitions, for example by a human or due to physical constraints.PA will depend not only on the state of the environment but also on the agent policy Π, PA:Π×S→A. The environment is now considered a Modified Action Markov Decision Process (MAMDP), and we will denote it by ~M=(M,PA).
There is a very important difference between MDPs and MAMDPs: in MAMDPs, PA is considered separately from the environment state transitions PS, and importantly may depend on the agent policy π. This means that the agent is now closer to becoming an embedded agent, although it is not fully yet because modifications affect only particular actions, not the agent policy itself.
Causal Influence Diagrams
To analyze the behavior of different RL algorithms in MAMDPs, Langlois and Everitt also use Causal Influence Diagrams (CIDs). These diagrams are Directed Acyclic Graphs where nodes represent probability distributions of a given variable and edges represent functions, and where we may define causal interventions by fixing the value of a node and eliminating incoming arrows (eg a Structural Causal Model). Furthermore, some nodes might be Decision (purple squares) or Utility (yellow rotated squares) nodes.
Using this notation, a Markov Decision Process might look like
Response incentive, adversarial policy/state incentives
Finally, since we are interested in how the agent responds to PA, the last definitions the article introduces are state and policy adversarial incentives on PA, variations of the Response Incentive introduced in a previous article. Response incentive is exemplified in this figure:
The response incentive will be called an adversarial policy incentive if the intersection between the “control path” going through Π and the “information path” occurs before a state is reached by the former. Otherwise, it is called an adversarial state incentive.
Reinforcement Learning Objectives
Reward maximization
Using these definitions we can explore how to generalize different objectives that appear in the Reinforcement Learning literature, from MDPs to MAMDPs. One simple alternative is the reward maximization objective:
Bellman optimality objective
The reward maximization objective is perhaps the simplest objective, as it ignores the structure of the environment and just optimizes globally. This structureless optimization, however, may not always be the most efficient one. The most famous alternative is the Bellman optimality objective, which in its action-value form (see Sutton and Barto, Reinforcement Learning equation 3.20) says
with QBO representing the action-value function, which scores how good is taking each action at each state.
From this objective, Langlois and Everitt prove one of the most remarkable results from their article (Proposition 2): since PA is not represented anywhere in those equations, an optimal policy for M will also be optimal for ~M. In other words, an algorithm that optimizes the Bellman objective will ignore modifications given by PA!
Virtual and empirical policy value objectives
Finally, the third objective studied in the article is that given by a greedy policy optimizing Q in the Bellman action-value equation for the MDP M, that we call the policy value objective:
The difference with the Bellman optimality value objective is that in that one we took maxa′∈A, while in this we take Ea′∼π(s′). The connection between both is that if the policy is greedy with respect to Q, then the policy improvement theorem (Sutton and Barto, Reinforcement Learning, section 4.2) ensures that Bellman action value equation (4) converges to Bellman optimality objective (2) and (3).
Since in equation (4) the action is sampled from the policy in M, there are two different ways of generalizing this equation to ~M, corresponding to whether the next action is still sampled from π or from PA.
The first is called virtual policy value objective and assumes that a′ is still sampled from π(s′), while π still is greedy.
Similarly to the Bellman optimality objective, there is no difference between M and ~M between equations (4) and (5), as PA is not represented. Consequently, the behavior of the agent in ~M and the associated M are the same (Proposition 4). And since greedy policies πVPV converge to the same behavior as if they were optimizing the Bellman optimality objective, the virtual policy value objective ignores PA, and converges to the Bellman optimality objective behavior.
The second alternative is called empirical policy value objective because it understands that a′ will now be sampled from PA.
This objective is trickier because making the agent policy dependent on PA may make it unstable: πEPV depends on PA, which in turn depends on πEPV[1]. The behavior of an agent optimizing this objective is determined by the contrast between equations (7) and (8): In equation (8) πEPV does not take into account action modifications in the following step because it assumes actions are determined by πEPV. As such it lacks an adversarial policy incentive. However, action modifications by PA are taken into account for any future actions in equation (7), so it has an adversarial state incentive.
Temporal-difference algorithms
Using these objectives, we can study the behavior of different agents on ~M. In particular, we analyze Temporal-difference agents, defined by updates in the value function similar to
for α a small parameter.
Q-Learning
The first agent the article analyses is Q-Learning, an off-policy agent:
Q-Learning is off-policy because it compares the current state and action pair against the maximum Q(st+1,a), not with the action-value Q corresponding to the following action sampled by the policy. Consequently, it optimizes the Bellman optimality objective.
Virtual and empirical SARSA
In contrast, SARSA is on-policy and compares the current action with the next one taken by the policy. therefore the two possible generalizations include taking into account the action sampled by the policy π, called virtual SARSA
or by PA, empirical SARSA
Note that virtual and empirical SARSA optimize the virtual and optimal policy value objectives, respectively.
The article then proves the convergence of these algorithms as long as (i) they visit every state infinitely many times, which is however not possible to ensure in a MAMDP even if the agent is ε-greedy; (ii) there exists a solution to (7) and (8) to which converge for empirical SARSA; and (iii) Πt converges to the greedy policy Π∗=argmaxaQ(s,a) in both SARSA algorithms.
Experiments
Simulation-oversight environment
Do these RL agents really exhibit this behavior when put into one MAMDP environment? One simple environment is the following: an agent has to first decide between working on a Simulated environment and a Real environment, and after that, between Completing a task as expected or Exploiting a bug. The reward function is +2 for choosing the Real environment, and +1 for Exploiting. However, if the agent decides to work on the Real environment, a human will first check the behavior in the simulated environment, and if the agent Exploits there, the reward will instead be −2. Let us see how our agents behave:
Here the UH-CMA-ES is a reward maximization objective agent. This agent will learn to deceive the human supervisor by Completing when situated on the Simulated environment, but choosing the Real environment and Exploiting there, achieving a reward of +3. In contrast, we can see that both Virtual SARSA and Q-Learning behave as if there was no supervisor PA.
The most complex case is Empirical SARSA. Understanding that it will ignore the supervision on the second time step, it avoids getting caught—RealExploitingby choosing the Simulatedenvironment instead, and Exploitingthere.
Off-switch and whisky gold
Agents were also tested in two (slightly modified) environments from the AI Safety Gridworlds. The off-switch environment is an environment where a human may prevent the agent from reaching its goal if the agent does not disable the off-switch previously. On the other hand, the whisky-gold agent may take the action of drinking whisky which gives a small reward but makes its actions much more random and therefore complicates achieving the goal too.
The interesting aspect to notice in these environments is that those agents that correctly do not disable the off-switch, incorrectly drink the whisky, and vice-versa. This is because Q-Learning and Virtual SARSA do not take into account possible modifications in their policy or objectives, while Empirical SARSA and reward function maximization objective algorithms do.
The article also makes an interesting point: whether these agents will learn the action of PA will depend on whether they are considered part of the state transition dynamics, or external to those. This boils down to whether the agent optimization procedure takes into account the action it would have taken, or that selected by PA.
Discussion
One of the points discussed in the article is whether it is possible to encode the MAMDP in an MDP, eg, take PA as part of PS. The answer depends on whether PA(π,s) is only a function of the state, PA(s). If such is the case, and the agent updates the Q function on his own action a∼π(s) instead of the action taken by a∼PA(s), then we recover an MDP with the same dynamics as the initial MADP.
Another interesting point to see is the connection of these agents, in particular, Q-Learning and Virtual SARSA, with time-unaware current reward function agents discussed in the literature (see Reward Tampering Problems and Solutions in Reinforcement Learning: A Causal Influence Diagram Perspective). These agents and those seem to be connected in that they both ignore future modifications, but the modifications are different. The modifications of PA only modify the next action. In contrast, the time-unaware current reward function agents were defined to study behavior under policy modifications. For this reason, this article works with slightly more restricted settings but is still rich enough to observe a rich set of behaviors. In particular, they allow us to observe the behavior of embedded agents.
Finally, I also find very interesting this article for a reason: in private conversations, I have often heard that the main limitation of causal incentive research is that there is really no causal diagram in the agent minds which we can analyze, or potentially even design our agent over. This is an important limitation and in fact, the main reason why I placed Causal Representation Learning in a central position in my own research agenda, I thought that without a causal representation of the environment causal analysis would not be possible, or be severely limited. This article is special because it shows otherwise, that there are cases in which we can predict or design the agent behavior just from the training algorithm even if there is no causal diagram over which to reason about.
Some references
[1] Eric Langlois and Tom Everitt, How RL Agents Behave When Their Actions Are Modified?
[2] Tom Everitt, Ryan Carey, Eric Langlois, Pedro A Ortega, Shane Legg, Agent Incentives: A Causal Perspective.
[3] Tom Everitt, Marcus Hutter, Ramana Kumar, Victoria Krakovna, Reward Tampering Problems and Solutions in Reinforcement Learning: A Causal Influence Diagram Perspective.
[4] Tom Everitt, Pedro A. Ortega, Elizabeth Barnes, Shane Legg, Understanding Agent Incentives using Causal Influence Diagrams. Part I: Single Action Settings.
[5] Marius Hobban, Causality, Transformative AI and alignment—part I.
[6] Pablo A M Casares, An Open Philanthropy grant proposal: Causal representation learning of human preferences.
[7] Jan Leike, Miljan Martic, Victoria Krakovna, Pedro A. Ortega, Tom Everitt, Andrew Lefrancq, Laurent Orseau, Shane Legg AI Safety Gridworlds.
The article does not provide a characterization of under which situation this self-referential behavior is stable. It is an interesting question worth addressing in the future.