Stochastic Approximation in Reinforcement Learning
The gradient bandit algorithm can be understood as a stochastic approximation to gradient ascent.
One Reward, Many Preferences
A gradient bandit algorithm does not merely assign probabilities to actions. It adjusts action preferences so that the resulting probabilities improve performance. Its performance objective is expected reward. The central difficulty is that the true expected reward of every action is unknown, so the algorithm must use information from the action selected and the reward observed on the current step.
The useful mental model is a chain: preferences influence action probabilities, action probabilities influence which rewards are experienced, and those choices determine expected reward. Gradient ascent tries to move the preferences in a direction that improves the expected reward.
Following an Exact Gradient Step
Gradient ascent is an adjustment procedure for maximizing an objective. Here, the objective is expected reward. An exact gradient-ascent update would use the performance gradient: the direction in preference space that increases expected reward. The gradient connects three ingredients: the probabilities assigned to actions, the preferences that produce those probabilities, and the true expected rewards of the actions.
The gradient bandit algorithm addresses this limitation by replacing unavailable complete information with a sampled performance-gradient contribution. It selects an action, observes its reward, and uses that experience to construct an update.
Tracing a Sampled Update
On a particular step, the update uses the selected action A_t, the observed reward R_t, the average reward R̄_t, and an indicator showing whether the preference under consideration belongs to the selected action. The selected action and observed reward provide the current sample. The average reward supplies the comparison level used by the sampled contribution.
One Sampled Preference Update
Consider one step on which the algorithm selects an action, observes its reward, and compares that reward with the current average reward.
Select: The algorithm identifies the action A_t selected on the current step.
Observe: It records the reward R_t produced by that selected action.
Compare: It uses the average reward R̄_t as the comparison level in the sampled performance-gradient contribution.
Identify: An indicator identifies whether the preference being considered belongs to the selected action.
Update: These available quantities form a sampled gradient contribution, which is used to adjust preferences rather than relying on the unavailable true action values for every action.
The single observed reward does not reveal the complete performance gradient. It supplies one stochastic contribution whose expected value participates in the exact gradient-ascent direction.
The key point is not that one reward equals the full gradient. It does not. The key point is that the reward provides a valid sample of the information needed for the update. Repeated sampled updates are connected to the exact gradient through their expected value.
Why Stochastic Updates Match
The performance gradient can be rewritten as an expectation over the random action A_t. The unknown true value q*(A_t) can be replaced inside this expectation by the observed reward because the expected reward conditioned on the selected action satisfies E[R_t | A_t] = q*(A_t). Therefore, sampling an action and observing its reward provides the stochastic information needed for the update.
| Update approach | Information used | Relationship to the performance gradient |
|---|---|---|
| Exact gradient ascent | The true action values for the actions | Directly calculates the exact direction |
| Stochastic gradient ascent | The selected action and its observed reward | Provides a sample whose expected value matches the exact direction |
The distinction is between complete unavailable information and sampled available information.
From Function Examples to Generalization
Function approximation uses examples to construct a broader representation of a desired function. In reinforcement learning, the desired function can be a value function. The learner may have examples from that function without having a complete description of the function itself.
The examples, the desired function, and the approximation are different things. The desired function is the function the learner is trying to represent. The examples are the evidence available from that function. The approximation is the generalized representation constructed from those examples. The purpose is not simply to store the examples; it is to generalize from them.
Function approximation is an instance of supervised learning. Supervised learning is also studied in artificial neural networks, pattern recognition, and statistical curve fitting. This relationship matters because reinforcement learning can combine its own methods with generalization methods already studied in those fields.
Common Reasoning Errors
Treating one observed reward as the complete performance gradient.
A sampled reward provides one stochastic contribution. It does not provide the complete set of true action values.
Fix:
View the update as a sample whose expected value connects to the exact gradient-ascent direction.Assuming exact gradient ascent can be applied directly when true action values are unknown.
The true action values q*(b) are not known in the setting described.
Fix:
Use the selected action and observed reward to construct a sampled update.Describing the gradient bandit algorithm as only a probability-assignment method.
The algorithm adjusts preferences so that the resulting action probabilities improve expected reward.
Fix:
Trace the relationship from preferences to probabilities to expected reward.Equating the desired function with the approximation.
The desired function, available examples, and constructed approximation are distinct parts of the process.
Fix:
Identify which object is the target, which information consists of examples, and which representation is built from them.Assuming reinforcement learning must create a completely new generalization method.
Function approximation connects reinforcement learning with existing work on generalization.
Fix:
Recognize that methods from related fields can take the role of a function approximator, although they may not all fit equally conveniently into reinforcement learning algorithms.
Apply the Two Connections
Explain the following process in your own words: a gradient bandit selects an action, observes a reward, compares that reward with the average reward, and changes preferences. Then explain why repeated updates can have the same direction in expectation as exact gradient ascent even though the true action values are unknown.
Hints
- Name the performance objective first.
- Separate one sampled contribution from the full performance gradient.
- Use the conditional expected-reward relationship to explain the equivalence.
- Mention the indicator for whether a preference belongs to the selected action.
A reinforcement-learning system has examples from a desired value function but does not have a complete description of that function. Identify the desired function, the examples, and the approximation in this situation. Finally, explain why supervised-learning methods may be relevant.
Hints
- The desired function is the function the learner is trying to represent.
- The examples are the available evidence.
- The approximation is the generalized representation constructed from that evidence.
- Function approximation is an instance of supervised learning.
The Complete Picture
- Gradient ascent treats expected reward as the performance objective and adjusts preferences in a direction intended to improve it.
- Exact gradient ascent requires true action values, but those values are unknown in the gradient bandit setting.
- A selected action and its observed reward provide a sampled performance-gradient contribution, with the average reward and an action indicator also involved in the update.
- Because the expected reward conditioned on the selected action equals its true action value, the expected value of sampled updates matches the exact gradient-ascent direction.
- Function approximation constructs a broader representation of a desired function from available examples and connects reinforcement learning with supervised-learning methods for generalization.
Key Takeaways
- The gradient bandit algorithm is a stochastic approximation to gradient ascent.
- Its goal is to improve expected reward by adjusting action preferences, not merely by assigning action probabilities.
- Observed rewards replace unavailable true action values in sampled updates, and the expected value of those updates matches the exact gradient direction.
- Function approximation uses examples of a desired function to construct a broader representation.
- Because function approximation is an instance of supervised learning, reinforcement learning can combine with established generalization methods.