Over a million developers have joined DZone.
{{announcement.body}}
{{announcement.title}}

Derivation of the Normal Equation for Linear Regression

DZone's Guide to

Derivation of the Normal Equation for Linear Regression

· Big Data Zone
Free Resource

Find your next Big Data job at DZone Jobs. See jobs focused on Big Data or create your profile and have the employers come to you!

I was going through the Coursera "Machine Learning" course, and in the section on multivariate linear regression something caught my eye. Andrew Ng presented the Normal Equation as an analytical solution to the linear regression problem with a least-squares cost function. He mentioned that in some cases (such as for small feature sets) using it is more effective than applying gradient descent; unfortunately, he left its derivation out.

Here I want to show how the normal equation is derived.

First, some terminology. The following symbols are compatible with the machine learning course, not with the exposition of the normal equation on Wikipedia and other sites - semantically it's all the same, just the symbols are different.

Given the hypothesis function:

<math>

We'd like to minimize the least-squares cost:

<math>

Where <math> is the i-th sample (from a set of m samples) and <math> is the i-th expected result.

To proceed, we'll represent the problem in matrix notation; this is natural, since we essentially have a system of linear equations here. The regression coefficients <math> we're looking for are the vector:

<math>

Each of the m input samples is similarly a column vector with n+1 rows, <math> being 1 for convenience. So we can now rewrite the hypothesis function as:

<math>

When this is summed over all samples, we can dip further into matrix notation. We'll define the "design matrix" X (uppercase X) as a matrix of m rows, in which each row is the i-th sample (the vector <math>). With this, we can rewrite the least-squares cost as following, replacing the explicit sum by matrix multiplication:

<math>

Now, using some matrix transpose identities, we can simplify this a bit. I'll throw the <math> part away since we're going to compare a derivative to zero anyway:

<math><math>

Note that <math> is a vector, and so is y. So when we multiply one by another, it doesn't matter what the order is (as long as the dimensions work out). So we can further simplify:

<math>

Recall that here <math> is our unknown. To find where the above function has a minimum, we will derive by <math> and compare to 0. Deriving by a vector may feel uncomfortable, but there's nothing to worry about. Recall that here we only use matrix notation to conveniently represent a system of linear formulae. So we derive by each component of the vector, and then combine the resulting derivatives into a vector again. The result is:

<math>

Or:

<math>

Now, assuming that the matrix <math> is invertible, we can multiply both sides by <math> and get:

<math>

Which is the normal equation.

 

Find your next Big Data job at DZone Jobs. See jobs focused on Big Data or create your profile and have the employers come to you!

Topics:

Published at DZone with permission of Eli Bendersky, DZone MVB. See the original article here.

Opinions expressed by DZone contributors are their own.

THE DZONE NEWSLETTER

Dev Resources & Solutions Straight to Your Inbox

Thanks for subscribing!

Awesome! Check your inbox to verify your email so you can start receiving the latest in tech news and resources.

X

{{ parent.title || parent.header.title}}

{{ parent.tldr }}

{{ parent.urlSource.name }}