Multivariate linear regression#

In machine learning, data samples contain multiple features.

We will generalize our previous definitions of linear regression to additional features. How to see this? Previously our data consisted of one column vector \(\boldsymbol{x}\) and its associated column vector of targets \(\boldsymbol{y}\). Additional input features can be visualized in the form of a table:

x1 x2 x3 x4
0 47.219100 -2.505674 2131.326537 -0.169858
1 45.360990 -1.445931 2096.101427 0.052191
2 49.326441 2.227667 2065.543031 0.034362
3 49.557048 -3.035812 2077.252509 -0.135052
4 44.629112 2.422030 1967.909728 0.080960
5 43.848614 2.607565 2041.685750 -0.145993

The data set can be represented as a 2D array with training examples listed as rows. Here each training example has four input features \(x_1\), \(x_2\), \(x_3\) and \(x_4\). Each column lists a given feature.

In particle physics, input features can be the properties of the particles in a collision event. For instance \(x_1\) could be the electron momentum, \(x_2\) the electron’s azimutal angle of its direction in the detector’s coordinate system, \(x_3\) could be another particle’s momentum, \(x_4\) could be a special angle between two particles.

Again let’s have consistent notations:

Terminology and Notation

  • The number of training examples is denoted with \(m\)

  • The number of features is denoted with \(n\)

  • The row index or rank of a training example is in superscript, from 1 to \(m\)

  • The feature or column of a training example is in subscript, from 1 to \(n\)

\(\Rightarrow\) the value of input feature \(j\) of training example in row \(i\) is

\[\begin{equation*} x_j^{(i)} \end{equation*}\]

The training examples form a (\(m \times n\)) matrix \(X\):

\[\begin{equation*} X = \begin{pmatrix} x_1^{(1)} & x_2^{(1)} & \cdots & x_j^{(1)} & \cdots & x_n^{(1)} \\[2ex] x_1^{(2)} & x_2^{(2)} & \cdots & x_j^{(2)} & \cdots & x_n^{(2)} \\ \vdots & \vdots & \ddots & \vdots & & \vdots \\ x_1^{(i)} & x_2^{(i)} & \cdots & x_j^{(i)} & \cdots & x_n^{(i)} \\ \vdots & \vdots & & \vdots & \ddots & \vdots \\ x_1^{(m)} & x_2^{(m)} & \cdots & x_j^{(m)} & \cdots & x_n^{(m)} \\ \end{pmatrix} \end{equation*}\]

Warning

The \(i^\text{th}\) training sample \(\boldsymbol{x^{(i)}}\) is not a scalar but a row vector of \(n\) elements.

Our hypothesis function is generalized to the following for a given training example (row \(i\)):

Definition 9

The hypothesis or mapping function for linear regression with \(n\) features is:

\[\begin{equation*} h_\theta(\boldsymbol{x^{(i)}}) = \theta_0 + \theta_1 x^{(i)}_1 + \theta_2 x^{(i)}_2 + \cdots + \theta_j x^{(i)}_j + \cdots + \theta_n x^{(i)}_n \end{equation*}\]

Warning

There are \(n+1\) parameters to optimize as we need to add the offset parameter \(\theta_0\).

If we set \(x_0^{(i)} = 1\), we can write the mapping function as a sum. For one training example \(x^{(i)}\), i.e. a row in the data set:

(7)#\[ h_\theta(\boldsymbol{x^{(i)}}) = \sum_{j=0}^n \theta_j x^{(i)}_j = x^{(i)} \theta^{\; T}\]

where \(x^{(i)}\) an row vector of \(n+1\) elements, \(x^{(i)} = (x^{(i)}_0, x^{(i)}_1, x^{(i)}_2, \cdots, x^{(i)}_n)\) and \(\theta\) is a row vector too of \(n+1\) elements as well, \(\theta = (\theta_0, \theta_1, \cdots, \theta_n)\). Thus taking the transposed is equivalent to a dot product:

\[\begin{equation*} \boldsymbol{x^{(i)}} \boldsymbol{\theta^{\; T}} = \begin{pmatrix} x^{(i)}_0, x^{(i)}_1, x^{(i)}_2, \cdots, x^{(i)}_n\\\end{pmatrix} \begin{pmatrix} \theta_0 \\\theta_1 \\ \\\vdots \\ \\\theta_n \\\end{pmatrix} = \; \boldsymbol{x^{(i)}} \cdot \boldsymbol{\theta} \end{equation*}\]

Gradient Descent in Multilinear Regression#

We will revisit our algorithm to generalize it to \(\theta_n\) parameters:

Algorithm 2 (Gradient Descent for Multivariate Linear Regression)

Inputs

  • Training data set \(X\) of \(m\) samples with each \(n\) input features, associated with their targets \(y\):

