Generalizing Across Multi-Objective Reward Functions in Deep Reinforcement Learning

2018·Arxiv

Abstract

Abstract

Many reinforcement-learning researchers treat the reward function as a part of the environment, meaning that the agent can only know the reward of a state if it encounters that state in a trial run. However, we argue that this is an unnecessary limitation and instead, the reward function should be provided to the learning algorithm. The advantage is that the algorithm can then use the reward function to check the reward for states that the agent hasn’t even encountered yet. In addition, the algorithm can simultaneously learn policies for multiple reward functions. For each state, the algorithm would calculate the reward using each of the reward functions and add the rewards to its experience replay dataset. The Hindsight Experience Replay algorithm developed by Andrychowicz et al. (2017) does just this, and learns to generalize across a distribution of sparse, goal-based rewards. We extend this algorithm to linearly-weighted, multi-objective rewards and learn a single policy that can generalize across all linear combinations of the multi-objective reward. Whereas other multi-objective algorithms teach the Q-function to generalize across the reward weights, our algorithm enables the policy to generalize, and can thus be used with continuous actions.

1. Introduction

Reinforcement Learning has successfully been used to accomplish many tasks, such as learning to play games (Mnih et al., 2015; Silver et al., 2017), control robots (Ng et al., 2006; Levine et al., 2015), and drive cars (Bojarski et al., 2016).

In model-free reinforcement learning, the environment model is considered a black box, so the agent can only interact with it through sample runs. The reward function, however, quantifies the task that the problem-designer wants the agent to accomplish, and so is not inherently part of the environment. Therefore, the reward function could be incorporated into the learning algorithm rather than being included in the environment model. The algorithm would then be able to check what the reward is for any state or action, even ones it hasn’t encountered in its sample runs. If the designer wants the agent to achieve a high reward for a number of reward functions, the algorithm could check what the reward is for each function and then use an off-policy algorithm to simultaneously learn the value function of each reward. Alternatively, if the distribution of reward functions can be parameterized, then the parameters can be fed into the policy, which will learn different behaviours for different reward functions.

For example, Andrychowicz et al. (2017) develop the Hindsight Experience Replay (HER) method which demonstrates that a neural network policy can generalize across sparse, goal-oriented rewards. The agent learns how to reach multiple goal states and can even interpolate to goal states that it did not specifically see before. They add the desired goal to the state representation and augment the experience replay dataset with alternate goals in order to learn to generalize to new goals.

We extend this result to another class of reward functions: multi-objective rewards with a linear scalarization function. Multi-objective reinforcement learning problems consider environments where the reward is vector valued rather than scalar. Roijers et al. (2014) provide a thorough overview. To solve a multi-objective problem, one can either find all policies on the Pareto front–that is, the set of policies in which no policy is strictly worse than any other–or one can specify a function that converts the reward vector to a scalar. The advantage of specifying a scalarization function is that one can then use the standard set of reinforcement learning algorithms to learn how to maximize the reward. One common scalarization function is a linear function that computes a weighted sum of the elements in the reward vector (Natarajan and Tadepalli, 2005; Barrett and Narayanan, 2008; Mossalam et al., 2016). The disadvantage, however, is that in order to use standard reinforcement learning techniques, the weights must be specified up front, before training the agent. For example, if you were to design an agent that would help navigate from home to work, you might choose a reward function that weights the commute time, the cost of the route, and the cost to the environment. Depending on the day, you might consider cost to be more important than time, or vice versa, and would need to retrain the algorithm each time you changed your mind.

We solve this problem by extending Hindsight Experience Replay to the multi-objective reward case. The weights can then be input to the agent in real time, even after the agent is trained. Rather than retraining your navigation agent with new priorities, you could just input your new priorities to the agent and it would adjust its policy accordingly. Our algorithm can be applied to any off-line, deep reinforcement-learning algorithm that uses an experience replay dataset, and can be easily applied with minimal changes to the underlying algorithm.

Ideas similar to ours exist, most notably the work of Castelletti et al. (2012), who use fitted Q-iteration (Ernst et al., 2005) and add the linear weight vector to the state representation in order to learn a Q-function that can generalize across all linear combinations of the reward. Our solution allows policy functions to generalize across weight vectors, in addition to Q-functions. Thus, our algorithm can also be applied to environments with continuous action spaces.

2. Background

The reinforcement learning problem can be represented as a Markov Decision Process. At each timestep, t, the agent observes the environment state , chooses an action using its policy ), receives a reward ), and the environment transitions to the next state with probability

The discounted sum of rewards is called the return, a discounting factor used to ensure that the return converges. The goal of reinforcement learning is to learn the optimal policy ) that maximizes the expected return, arg max

