introduction_to_svm.markdown 11.2 KB
Newer Older
1 2 3 4 5 6 7 8
Introduction to Support Vector Machines {#tutorial_introduction_to_svm}
=======================================

Goal
----

In this tutorial you will learn how to:

9 10
-   Use the OpenCV functions @ref cv::ml::SVM::train to build a classifier based on SVMs and @ref
    cv::ml::SVM::predict to test its performance.
11 12 13 14 15 16 17 18 19 20 21 22 23

What is a SVM?
--------------

A Support Vector Machine (SVM) is a discriminative classifier formally defined by a separating
hyperplane. In other words, given labeled training data (*supervised learning*), the algorithm
outputs an optimal hyperplane which categorizes new examples.

In which sense is the hyperplane obtained optimal? Let's consider the following simple problem:

For a linearly separable set of 2D-points which belong to one of two classes, find a separating
straight line.

24
![](images/separating-lines.png)
25 26 27 28 29

@note In this example we deal with lines and points in the Cartesian plane instead of hyperplanes
and vectors in a high dimensional space. This is a simplification of the problem.It is important to
understand that this is done only because our intuition is better built from examples that are easy
to imagine. However, the same concepts apply to tasks where the examples to classify lie in a space
30 31
whose dimension is higher than two.

32 33 34 35 36
In the above picture you can see that there exists multiple lines that offer a solution to the
problem. Is any of them better than the others? We can intuitively define a criterion to estimate
the worth of the lines: <em> A line is bad if it passes too close to the points because it will be
noise sensitive and it will not generalize correctly. </em> Therefore, our goal should be to find
the line passing as far as possible from all points.
37 38 39 40 41 42

Then, the operation of the SVM algorithm is based on finding the hyperplane that gives the largest
minimum distance to the training examples. Twice, this distance receives the important name of
**margin** within SVM's theory. Therefore, the optimal separating hyperplane *maximizes* the margin
of the training data.

43
![](images/optimal-hyperplane.png)
44 45 46 47 48 49

How is the optimal hyperplane computed?
---------------------------------------

Let's introduce the notation used to define formally a hyperplane:

50
\f[f(x) = \beta_{0} + \beta^{T} x,\f]
51 52 53

where \f$\beta\f$ is known as the *weight vector* and \f$\beta_{0}\f$ as the *bias*.

54
@sa A more in depth description of this and hyperplanes you can find in the section 4.5 (*Separating
55
Hyperplanes*) of the book: *Elements of Statistical Learning* by T. Hastie, R. Tibshirani and J. H.
56
Friedman (@cite HTF01).
57 58

The optimal hyperplane can be represented in an infinite number of different ways by
59 60 61
scaling of \f$\beta\f$ and \f$\beta_{0}\f$. As a matter of convention, among all the possible
representations of the hyperplane, the one chosen is

62
\f[|\beta_{0} + \beta^{T} x| = 1\f]
63 64 65 66 67 68 69 70 71 72 73 74 75

where \f$x\f$ symbolizes the training examples closest to the hyperplane. In general, the training
examples that are closest to the hyperplane are called **support vectors**. This representation is
known as the **canonical hyperplane**.

Now, we use the result of geometry that gives the distance between a point \f$x\f$ and a hyperplane
\f$(\beta, \beta_{0})\f$:

\f[\mathrm{distance} = \frac{|\beta_{0} + \beta^{T} x|}{||\beta||}.\f]

In particular, for the canonical hyperplane, the numerator is equal to one and the distance to the
support vectors is

76
\f[\mathrm{distance}_{\text{ support vectors}} = \frac{|\beta_{0} + \beta^{T} x|}{||\beta||} = \frac{1}{||\beta||}.\f]
77 78 79 80

Recall that the margin introduced in the previous section, here denoted as \f$M\f$, is twice the
distance to the closest examples:

81
\f[M = \frac{2}{||\beta||}\f]
82 83 84 85 86

Finally, the problem of maximizing \f$M\f$ is equivalent to the problem of minimizing a function
\f$L(\beta)\f$ subject to some constraints. The constraints model the requirement for the hyperplane to
classify correctly all the training examples \f$x_{i}\f$. Formally,

87
\f[\min_{\beta, \beta_{0}} L(\beta) = \frac{1}{2}||\beta||^{2} \text{ subject to } y_{i}(\beta^{T} x_{i} + \beta_{0}) \geq 1 \text{ } \forall i,\f]
88 89 90 91 92 93 94 95 96

where \f$y_{i}\f$ represents each of the labels of the training examples.

This is a problem of Lagrangian optimization that can be solved using Lagrange multipliers to obtain
the weight vector \f$\beta\f$ and the bias \f$\beta_{0}\f$ of the optimal hyperplane.

Source Code
-----------

97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
@add_toggle_cpp
-   **Downloadable code**: Click
    [here](https://github.com/opencv/opencv/tree/3.4/samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp)

-   **Code at glance:**
    @include samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp
@end_toggle

@add_toggle_java
-   **Downloadable code**: Click
    [here](https://github.com/opencv/opencv/tree/3.4/samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java)

-   **Code at glance:**
    @include samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java
@end_toggle

@add_toggle_python
-   **Downloadable code**: Click
    [here](https://github.com/opencv/opencv/tree/3.4/samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py)

-   **Code at glance:**
    @include samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py
@end_toggle
120 121 122 123

Explanation
-----------

124 125 126 127 128 129 130 131 132 133 134 135
-   **Set up the training data**

The training data of this exercise is formed by a set of labeled 2D-points that belong to one of
two different classes; one of the classes consists of one point and the other of three points.

@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp setup1
@end_toggle

@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java setup1
@end_toggle
136

137 138 139
@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py setup1
@end_toggle
140

141 142 143
The function @ref cv::ml::SVM::train that will be used afterwards requires the training data to be
stored as @ref cv::Mat objects of floats. Therefore, we create these objects from the arrays
defined above:
144

145 146 147
@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp setup2
@end_toggle
148

149 150 151
@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java setup2
@end_toggle
152

153 154 155 156 157
@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py setup1
@end_toggle

-   **Set up SVM's parameters**
158 159 160 161 162 163

    In this tutorial we have introduced the theory of SVMs in the most simple case, when the
    training examples are spread into two classes that are linearly separable. However, SVMs can be
    used in a wide variety of problems (e.g. problems with non-linearly separable data, a SVM using
    a kernel function to raise the dimensionality of the examples, etc). As a consequence of this,
    we have to define some parameters before training the SVM. These parameters are stored in an
164 165
    object of the class @ref cv::ml::SVM.

166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp init
@end_toggle

@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java init
@end_toggle

@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py init
@end_toggle

Here:
-   *Type of SVM*. We choose here the type @ref cv::ml::SVM::C_SVC "C_SVC" that can be used for
    n-class classification (n \f$\geq\f$ 2). The important feature of this type is that it deals
    with imperfect separation of classes (i.e. when the training data is non-linearly separable).
    This feature is not important here since the data is linearly separable and we chose this SVM
    type only for being the most commonly used.

-   *Type of SVM kernel*. We have not talked about kernel functions since they are not
    interesting for the training data we are dealing with. Nevertheless, let's explain briefly now
    the main idea behind a kernel function. It is a mapping done to the training data to improve
    its resemblance to a linearly separable set of data. This mapping consists of increasing the
    dimensionality of the data and is done efficiently using a kernel function. We choose here the
    type @ref cv::ml::SVM::LINEAR "LINEAR" which means that no mapping is done. This parameter is
    defined using cv::ml::SVM::setKernel.

-   *Termination criteria of the algorithm*. The SVM training procedure is implemented solving a
    constrained quadratic optimization problem in an **iterative** fashion. Here we specify a
    maximum number of iterations and a tolerance error so we allow the algorithm to finish in
    less number of steps even if the optimal hyperplane has not been computed yet. This
    parameter is defined in a structure @ref cv::TermCriteria .

-   **Train the SVM**
200
    We call the method @ref cv::ml::SVM::train to build the SVM model.
201

202 203 204 205 206 207 208
@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp train
@end_toggle

@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java train
@end_toggle
209

210 211 212 213 214
@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py train
@end_toggle

-   **Regions classified by the SVM**
215

216 217 218 219 220
    The method @ref cv::ml::SVM::predict is used to classify an input sample using a trained SVM. In
    this example we have used this method in order to color the space depending on the prediction done
    by the SVM. In other words, an image is traversed interpreting its pixels as points of the
    Cartesian plane. Each of the points is colored depending on the class predicted by the SVM; in
    green if it is the class with label 1 and in blue if it is the class with label -1.
221

222 223 224 225 226 227 228
@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp show
@end_toggle

@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java show
@end_toggle
229

230 231 232 233 234
@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py show
@end_toggle

-   **Support vectors**
235

236 237 238
    We use here a couple of methods to obtain information about the support vectors.
    The method @ref cv::ml::SVM::getSupportVectors obtain all of the support
    vectors. We have used this methods here to find the training examples that are
239
    support vectors and highlight them.
240

241 242 243 244 245 246 247 248 249 250 251
@add_toggle_cpp
@snippet samples/cpp/tutorial_code/ml/introduction_to_svm/introduction_to_svm.cpp show_vectors
@end_toggle

@add_toggle_java
@snippet samples/java/tutorial_code/ml/introduction_to_svm/IntroductionToSVMDemo.java show_vectors
@end_toggle

@add_toggle_python
@snippet samples/python/tutorial_code/ml/introduction_to_svm/introduction_to_svm.py show_vectors
@end_toggle
252

253 254 255 256 257 258 259 260 261 262
Results
-------

-   The code opens an image and shows the training examples of both classes. The points of one class
    are represented with white circles and black ones are used for the other class.
-   The SVM is trained and used to classify all the pixels of the image. This results in a division
    of the image in a blue region and a green region. The boundary between both regions is the
    optimal separating hyperplane.
-   Finally the support vectors are shown using gray rings around the training examples.

263
![](images/svm_intro_result.png)