Allen's reinforcement learning notes
Contents
Links
Motivation
Consider a problem where we have to train a robot to pick up some object. A traditional ML algorithm might try to learn some function f(x) = y, where given some position x observed via the camera we output some behavior y. The trouble is that in the real world, the correct grab location is some function of the object and the physical environment, which is hard to intuitively ascertain by observation.
The motivation behind reinforcement learning is to repeatedly take observations, then sample the effects of actions on those observations (reward and new observation/state). Ultimately, we hope to create a policy pi that maps states or observations to actions.
Learning
Learning involves the agent taking actions and the environment returning a new state and reward.
- Input: : States at each time step
- Output: : Actions at each time step
- Data:
- Learn Failed to parse (unknown function "\math"): {\displaystyle \pi_\theta : s_t -> a_t <\math> to maximize <math> \sum_t r_t <\math> === State vs. Observation === A state is a complete representation of the physical world while the observation is some subset or representation of s. They are not necessarily the same in that we can't always infer s_t from o_t, but o_t is inferable from s_t. To think of it as a bayes net, we have * s_1 -> o_1 - (pi_theta) -> a_1 (policy) * s_1, a_1 - (p(s_{t+1} | s_t, a_t) -> s_2 (dynamics) Note that theta represents the parameters of the policy (for example, the parameters of a neural network). Assumption: Markov Property - Future states are independent of past states given present states. This is the fundamental difference between states and observations. === Problem Representation === States and actions are typically continuous - thus, we often want to model our output policy as a density function, which tells us the distribution of probabilities of actions at some given state. The reward is a function of the state and action r(s, a) -> int, which tells us what states and actions are better. When choosing hyperparameters we need to be careful to make sure that we go for completing long term goals instead of always looking for immediate reward. === Markov Chain & Decision Process=== Markov Chain: <math> M = {S, T} <\math>, where S - state space, T- transition operator. The state space is the set of all states, and can be discrete or continuous. The transition probabilities is represented in a matrix, where the i,j'th entry is the probability of going into state i at state j, and we can express the next time step by multiplying the current time step with the transition operator. Markov Decision Process: <math> M = {S, A, T, r} <\math>, where A - action space. T is now a tensor, containing the current state, current action, and next state. We let T_{i, j, k} = p(s_t + 1 = i | s_t = j, a_t = k). r is the reward function. === Reinforcement Learning Algorithms - High-level === # Generate Samples (run policy) # Fit a model/estimate something about how well policy is performing # Improve policy # Repeat === Temporal Difference Learning === Temporal Difference (TD) is a model for estimating the utility of states given some state-action-outcome information. Suppose we have some initial value <math>V_0(s) } , and we get some information . We can then use the update equation . Here \alpha represents the learning rate, which is how much new information is weighted relative to old information, while \gamma represents the discount factor, which can be thought of how much getting a reward in the future factors into our current reward.
Q Learning
Q Learning gives us a way to extract the optimal policy after learning. Instead of keeping track of the values of individual states, we keep track of Q values for state-action pairs, representing the utility of taking action a at state s.
How do we use this Q value? Two main ideas.
Idea 1: Policy iteration - if we have a policy and we know , we can improve the policy, by deterministically setting the action at each state be the argmax of all possible actions at the state.
Idea 2: Gradient update - If , then a is better than average. We will then modify the policy to increase the probability of a.
