So we have our hypothesis function and we have a way of measuring how well it fits into the data. Now we need to estimate the parameters in the hypothesis function. That’s where gradient descent comes in.
Imagine that we graph our hypothesis function based on its fields θ 0 θ_0 θ0 and θ 1 θ_1 θ1 (actually we are graphing the cost function as a function of the parameter estimates). We are not graphing x and y itself, but the parameter range of our hypothesis function and the cost resulting from selecting a particular set of parameters.
We put
θ
0
θ_0
θ0 on the x axis and
θ
1
θ_1
θ1 on the y axis, with the cost function on the vertical z axis. The points on our graph will be the result of the cost function using our hypothesis with those specific theta parameters. The graph below depicts such a setup.
We will know that we have succeeded when our cost function is at the very bottom of the pits in our graph, i.e. when its value is the minimum. The red arrows show the minimum points in the graph.
The way we do this is by taking the derivative (the tangential line to a function) of our cost function. The slope of the tangent is the derivative at that point and it will give us a direction to move towards. We make steps down the cost function in the direction with the steepest descent. The size of each step is determined by the parameter α, which is called the learning rate.
For example, the distance between each ‘star’ in the graph above represents a step determined by our parameter α. A smaller α would result in a smaller step and a larger α results in a larger step. The direction in which the step is taken is determined by the partial derivative of J ( θ 0 , θ 1 ) J(θ_0,θ_1) J(θ0,θ1)). Depending on where one starts on the graph, one could end up at different points. The image above shows us two different starting points that end up in two different places.
The gradient descent algorithm is:
repeat until convergence:
θ j : = θ j − α ∂ ∂ θ j J ( θ 0 , θ 1 ) θ_j:=θ_j−α\dfrac {∂}{∂θ_j}J(θ_0,θ_1) θj:=θj−α∂θj∂J(θ0,θ1)
where
j=0,1 represents the feature index number.
At each iteration j, one should simultaneously update the parameters
θ
1
,
θ
2
,
.
.
.
,
θ
n
θ_1,θ_2,...,θ_n
θ1,θ2,...,θn. Updating a specific parameter prior to calculating another one on the
j
(
t
h
)
j^{(th)}
j(th) iteration would yield to a wrong implementation.