\[\begin{equation*} X = \begin{pmatrix} x_1^{(1)} & x_2^{(1)} & \cdots & x_j^{(1)} & \cdots & x_n^{(1)} \\[2ex] x_1^{(2)} & x_2^{(2)} & \cdots & x_j^{(2)} & \cdots & x_n^{(2)} \\ \vdots & \vdots & \ddots & \vdots & & \vdots \\ x_1^{(i)} & x_2^{(i)} & \cdots & x_j^{(i)} & \cdots & x_n^{(i)} \\ \vdots & \vdots & & \vdots & \ddots & \vdots \\ x_1^{(m)} & x_2^{(m)} & \cdots & x_j^{(m)} & \cdots & x_n^{(m)} \\ \end{pmatrix} \hspace{10ex} y = \begin{pmatrix} y^{(1)} \\[2ex] y^{(2)} \\[2ex] \vdots \\ y^{(i)}\\ \vdots \\[2ex] y^{(m)}\end{pmatrix} \end{equation*}\]

Hyperparameters

  • Learning rate \(\alpha\)

  • Number of epochs \(N\)

Outputs
The optimized values of \(\theta\) parameters, \(\theta_0\), \(\theta_1\), … , \(\theta_n\) minimizing \(J(\theta)\).

  1. Initialization: Set values for all \(\theta\) parameters

  2. Iterate N times or while the exit condition is not met:

    1. Derivatives of the cost function:
      Compute the partial derivatives \(\frac{\partial }{\partial \theta_j} J(\theta)\) for each \(\theta_j\)

    2. Update the parameters:
      Calculate the new parameters according to:

    (8)#\[\begin{split}\begin{equation*} \\ \theta'_j = \theta_j-\alpha \frac{\partial}{\partial \theta_j} J\left(\theta\right) \; \; \; \; \; \; \forall j \in [0..n]\\ \end{equation*}\end{split}\]

             Reassign the new \(\theta\) parameters to prepare for next iteration

(9)#\[\begin{split}\begin{equation*} \theta_j = \theta'_j \; \; \; \; \; \; \forall j \in [0..n]\\ \end{equation*}\end{split}\]

Exit conditions

  • After the maximum number of epochs \(N\) is reached

  • If all derivatives are zero:

\[\begin{equation*} \frac{\partial}{\partial \theta_j} J\left(\theta\right) = 0 \; \; \; \; \; \; \forall j \in [0..n] \end{equation*}\]

The partial derivatives of \(J(\theta)\) for each parameter \(\theta_j\) will be of the form:

(10)#\[\frac{\partial}{\partial \theta_j} J(\theta) = \frac{1}{m} \sum_{i=1}^{m} \left( h_\theta(x^{(i)}) - y^{(i)}\right) \; x_j^{(i)}\]

Feature Scaling & Normalization#

In the example of the previous section, the 3D plot of the cost function with respect to the \(\theta\) parameters was not really bowl-shaped but stretched. This happens when the features are of different ranges. One parameter will have an increment size larger than the other.

Exercise

If let’s say \(x_1\) has values between \(10 000\) and \(100000\) and \(x_2\) between 1 and 10.

  1. Which parameter, \(\theta_1\) or \(\theta_2\) will be updated with a very large or very small step? (independently of the sign). Hint: ask yourself in which direction the contour is going to be stretched.

  2. What are the consequences on the gradient descent algorithm?

What is feature scaling?#

Definition 10

Feature scaling is a data preparation process consisting of harmonizing the values of the input variables taken as features for a machine learning algorithm.

Feature scaling can be implemented in different ways. We will see the two classics: mean normalization and standardization.

Mean normalization#

Definition 11

Mean normalization consists of calculating the mean \(\mu_j\) of all training examples \(x_j^{(1)}, x_j^{(2)}, \cdots, x_j^{(m)}\) of a given feature \(j\) and substract it to each example \(x_j^{(i)}\) of that feature:

\[\begin{equation*} \left(x^{(i)}_j\right)^\text{ norm} = x^{(i)}_j - \mu_j \; \; \; \; \; \; \forall i \in [1..m] \end{equation*}\]

It is usual to divide by the range of the features:

\[\begin{equation*} \left(x^{(i)}_j\right)^\text{ norm} = \frac{x^{(i)}_j - \mu_j}{x_j^\max - x_j^\min} \; \; \; \; \; \; \forall i \in [1..m] \end{equation*}\]

Consequence: the mean of the new normalized sample collection for that feature - think of it as an extra column in the matrix \(X\) - will be zero. The data distribute the same as before, the values are just representing the distance to the mean.

Standardization#

Definition 12

The mean normalization procedure using as denominator the standard deviation \(\sigma_j\) all the samples for this feature is called Standardization

\[\begin{equation*} \widehat{x}^{(i)}_j = \frac{x^{(i)}_j - \mu_j}{\sigma_j} \; \; \; \; \; \; \forall i \in [1..m] \end{equation*}\]

Consequence: the mean and standard deviation of the new normalized collection of feature \(j\) will be zero and one respectively.

When is feature scaling relevant? When is it not?
It all depends on how the data look like. This is why it is important before starting the machinery of (fancy) learning techniques to inspect the data, plot several distributions if possible and dedicate time for the preparedness steps: data cleaning, scaling etc. This is essential for the success of the fitting algorithm.