Policy Gradient and Actor–Critic

English translation

Graduate course · Research application activities

Examine direct updates to a stochastic policy, then investigate how a critic changes learning over long episodes.

Before practice

01 What does it mean to update a stochastic policy directly?

Description

REINFORCE weights the log probability of the selected action by the return. Subtracting a baseline does not change the reward function itself.

02 Does an incorrect critic also affect the actor?

Description

A2C uses a critic to estimate future values, computes advantages, and updates the actor. Rollout length changes the balance between actual rewards and bootstrapping.

Execution conditions

If the required package is missing, run %pip install gymnasium==1.3.0 stable-baselines3==2.9.2 in a Colab or Jupyter code cell.

Applications

Throughput and cost of equipment and communication resources

An average of 2 items arrive at the synthetic queue each time step. Increasing service capacity reduces queue length but increases energy cost.

Observation·State
Queue length q=0q=0…20. Observation q/20q/20.
Action
Choose a processing capacity of 0, 2, or 4.
Reward·Constraint
−(next queue length + 0.2×capacity² + 10×dropped items). Time limit at 100 steps.
Comparison baseline
Constant capacity of 2, and a threshold policy based on queue length.
Metrics to check
Evaluation return, mean queue length, mean capacity², and dropped-item count.

Mobile robot paths and safety

An application design case choosing either grid paths or continuous velocity control.

Observation·State
Position, target-relative position, and nearby obstacle observations.
Action
Grid direction selection or bounded linear and angular velocities.
Reward·Constraint
Separate arrival, collision, and movement costs. Distinguish collisions from timeouts.
Comparison baseline
Shortest path and rule-based avoidance.
Metrics to check
Success rate, collision rate, travel distance, and time. Evaluate on unseen maps.

Practice

Preview practice notebookDownload practice notebook

1. Update a policy from one episode

Starting conditions

Queue, softmax policy 1→16→3, γ=.95\gamma=.95, Adam .005, and 30 episodes. Use a simple baseline that subtracts the mean return for each episode.

Procedure

  1. Calculate return-to-go by hand for the fixed rewards [-1,-2,-3], then compare with the loop output.
  2. Run the provided REINFORCE implementation and record training returns separately from deterministic evaluation returns.
  3. Explain the difference between stochastic actions during collection and argmax selection during evaluation.

Results to retain

Items to record and points to check
Time point
reward
Return-to-go
Baseline
Advantage

Points to check

  • Is the log probability that of the selected action?
  • Can you explain how the probability of an action with negative advantage changes?

2. Rule-based control and A2C

Starting conditions

Discrete actions in Queue

Procedure

  1. Pass the two preceding rule-based policies policies to evaluate and record results for the common evaluation seeds. Find where QueueGym transforms queue length to q/20q/20 and reward to 1/201/20 of its original value.
  2. Run fit_sb3("A2C",seed=0,steps=2048). Select the learned policy with model.predict(...,deterministic=True) and evaluate it on the same seeds using evaluate in the notebook.
  3. Record episode return, mean queue length, mean squared capacity, and total dropped items for the rule-based and learned policies in a table with the same columns. Identify improvements and regressions, and state that your judgment is based on this training budget and evaluation seed.

Results to retain

Items to record and points to check
Policy
Training budget
Evaluation conditions
Main error and delay
Input cost
Constraint violations

Points to check

  • Did you distinguish training return from evaluation metrics in their original units?
  • Did you first validate with a synthetic model instead of real equipment?

3. Apply this to your research problem

Starting conditions

Based on the preceding experiments, choose Queue, Motor, or one of your research problems and design an experiment for real application.

Procedure

  1. Define state, observation, action, reward, termination, and time limit in one table. Record each variable's units and the information you can actually obtain.
  2. Write 3 expected transitions numerically and check whether rewards for good and bad actions match your intent.
  3. Choose one rule-based comparison baseline, one evaluation condition, and one failure case of greatest concern. Also list the required measurements or simulator.

Results to retain

You may continue with the research problem and design table selected earlier. Revisit existing entries and transition examples using the concepts learned in this activity, and update changed entries or reasons for retaining them.

Items to record and points to check
Item
Concrete definition and units
Observability and verification method

Points to check

  • Did you exclude future information and evaluation labels from observations?
  • Did you define domain-specific metrics rather than judging the research contribution solely by training return?

Official examples and data