本文主要是介绍斯坦福CS231n assignment1:SVM图像分类原理及实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
斯坦福CS231n assignment1:SVM图像分类原理及实现
- SVM模型原理
- SVM的一种直观解释
- 损失函数
- 损失函数加入正则化项
- 梯度下降和梯度检验
- 图像预处理
- 小批量数据梯度下降(Mini-batch gradient descent)
- 代码实现
本文Github代码
斯坦福CS231n课程讲解了实现图像分类的方法,从传统的KNN,SVM,到CNN,LSTM模型,讲解的非常专业精准。同时该课程提供了相应的习题来检验和巩固讲授的知识,如果能按部就班的完成,对神经网络将会有深刻的体会和理解。本文将结合代码实现讲解其中的SVM方法实现图像分类的原理和方法,以及需要注意的知识细节。
SVM模型原理
SVM通过平面将空间中的实例划分到不同的类别,从而实现分类。这里的空间包括二维空间,三维空间,一直到高维空间,具体的维数等于实例的特征数量,如果我们待分类的图片是32*32*3(长宽分别是32个像素,RGB3个颜色通道)维的,那么图片所处的空间就是3072维的空间。在这个高维空间,我们通过由权重向量W和偏置项b确定的一个(实际上是一组)超平面来将图片进行分类。为了可视化,我们将多维空间压缩到二维空间,那么就是下面的图像:
这里每一个平面都将整个高维空间划分成两部分,平面的一侧是某一类图片,另一侧是这个类别之外的其他图片。比如红色的平面一侧是汽车这个类别,另一侧是非汽车类别。每一个类别都对应一个平面,这些平面互相之间不存在关联,利用SVM模型进行分类的目的就是确定这样一组平面,使得同一类尽可能划分在该类对应的平面的一侧,其他类尽可能在另一侧,而且两种类别离平面的距离越大越好(平面尽可能把两类分的更开),这是SVM模型的思路。
所有这些类别对应的平面通过下面的矩阵唯一确定:
其中改变W可以使平面旋转,而改变b使平面平移。如果b为0,此时W*0=0,那么平面会经过原点。
SVM的一种直观解释
SVM模型用于图像分类可以看做给每一种图像的类别生成一个图像模板,然后拿待分类的图像和这个图像模板做内积,计算他们的相似度,相似度最高的类别就是分类类别。根据这个思想,生成的权重向量可视化如下:
可以看出,这些图像模板比较能够代表某种类别的共性,比如car类别是一辆红色的车的形象,而horse类型是左右两匹马的形象,这些是集合了所有训练样本得出的模板。从这个角度,SVM可以看做KNN模型的一种简化,KNN模型对一张图片分类时需要和所有训练样本做比较,而SVM只需要和抽象出来的每个类别下的一个图像模板做比较即可,显然更高效。
损失函数
SVM模型有多种不同的实现,区别主要体现在损失函数的定义上,可以根据实现分为:
- 经典SVM
- Structured SVM
其中经典SVM模型核心思路是找一个超平面将不同类别分开,同时使得离超平面最近的点的距离最大,这样能保证即使是最难区分的点,也有较大的确信度将它
这篇关于斯坦福CS231n assignment1:SVM图像分类原理及实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!