1
$\begingroup$

Specifically, according to this post:

How is the policy gradient calculated in REINFORCE

the function I need to minimise is:

$−Gt \log \pi(At|St,θt)$

where $Gt$ is the discounted reward, and $\pi$ is the policy which outputs a probability distribution of actions given the state and network parameters.

Since the negative log of probability is always positive or zero, this expression also achieves a minimum of zero if $Gt$ is zero i.e. no reward, assuming the reward of the game is always non-negative.

This seems to be the opposite of what we want to do which is to find a policy network parameters that maximises the discounted reward $Gt$. What am I misunderstanding here?

$\endgroup$
1
  • $\begingroup$ Can you please put your specific question in the title? Clearly, "what am I misunderstanding here?" is a bit vague and it would be great if you could formulate a clearer question in the title. $\endgroup$
    – nbro
    Commented Jun 6, 2023 at 10:05

1 Answer 1

3
$\begingroup$

The expression you are looking at in the link, $G_t \nabla_{\theta} \text{ln}(\pi(A_t|S_t,\theta_t))$, is not the cost function, but the estimated gradient of the cost function. You cannot simply remove the $\nabla_{\theta}$ to see the cost function, because the state distribution that weights components of the cost function (i.e. how often you will observe each trajectory) depends critically on $\theta$.

The real cost function would be the negative of expected return from the start (or stable for non-episodic) state distribution, i.e. $-J(\theta)$. This does not have a simple expansion, because it must be summed up over the distribution of states and the policy, both of which depend on $\theta$.

Usually in reinforcement learning, the optimisation is not stated as a cost function to minimise, but as an expression to maximise. We want to maximise $J(\theta)$ which summarises our expected return. Hence a factor of the gradient is added to $\theta$ during update steps.

From comments:

if the policy stumbles upon a set of parameters which makes $G_t=0$, doesn't that make the estimated gradient zero hence causing the policy to stop improving

Partially correct.

A return of zero from any single trajectory will make the updates from all actions also be zero, regardless of whether or not a return of zero is a good result. This motivates REINFORCE with baseline or Actor-Critic approaches that don't use final return $G_t$, but advantage, or a similar baselined value instead. These methods are less sensitive to absolute value of returns.

For the agent to get stuck with all zero gradients though, all possible trajectories would have to give a return of zero. This would be a problem for any experience-based system, and there is no reinforcement learning system that can find better actions if the feedback is all the same.

In fact this is also a problem regardless of what the fixed value is - if all experience returns the same end return, then nothing can be learned to differentiate actions. Simple REINFORCE would actually generate gradients (because $G_t$ multiplier will be non-zero), and it might look like it is learning something because $\theta$ will change with gradient-based steps. But actually it will end up choosing a policy pseudo-randomly, because as far as the agent can tell all policies are the same.

One thing to bear in mind is that the gradient estimates generated in REINFORCE (and policy gradients in general) are biased. You don't know the true gradient, and there is no simple way to remove the bias for a single sample - you have to rely on the gradient steps across very many samples to even out this bias. In fact you can add almost any offset to the multiplier $G_t$, provided it doesn't depend immediately on the policy, and you still have a valid gradient estimate for update steps. This is why advantage-based systems work, they choose an offset based on state value which conveniently reduces variance on each sample.

$\endgroup$
3
  • $\begingroup$ Sorry, I meant loss, not cost function. I didn't remove ∇θ, it was stated in the link that the function we want to minimise is −Gt ln π(At|St,θt). Example from other sources such as github.com/Finspire13/pytorch-policy-gradient-example/blob/… also seems to indicate this is indeed the function we want to minimize (correct me if I am wrong). Anyway, my question still remains - if the policy stumbles upon a set of parameters which makes Gt=0, doesn't that make the estimated gradient zero hence causing the policy to stop improving? $\endgroup$
    – Jason L
    Commented Jun 5, 2023 at 13:17
  • $\begingroup$ @JasonL That doesn't make much difference - a cost function is conventially the sum of some loss functions. $J(\theta)$ is the cost function, and components of it weighted by state distribution are the loss functions. The quote from the link is incorrect. I will answer your comment in the body of the question $\endgroup$ Commented Jun 5, 2023 at 13:35
  • $\begingroup$ Thanks for your input @Neil. I don't think the quote from the link is incorrect. I've looked at a number of example implementations for vanilla policy gradient and −Gt log π(At|St,θt) is indeed the loss function being minimized. I think my question mainly originates from observing the loss value as training progresses - that I noticed sometimes the agent plays very badly achieving a low score, the loss decreases. Then I realised it could be term Gt in the loss that contributes to this. So it would appear that the agent is being 'rewarded' for bad play. $\endgroup$
    – Jason L
    Commented Jun 6, 2023 at 13:08

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .