By R. Gayathri
Implementing state-action-reward-state-action Algorithm by Reinforcement learning technique in Python.
A Machine can be trained to make a sequence of decisions to achieve a definite goal. In artificial intelligence, this is usually done by a giving a reward or penalty for every action taken by the machine. This method is called reinforcement learning and it belongs to the category of unsupervised learning.
To understand SARSA algorithm which is based on Markov Decision process, we need to understand the concept of Temporal Difference. SARSA is related to Q-learning which are both TD based, but SARSA is on-policy (agent learns value function from action of current policy) and Q-learning is off-policy (agent learns value function from action of another policy)
The main advantage of SARSA is that, the reward/penalty of an action can be updated after each state, and need not wait till the end of the episode.
ALGORITHM
A Markov Decision Process (MDP) model contains:
• S = set of possible states.
• A = set of actions.
• r (S, A) = reward value
• A policy the solution of Markov Decision Process.
The aim is to achieve at the best possible solution for a problem
TD(0) Update Rule:
This involves taking an action at the starting state, move to next state and update the value of the previous state.
S1 -> a -> S2 to -> r
V(S) => V(S1) +α [ r + γ * V(S2) – V(S1) ]
Where V(S) is the new value of state, V(S1) is the current value of state and V(S2) is the estimated value of the new state.
α is the learning rate, γ is the discount factor and r is the reward
[ r + γ * V(S2) – V(S1) ] is the TD error
The aim is to move the current state value towards the sum of reward value and γ times the value of new state ( r + γ * V(S2) ) and this done at the rate of α.
SARSA:
This aims to achieve an optimal Q value and not just the optimal value of the state.
α, which is the learning rate needs to be initialized, which determines how quickly we make changes to the Q function.
Q (Si, Ai)) = Q (Si, Ai)) + α(ri+1 + γ * Q (Si+1, Ai+1) – Q (Si, Ai))
This procedure is used for all the episodes taking one episode at a time.
IMPLEMENTATION:
Chosen Environment:
The agent controls the movement of a character in a grid world. Some tiles of the grid are walkable, and others lead to the agent falling into the water. Additionally, the movement direction of the agent is uncertain and only partially depends on the chosen direction. The agent is rewarded for finding a walkable path to a goal tile.
Code:
Importing all the needed libraries.
import numpy as np import gym
numpy is used for dealing with arrays gym library is used for developing reinforcement algorithms
env = gym.make('FrozenLake-v0') epsilon = 0.9 total_episodes = 100 max_steps = 10 alpha = 0.85 gamma = 0.95 Q = np.zeros((env.observation_space.n, env.action_space.n))
The FrozenLake-v0 environment is set and loaded into a variable. The necessary variables are initialized. The number of episodes is 100 and the maximum number of steps per episode is 10
def choose_action(state): action=0 if np.random.uniform(0, 1) < epsilon: action = env.action_space.sample() else: action = np.argmax(Q[state, :]) return action def update(state, state2, reward, action, action2): predict = Q[state, action] target = reward + gamma * Q[state2, action2] Q[state, action] = Q[state, action] + alpha * (target - predict)
The function to choose the net action and the function to learn the Q value is defined.
reward=0 for episode in range(total_episodes): t = 0 state1 = env.reset() action1 = choose_action(state1) while t < max_steps: env.render() state2, reward, done, info = env.step(action1) action2 = choose_action(state2) update(state1, state2, reward, action1, action2) state1 = state2 action1 = action2 t += 1 reward += 1 if done: break
The machine is trained using SARSA algorithm by iterating through all the episodes.
The red mark is the current posion of the agent and the direction in the brackets is the direction of movement of the agent.
print ("Performace : ", reward/total_episodes) print(Q)
The performance of the algorithm is calculated by dividing the final reward by the total number of episodes.
The updated Q-table and the performance is printed
CONCLUSION:
Reinforcement learning is a very important concept that has great potential and plays a critical role in the technological industry. This can be used to arrive at solutions for complex problems with proper training and testing. Thus, SARSA Algorithm can be used to serve this purpose effeciently and successfully
Submitted by R. Gayathri (Gayathri)
Download packets of source code on Coders Packet
Comments