Concepts / Expected Reward and Decision Making

Expected Reward and Decision Making

A k-Armed Bandit Problem is a repeated decision problem with k possible actions.

  • Machine Learning

The Decision Repeats

Imagine facing the same kind of decision again and again. On every round, you choose one option from k available options, and that choice produces a numerical reward. The central challenge is not simply to make one good choice. It is to keep choosing over a period of time so that the total reward collected is as large as possible.

A k-Armed Bandit Problem is a repeated decision problem with k possible actions.

One Choice or Many Rounds

producesproducesfollowed bycontributes toOne choiceone selectionRound 1select an actionOutcomedecision endsRewardnumerical resultNext roundselect againTotal rewardaccumulated over time
What makes a k-Armed Bandit Problem different from selecting an option only once?

In the one-time case, there is no later decision in the process described here. In the k-Armed Bandit Problem, every reward is followed by another opportunity to choose. Each action selection therefore has two roles: it produces a reward for the current round, and it is part of a longer sequence whose rewards contribute to the total over the specified time period.

Actions Select Reward Distributions

A single cycle has a clear structure. First, the agent selects one action from the k available actions. The selected action determines which stationary probability distribution is used to choose the numerical reward. After the reward is received, the agent faces another choice at the next time step.

A stationary probability distribution is the action-associated probability distribution that does not change over time. Different actions can have different stationary probability distributions.

determinesselectsSelected actionone of k actionsStationarydistributionlinked to the actionNumerical rewardselected from thedistribution
How does choosing one action determine the stationary probability distribution from which the numerical reward is sampled?

Tracing Several Rounds

A generated three-round trace

Suppose a bandit problem has three available actions: Action A, Action B, and Action C. Consider a generated sequence in which the agent selects one action on each round and receives one numerical reward.

Round 1: The agent selects Action B. Action B determines the stationary probability distribution used to select the numerical reward for this round.

Round 2: The agent faces another choice and selects Action A. Action A determines the stationary probability distribution used for the new numerical reward.

Round 3: The agent selects Action B again. The reward is again selected from the stationary probability distribution associated with Action B.

The trace contains three action selections and three numerical rewards. The rewards from the rounds contribute to the total reward for the time period. The example does not assign particular distributions or reward values; its purpose is to show the repeated action-to-distribution-to-reward structure.

producesfollowed byproducesfollowed bycontributes toSelect actionone of k optionsReward 1numerical rewardSelect actionnext time stepReward 2numerical rewardFurther selectionsrepeated processTotal rewardover the time period
What happens across repeated rounds when an agent selects one of k actions and receives a reward each time?

The trace shows why the problem is about decision making over time. A reward from one round does not end the process. The agent must make another selection, receive another reward, and continue contributing to the total reward. Repeating the process also supplies experience about the available choices while rewards continue to accumulate.

Accumulating the Objective

The agent's objective is to maximize expected total reward over a specified time period. This objective evaluates the sequence of decisions together rather than judging only one action in isolation. A choice is valuable in the context of the repeated process because its reward contributes to the total collected during the period.

first roundnext roundnext roundcontributeStartdecision periodReward 1first selectionReward 2second selectionReward 3later selectionExpected total rewardobjective over the period
How do rewards from multiple decisions accumulate into expected total reward over a specified time horizon?

What do you think happens?

In the generated three-round trace, should the agent judge the process from only the reward on Round 1?

  • Yes, because the first decision determines the whole result
  • No, because the objective concerns expected total reward across the specified period
Reveal answer

Answer: No, because the objective concerns expected total reward across the specified period.

The bandit problem is a repeated decision problem. Each round supplies a reward, and the rewards from the repeated selections contribute to the total reward for the period.

Common Misunderstandings

  • Treating each action as if it always produces one fixed reward.

    The reward is selected from the probability distribution associated with the selected action.

    Fix: Remember that an action determines a stationary probability distribution, and the numerical reward is selected from that distribution.

  • Treating a bandit problem as a one-time choice.

    The k-Armed Bandit Problem requires repeated choices, with another decision at the next time step after a reward is received.

    Fix: Track the sequence of actions and rewards across the specified time period.

  • Optimizing only the reward from the current round.

    The stated objective is to maximize expected total reward over a time period.

    Fix: Evaluate decisions by how they contribute to the expected total reward across repeated rounds.

  • Assuming every action has the same reward distribution.

    Different actions can have different stationary probability distributions.

    Fix: Keep the selected action linked to its own stationary probability distribution.

The Lever Analogy

The problem is named after a slot machine with multiple levers. Each lever represents an available action, and repeatedly selecting levers produces rewards over time. The phrase best levers refers to the actions that are most valuable for the repeated-decision objective. Treatment selection provides another analogy: a decision-maker repeatedly selects among available options and considers the rewards associated with those selections.

Practice the Decision Cycle

EASY

Describe one complete cycle of a k-Armed Bandit Problem in the correct order. Your response should include the action selection, the action's stationary probability distribution, the numerical reward, and the next decision.

Hints
  • Begin with the agent selecting one of the k available actions.
  • Explain what the selected action determines.
  • End by stating what happens after the reward is received.
MEDIUM

Explain why the following is not a complete description of the objective: Select the action that gives the largest reward once.

Hints
  • Focus on the words repeated decision problem.
  • Include the specified time period in your explanation.
  • Mention expected total reward rather than one isolated reward.

Key Takeaways

  1. A k-Armed Bandit Problem presents k possible actions and requires the agent to choose repeatedly.
  2. The selected action determines the stationary probability distribution from which a numerical reward is selected.
  3. A stationary distribution does not change over time, although different actions can have different stationary distributions.
  4. The agent seeks to maximize expected total reward over a specified time period, not merely the reward from one choice.
  5. The repeated action-and-reward cycle distinguishes the bandit problem from a one-time decision.

Key Takeaways

  • The k-Armed Bandit Problem is a repeated decision problem with k possible actions.
  • Each selected action is linked to a stationary probability distribution that produces a numerical reward.
  • The agent makes another choice after each reward, creating a sequence of decisions and rewards.
  • The objective is to maximize expected total reward across a specified time period.
  • This repeated process is different from making one choice once.