微波EDA网,见证研发工程师的成长!
首页 > 硬件设计 > 硬件工程师文库 > 基于机器学习算法的SVM优化

基于机器学习算法的SVM优化

时间:09-18 来源: 点击:

  机器学习必将会设计算法的优化问题,主要是实现Platt SMO算法,那么,下面本文对SVM的优化进行了介绍,主要实现了Platt SMO算法优化SVM模型,并尝试使用遗传算法框架GAFT对初始SVM进行了优化。

  SMO中启发式选择变量

  在SMO算法中,我们每次需要选取一对α来进行优化,通过启发式的选取我们可以更高效的选取待优化的变量使得目标函数下降的最快。

  针对第一个α1和第二个α2 Platt SMO采取不同的启发式手段。

  第一个变量的选择

  第一个变量的选择为外循环,与之前便利整个αα列表不同,在这里我们在整个样本集和非边界样本集间进行交替:

  首先我们对整个训练集进行遍历, 检查是否违反KKT条件,如果改点的αiαi和xi,yixi,yi违反了KKT条件则说明改点需要进行优化。

  Karush-Kuhn-Tucker(KKT)条件是正定二次规划问题最优点的充分必要条件。针对SVM对偶问题,KKT条件非常简单:

  

  在遍历了整个训练集并优化了相应的α后第二轮迭代我们仅仅需要遍历其中的非边界α。 所谓的非边界α就是指那些不等于边界0或者C的α值。 同样这些点仍然需要检查是否违反KKT条件并进行优化。

  之后就是不断地在两个数据集中来回交替,最终所有的α都满足KKT条件的时候,算法中止。

  为了能够快速选取有最大步长的α,我们需要对所有数据对应的误差进行缓存,因此特地写了个SVMUTIl类来保存svm中重要的变量以及一些辅助方法:

  

  下面为第一个变量选择交替遍历的大致代码,相应完整的Python实现(完整实现见https://github.com/PytLab/MLBox/blob/master/svm/svm_platt_smo.py):

  

  第二个变量的选择

  SMO中的第二个变量的选择过程为内循环,当我们已经选取第一个α1之后,我们希望我们选取的第二个变量α2优化后能有较大的变化。根据我们之前推导的式子

  

  可以知道,新的α2的变化依赖于|E1−E2|, 当E1为正时, 那么选择最小的Ei作为E2,通常将每个样本的Ei缓存到一个列表中,通过在列表中选择具有|E1−E2|的α2来近似最大化步长。

  有时候按照上述的启发式方式仍不能够是的函数值有足够的下降,这是按下述步骤进行选择:

  在非边界数据集上选择能够使函数值足够下降的样本作为第二个变量

  如果非边界数据集上没有,则在整个数据仅上进行第二个变量的选择

  如果仍然没有则重新选择第一个α1

  第二个变量选取的Python实现:

  

  KKT条件允许一定的误差

  在Platt论文中的KKT条件的判断中有一个tolerance允许一定的误差,相应的Python实现:

  

  关于Platt SMO的完整实现详见:https://github.com/PytLab/MLBox/blob/master/svm/svm_platt_smo.py

  针对之前的数据集我们使用Platt SMO进行优化可以得到:

  

  将分割线和支持向量可视化:

  

  可见通过Platt SMO优化出来的支持向量与简化版的SMO算法有些许不同。

  使用遗传算法优化SVM

  由于最近自己写了个遗传算法框架,遗传算法作为一个启发式无导型的搜索算法非常易用,于是我就尝试使用遗传算法来优化SVM。

  使用遗传算法优化,我们就可以直接优化SVM的最初形式了也就是最直观的形式:

  

  顺便再安利下自己的遗传算法框架,在此框架的帮助下,优化SVM算法我们只需要写几十行的Python代码即可。其中最主要的就是编写适应度函数,根据上面的公式我们需要计算数据集中每个点到分割线的距离并返回最小的距离即可,然后放到遗传算法中进行进化迭代。

  遗传算法框架GAFT项目地址: https://github.com/PytLab/gaft , 使用方法详见README。

  Ok, 我们开始构建种群用于进化迭代。

  创建个体与种群

  对于二维数据点,我们需要优化的参数只有三个也就是[w1,w2]和b, 个体的定义如下:

  

  种群大小这里取600,创建种群

  

  创建遗传算子和GA引擎

  这里没有什么特别的,直接使用框架中内置的算子就好了。

  

  适应度函数

  这一部分只要把上面svm初始形式描述出来就好了,只需要三行代码:

  

  开始迭代

  这里迭代300代种群

  

  绘制遗传算法优化的分割线

  

  得到的分割曲线如下图:

  

Copyright © 2017-2020 微波EDA网 版权所有

网站地图

Top