policy from each action from a given state and is defined as

. A multi-objective reward will also typically be defined with a scalarization function that maps the vector reward to a single, optimizable scalar. In this paper, we consider a linear scalar function, is a set of weights 1] such that the weights sum to 1.

2.1 Hindsight Experience Replay

Many neural-network based reinforcement-learning algorithms employ a replay dataset of prior experiences. For example, deep Q-networks (Mnih et al., 2015) use Q-learning with a replay dataset and some other tricks to learn an optimal Q function. Actor critic with experience replay (Wang et al., 2016) is an off-policy actor critic method that adds an experience replay dataset and trust-region updates. Deep deterministic policy gradients (DDPG) (Lillicrap et al., 2015) is an off-policy policy-gradient algorithm for deterministic policies. The replay dataset contains transition tuples that are aggregated into mini-batches and used to fit the neural networks.

Andrychowicz et al. (2017) develop HER to deal with the case where the reward is goal-based. The reward is 0 when the agent reaches the goal state and 1 in all other states. They include the goal, g, in the transition tuples–and feed the goal to the policy and Q function. When they replay the dataset to the neural network, they swap the original goal state in some of the tuples with the state that the agent actually reached. This allows the agent to learn both from trial runs where the agent manages to achieve its goal and from the ones where it fails. In addition, the agent learns to generalize across goals, so that when the agent is online, the user can input a new goal that the agent might never have seen during training.

3. Reward Generalization

The data-augmentation technique of Hindsight Experience Replay should not, in principle, be limited to generalizing across goal-based reward functions. Other families of reward functions should also be generalizable, so long as the reward function can be parameterized and fed to the policy and Q functions.

Consider a class of reward function, ), parameterized by w. Additional rewards can be sampled from this class and added to the replay dataset, so that the agent can learn to generalize across the reward class. At every timestep, k additional rewards can be chosen from the reward family and add them to the replay dataset . The reward parameters are then fed to both the policy network, ), and the state-action network, ), and trained as normal.

The agent should be able to generalize across a variety of reward classes. Andrychowicz et al. (2017) demonstrate generalization over goal-based rewards. In this work we demonstrate generalization over linearly weighted multi-objective rewards. Another reward class

Figure 1: Results on 1D double-integrator as a function of –the weight for prioritizing time to goal (). Note that . The agent started from position needed to reach x = 0. The orange line shows the theoretical optimal solution and the blue line shows the solution achieved by the agent. For this experiment k = 4. 1a) Total time to reach goal as a function of 1b) Total fuel used as a function of 1c) Total cost incurred by agent as a function of

that would be interesting to explore would be to generalize over the parameter, so that the agent could dynamically adjust its reward horizon.

3.1 Multi-objective Generalization

We apply this data-augmentation technique to multi-objective rewards with linear weights. At every timestep, we add k tuples to the replay dataset For example, if k = 2 and the agent sees state , performs action , and received reward we can add both the tuple , and the tuple dataset. We feed the weight vector to both the policy network, ), and state-action network, ), and train as normal using the augmented dataset.

Since the learning algorithm has access to the reward function, it can choose a strategy for sampling from the reward distribution. For now, though, we uniformly sample the weights we use for augmenting the dataset.

4. Environment

We tested the algorithm on a double integrator environment. The agent controls the acceleration of a particle moving along a line and tries to reach a specified goal position. Every time step, the agent receives a two dimensional reward. The first component, encourages the agent to reach the goal quickly: 1 until the agent reaches the goal, at which point = 0. The second component encourages the agent to minimize fuel use: where 1] is the acceleration.

The optimal solution to this problem depends on the weighting of the two reward components ). If all the weight goes toward reaching the goal quickly, then the optimal solution takes the form of a bang-bang control system–the particle accelerates half the distance to the goal and decelerates the rest of the way. This solution does not care about the increased fuel use that the extra speed requires. If, however, more weight is put

Figure 2: A sample run of the 1D double-integrator starting from position . Blue is the learned agent’s action and orange is the optimal solution. 2a) The action that the agent chose. 2b) The velocity of the agent over the course of the episode. 2c) The position of the agent over the course of the episode.

on minimizing fuel use, then the optimal solution is a bang-off-bang controller–the particle accelerates part of the way, coasts for a while, and then decelerates.

We also scaled this problem up to the N-dimensional case, in which the agent controls the acceleration along each axis of an N-dimensional space. In order to make the N-dimensional case harder than N one dimensional cases, the total acceleration that the agent could apply was limited to 1. In other words, = 1. The reward is a 2N-dimensional vector, with two reward components for each axis.

