Policy Gradients (REINFORCE)

Reinforcement Learning

A glimpse of Policy Gradients (REINFORCE)

An agent has two buttons. Pressing left ends the attempt with reward 0; pressing right ends it with reward 2. It starts by choosing each button with probability 0.5.

We want to change those probabilities using the reward from a button it actually pressed. This is the simplest setting for a policy gradient: improve the action-selection rule directly.

Continue learning

Pro unlocks guided learning, complete notes, practical walkthroughs, exercises, and every AI Research lesson.