Action Selection in Reinforcement Learning
A k-Armed Bandit Problem is a repeated decision problem with k possible actions.
The Decision Repeats
Imagine facing the same kind of decision again and again. At every time step, an agent selects one option from k available actions and receives a numerical reward. The central challenge is not simply to make one good choice. It is to choose repeatedly so that the total reward collected over a specified period is as large as possible.
A k-Armed Bandit Problem is a repeated decision problem with k possible actions.
One Action and Its Reward
A single cycle has a clear structure. The agent selects one action from the available options. That selected action identifies the stationary probability distribution used to select the numerical reward. The agent then receives the reward and reaches the next time step, where another action must be selected.
The reward is not necessarily a fixed value that always follows an action. Instead, it is selected from the probability distribution associated with that action. Different actions can have different stationary probability distributions. Stationary means that the distribution associated with an action does not change over time.
Tracing Three Repeated Choices
An agent has three available actions and must make a sequence of decisions. Trace what determines the reward on each step.
First choice: The agent selects one of the three actions. The selected action determines which stationary probability distribution is used to select the first numerical reward.
Second choice: After receiving the first reward, the agent faces another choice. It may select an action again, and that selected action determines the distribution used for the next reward.
Third choice: The process repeats at the next time step. The reward on this step comes from the distribution linked to the action selected on this step.
Each reward is connected to the action selected at its own time step. Repeating the process produces a sequence of action-reward pairs.
Stationary Reward Distributions
The word stationary describes the reward distribution linked to an action, not a promise that every reward will be identical. The distribution does not change over time, but a reward is selected from that distribution whenever the action is chosen. Therefore, an action and its reward distribution are connected, while the numerical reward produced on an individual selection is drawn from that distribution.
Selecting an action does two things conceptually: it chooses which action is used, and it identifies the stationary probability distribution from which that step's numerical reward is selected.
The Total-Reward Objective
The agent's objective is to maximize expected total reward over a time period. This objective changes how the problem should be understood. The agent is not judged only by the reward from one isolated selection. Rewards from repeated decisions contribute to the total collected over the specified period.
The phrase best levers refers to actions that are most valuable for this repeated objective. However, the problem statement does not provide the action distributions in advance. Its essential structure is that every action has a distribution and the agent must repeatedly select actions while seeking greater expected total reward.
Repeated Choice Versus One-Time Choice
| One-time choice | k-Armed Bandit Problem |
|---|---|
| One decision is made. | Decisions are made repeatedly. |
| The focus is a single selection. | The agent receives a reward after each selected action. |
| There is no repeated action-reward cycle in the described decision. | After receiving a reward, the agent faces another choice at the next time step. |
| The objective is not described as maximizing expected total reward across repeated selections. | The goal is to maximize expected total reward over a specified time period. |
The problem is named after a slot machine with multiple levers. Treatment selection provides another analogy: a decision-maker repeatedly selects among available options and receives an outcome associated with the selected option.
Common Interpretation Errors
Treating the problem as a one-time decision
A k-Armed Bandit Problem is defined as a repeated decision problem. After a reward is received, another choice occurs at the next time step.
Fix:
Trace the process across multiple action-reward cycles.Assuming an action always produces one fixed numerical reward
The reward is selected from the stationary probability distribution associated with the selected action.
Fix:
Describe the action as determining a distribution, then describe the reward as being selected from that distribution.Optimizing only the next reward
The objective is to maximize expected total reward over a specified time period.
Fix:
Consider how repeated rewards contribute to the total objective.Assuming stationary means the reward never varies
Stationary means the distribution does not change over time; it does not state that every selected reward must be identical.
Fix:
Keep the stable distribution separate from the individual numerical reward selected from it.
Check Your Understanding
Describe one complete cycle in a k-Armed Bandit Problem. Include the agent's action, the action's stationary probability distribution, the numerical reward, and what happens at the next time step.
Hints
- Begin with the agent selecting one of k available actions.
- Explain what the selected action determines.
- End by identifying why the process continues.
Explain why maximizing the reward from one action is not the complete objective in a k-Armed Bandit Problem.
Hints
- The decision process repeats.
- Rewards from multiple time steps contribute to the objective.
- Use the phrase expected total reward.
Key Takeaways
- The k-Armed Bandit Problem consists of repeated choices among k possible actions. Each selected action is associated with a stationary probability distribution, and the numerical reward for that step is selected from the distribution linked to the chosen action. The agent's goal is to maximize expected total reward over a specified time period. The repeated action-reward cycle, rather than a single isolated choice, is the defining structure.
Key Takeaways
- A k-Armed Bandit Problem is a repeated decision problem with k available actions.
- The selected action determines the stationary probability distribution used to select that step's numerical reward.
- Stationary means the action's distribution does not change over time; it does not mean every reward is identical.
- The objective is to maximize expected total reward over a specified time period.
- The repeated action-reward cycle distinguishes the problem from a one-time choice.