SARSA Agent
The SARSA algorithm is an on-policy reinforcement learning method for environments with a discrete action space. A SARSA agent trains a value function based critic to estimate the expected discounted cumulative long-term reward of the current policy. Therefore, SARSA is the on-policy version of Q-learning. For more information on the different types of reinforcement learning agents, see Reinforcement Learning Agents.
In Reinforcement Learning Toolbox™, a SARSA agent is implemented by an rlSARSAAgent
object.
Note
SARSA agents do not support recurrent networks.
SARSA agents can be trained in environments with the following observation and action spaces.
Observation Space | Action Space |
---|---|
Continuous or discrete | Discrete |
SARSA agents use the following critic.
Critic | Actor |
---|---|
Q-value function critic
Q(S,A), which you create
using | SARSA agents do not use an actor. |
During training, the agent explores the action space using epsilon-greedy exploration. During each control interval the agent selects a random action with probability ϵ, otherwise it selects the action for which the action-value function is greatest with probability 1–ϵ.
Critic Function Approximator
To estimate the value of the current policy, a SARSA agent uses a critic. The critic is a function approximator object that implements the parametrized action-value function Q(S,A;ϕ), using parameters ϕ. For a given observation S and action A, the critic stores the corresponding estimate of the expected discounted cumulative long-term reward given the current policy. During training, the critic tunes its parameters to improve its estimation.
For critics that use table-based value functions, the parameters in ϕ are the actual Q(S,A) values in the table.
For more information on creating critics for value function approximation, see Create Policies and Value Functions.
During training, the agent tunes the parameter values in ϕ. After training, the parameters remain at their tuned value and the trained value function approximator is stored in critic Q(S,A).
Agent Creation
To create a SARSA agent:
Create a critic using an
rlQValueFunction
orrlVectorQValueFunction
object.Specify agent options using an
rlSARSAAgentOptions
object. Alternatively, you can create the agent first (step 3) and then, using dot notation, access its option object and modify the options.Create the agent using an
rlSARSAAgent
object.
Training Algorithm
SARSA agents use the following training algorithm. To configure the training algorithm,
specify options using an rlSARSAAgentOptions
object.
Initialize the critic Q(S,A;ϕ) with random parameter values in ϕ.
For each training episode:
Get the initial observation S from the environment.
For the current observation S, select a random action A with probability ϵ. Otherwise, select the action for which the critic value function is greatest.
To specify ϵ and its decay rate, use the
EpsilonGreedyExploration
option.Repeat the following for each step of the episode until S is a terminal state:
Execute action A0. Observe the reward R and next observation S'.
For the current observation S', select a random action A' with probability ϵ. Otherwise, select the action for which the critic value function is greatest.
If S' is a terminal state, set the value function target y to R. Otherwise, set it to
To set the discount factor γ, use the
DiscountFactor
option.Compute the difference ΔQ between the value function target and the current Q(S,A;ϕ) value.
Update the critic using the learning rate α. Specify the learning rate when you create the critic by setting the
LearnRate
option in therlCriticOptimizerOptions
property within the agent options object.For table-based critics, update the corresponding Q(S,A) value in the table.
For all other types of critics, compute the gradients Δϕ of the loss function with respect to the parameters ϕ. Then, update the parameters based on the computed gradients. In this case, the loss function is the square of ΔQ.
Set the observation S to S'.
Set the action A to A'.
References
[1] Sutton, Richard S., and Andrew G. Barto. Reinforcement Learning: An Introduction. Second edition. Adaptive Computation and Machine Learning. Cambridge, Mass: The MIT Press, 2018.