5. Results

We used the DDPG algorithm (Lillicrap et al., 2015) with policy and value networks each consisting of two hidden layers of 256 neurons with ReLU activation functions, a sigmoid activation on the policy output, and a linear activation on the value network output.

5.1 1D Results

Figure 1 compares the theoretical optimal results with those of the algorithm as a function of the weight on increases, the agent takes less time to reach the goal (Figure 1a), which makes sense, since prioritizes a low travel time. Note, that when is 0, the time to reach the goal approaches infinity. This is because all the weight is on minimizing fuel use, so the best strategy is to not move at all. The algorithm, however, outputs a small, constant acceleration and did not reach the goal, which is why fuel use is not 0 (Figure 1b).

Except for the outlier at = 0, fuel use does increase as increases, as it should. For smaller , the optimal strategy would be to accelerate with maximum acceleration, then coast, and finally decelerate with maximum deceleration. The algorithm instead learned to reduce fuel by minimizing its maximum acceleration. The agent does not learn the optimal solution as can be seen in Figure 2, but it does come close and it definitely learns an appropriate policy for each weight.

5.2 2D Results

In the 2D case, the agent controls the acceleration along two axes. The reward, therefore, is a 4-dimensional vector: travel time along axis 1, fuel use of axis 1, travel time along axis

Figure 3: Fuel use and travel time of agent in 2-D double-integrator. The agent started from position 90 and needed to reach x = 0. Figures 3a and 3c show the total fuel use and travel time, respectively, of the agent along axis 1 (blue) and axis 2 (orange) as the axis 1 weights trade off between minimizing travel time () and minimizing fuel use (). Figures 3b and 3d show the same thing, but trading off between minimizing travel time () and minimizing fuel use (axis 2.

2, fuel use of axis 2. The weights, trade off between the different reward dimensions. As the dimension of the weight vector increases, more samples should be necessary for the agent to learn to generalize because the volume of the weight space is bigger. Despite, this the agent was able to learn the correct policy for each given weight.

Figure 3 shows results of the training as the weights trade off between minimizing fuel use and minimizing travel time along axis 1 (Figures 3a and 3c) and axis 2 (Figures 3b and 3d). When the weight varies between minimizing fuel along axis 1 to minimizing time along axis 1, the fuel use of axis 1 increases and the time along axis 1 decreases. The same is true when the weight varies on axis 2. This behaviour makes sense, since more fuel is required to reach the destination more quickly. Note that when the weights prioritize axis 1, the agent does not care about minimizing fuel use or time on axis 2. The same is true when prioritizing axis 2–in Figure 3b, the fuel use along axis 2 increases, but along axis 1 remains flat, and in Figure 3d the time decreases along axis 2, but jumps up along axis 1.

5.3 Augmentation Rate

Our algorithm introduces one main hyperparameter, the augmentation rate k. This parameter controls how many augmented experiences should be added to the replay dataset in relation to normal experiences. For example, if k = 3, then for every piece of data collected from the environment three augmented experiences are added. For the 1D environment, we examined the cost received (negative reward) as a function of augmentation rate k, where

Figure 4: Augmentation rate, k, as a function of average cost

the cost is averaged across all weights. Figure 4 shows the results. The cost is highly variable as a function of k, but there is a distinct minimum at k = 2. If k is too low, then the algorithm would not have enough alternate experiences to generalize, but if k is too high, then the algorithm is only learning from off-policy data, and learning would be slower. Andrychowicz et al. Andrychowicz et al. (2017), however, report that in their experiments k = 4 was optimal, which means that the optimal augmentation rate highly depends on the environment or the intended reward class that should be generalized.

6. Conclusion

We demonstrate that providing the reward function to the learning algorithm can be useful for teaching the agent to generalize across a distribution of rewards. Specifically, we demonstrate that a neural network policy can learn to generalize across the weights of a linearly weighted, multi-objective reward function.

Future work would extend these results to other classes of reward functions, such as learning to generalize across non-linearly weighted multi-objective rewards, or across differ-ent values of

References

Marcin Andrychowicz, Filip Wolski, Alex Ray, Jonas Schneider, Rachel Fong, Peter Welin- der, Bob McGrew, Josh Tobin, Pieter Abbeel, and Wojciech Zaremba. Hindsight experience replay. CoRR, abs/1707.01495, 2017. URL http://arxiv.org/abs/1707.01495.

Leon Barrett and Srini Narayanan. Learning all optimal policies with multiple criteria. In Proceedings of the 25th International Conference on Machine Learning, ICML ’08, pages 41–47, New York, NY, USA, 2008. ACM. ISBN 978-1-60558-205-4. doi: 10.1145/1390156. 1390162. URL http://icml2008.cs.helsinki.fi/papers/257.pdf.

Mariusz Bojarski, Davide Del Testa, Daniel Dworakowski, Bernhard Firner, Beat Flepp, Prasoon Goyal, Lawrence D. Jackel, Mathew Monfort, Urs Muller, Jiakai Zhang, Xin Zhang, Jake Zhao, and Karol Zieba. End to end learning for self-driving cars. CoRR, abs/1604.07316, 2016. URL http://arxiv.org/abs/1604.07316.

A. Castelletti, F. Pianosi, and M. Restelli. Tree-based fitted q-iteration for multi-objective markov decision problems. In The 2012 International Joint Conference on Neural Networks (IJCNN), pages 1–8, 2012.

Damien Ernst, Pierre Geurts, and Louis Wehenkel. Tree-based batch mode reinforcement learning. J. Mach. Learn. Res., 6:503–556, December 2005. ISSN 1532-4435. URL http://www.jmlr.org/papers/v6/ernst05a.html.

Sergey Levine, Chelsea Finn, Trevor Darrell, and Pieter Abbeel. End-to-end training of deep visuomotor policies. CoRR, abs/1504.00702, 2015. URL http://arxiv.org/abs/ 1504.00702.

Timothy P. Lillicrap, Jonathan J. Hunt, Alexander Pritzel, Nicolas Heess, Tom Erez, Yuval Tassa, David Silver, and Daan Wierstra. Continuous control with deep reinforcement learning. CoRR, abs/1509.02971, 2015. URL http://arxiv.org/abs/1509.02971.

Volodymyr Mnih, Koray Kavukcuoglu, David Silver, Andrei A. Rusu, Joel Veness, Marc G. Bellemare, Alex Graves, Martin Riedmiller, Andreas K. Fidjeland, Georg Ostrovski, Stig Petersen, Charles Beattie, Amir Sadik, Ioannis Antonoglou, Helen King, Dharshan Kumaran, Daan Wierstra, Shane Legg, and Demisdq Hassabis. Human-level control through deep reinforcement learning. Nature, 518:529, Feb 2015. URL https: //storage.googleapis.com/deepmind-media/dqn/DQNNaturePaper.pdf.

Hossam Mossalam, Yannis M. Assael, Diederik M. Roijers, and Shimon Whiteson. Multi- objective deep reinforcement learning. CoRR, abs/1610.02707, 2016. URL http:// arxiv.org/abs/1610.02707.

Sriraam Natarajan and Prasad Tadepalli. Dynamic preferences in multi-criteria reinforce- ment learning. In Proceedings of the 22Nd International Conference on Machine Learning, ICML ’05, pages 601–608, New York, NY, USA, 2005. ACM. ISBN 1-59593-180-5. doi: 10.1145/1102351.1102427. URL http://homes.sice.indiana.edu/natarasr/Papers/ DMCRL.pdf.

Andrew Y. Ng, Adam Coates, Mark Diel, Varun Ganapathi, Jamie Schulte, Ben Tse, Eric Berger, and Eric Liang. Autonomous inverted helicopter flight via reinforcement learning. In Marcelo H. Ang and Oussama Khatib, editors, Experimental Robotics IX, pages 363– 372, Berlin, Heidelberg, 2006. Springer Berlin Heidelberg. ISBN 978-3-540-33014-1.

Diederik Marijn Roijers, Peter Vamplew, Shimon Whiteson, and Richard Dazeley. A survey of multi-objective sequential decision-making. CoRR, abs/1402.0590, 2014. URL http: //arxiv.org/abs/1402.0590.

David Silver, Julian Schrittwieser, Karen Simonyan, Ioannis Antonoglou, Aja Huang, Arthur Guez, Thomas Hubert, Lucas Baker, Matthew Lai, Adrian Bolton, Yutian Chen, Timothy Lillicrap, Fan Hui, Laurent Sifre, George van den Driessche, Thore Graepel, and Demis Hassabis. Mastering the game of go without human knowledge. Nature, 550:354, Oct 2017. URL https://deepmind.com/documents/119/agz_unformatted_ nature.pdf. Article.

Ziyu Wang, Victor Bapst, Nicolas Heess, Volodymyr Mnih, R´emi Munos, Koray Kavukcuoglu, and Nando de Freitas. Sample efficient actor-critic with experience replay. CoRR, abs/1611.01224, 2016. URL http://arxiv.org/abs/1611.01224.

designed for accessibility and to further open science