摘 要:提出了一种基于Adaboost算法和CART算法结合的分类算法。以特征为节点生成CART二叉树,用CART二叉树代替传统Adaboost算法中的弱分类器,再由这些弱分类器生成强分类器。将强分类器对数字样本和人脸样本分类,与传统Adaboost算法相比,该方法的错误率分别减少20%和86.5%。将分类器应用于目标检测上,实现了对这两种目标的快速检测和定位。结果表明,改进算法既减小了对样本分类的错误率,又保持了传统Adboost算法对目标检测的快速性。
关键词:Adaboost;CART;数据挖掘;目标识别;模式分类
数据挖掘是从大量的数据中提取出隐含有用信息的过程[1]。分类是数据挖掘的一种重要形式,在分类算法中,Adaboost算法和CART(Classification and Regression Tree)算法在对数据的分类中都有着重要的作用。Adaboost算法是一种迭代算法,其核心思想是针对同一个分类集训练不同的弱分类器,然后把这些弱分类器结合起来形成一个强分类器进而实现对数据分类,其分类速度快、精度高。2001年,由VIOLA P和JONES M将该算法应用于人脸定位[2],算法开始得到快速的发展。此后,LIENHART R和MAYDT J又用此算法成功实现了对不同方位人脸的检测[3]。决策树算法最早是由HUNT等人于1966年提出的CLS(Concept Learning System)。当前,最有影响的决策树算法是QUINLAN于1986年提出的ID3和1993年提出的C4.5。CART算法是基于以上两种方法的改进算法,它采用一种二分递归分割的技术,将当前的样本集分为两个子样本集,使得生成的决策树的每个非叶子节点都有两个分支。因此,CART算法生成的决策树是结构简洁的二叉树[4],比ID3和C4.5算法具有更好的抗噪声性能。
本算法是基于以上两种算法的改进算法,在算法的训练过程中,用CART算法生成的二叉树代替传统Adaboost算法中的弱分类器,然后级联成最终的强分类器,最后通过以实验验证了该算法的可靠性。实验分别以数字图像和人脸图像为样本,训练生成分类器,再分别对若干张测试样本分类并计算出分类误差及误差减小率。在目标检测的实验上,比较了改进算法和传统Adaboost算法的优越性,两种算法都能完全检测到目标,且耗时相当。
1 Adaboost和CART算法
1.1 Adaboost算法
Adaboost算法的训练过程就是找出若干个弱分类器[5]。设n个弱分类器(h1,h2,…,hn)是由相同的学习算法形成的,每个弱分类器能单独对未知样本分类成正样本或负样本(二分类情况),通过加权统计弱分类器的分类结果得出最终的分类结果。选择弱分类器的过程中,只要求分类器对样本的分类能力大于自然选择就可以了,即分类错误率小于0.5。凡是分类错误率低于0.5的分类器都可以作为弱分类器,但在实际的训练过程中,还是选择错误率最低的分类器作为该轮选择的弱分类器,表示如下:
特征为矩形图像中白色区域内的像素总和减去黑色区域的像素总和,它反映了白色区域到黑色区域的梯度变化情况。
试验中对特征的提取一般都是基于特征图的,特征图可以使计算量大大减少。积分图就是对要处理的图像二次积分,表示如下:
1.1.2 特征生成
特征生成即是将样本图像表示成矢量的形式。以24×24样本图为例,生成积分图之后,选择有效的Haar-Like特征模板,在积分图中移动,并保存特征值。当一次移动完之后,改变模板大小继续移动取特征值,然后将所有特征按先后顺序排列成一维向量成为代表样本的特征向量。由于模板是在积分图上移动的,因此,每次只需要知道模板的4个顶点坐标就可以通过加减法轻松计算出特征值。生成的特征数量相对较多,参考文献[3]具体分析了每个模板对应的特征的个数及其计算公式,统计了所有模板在24×24图像上移动生成的特征总数为117 941个,即以117 941维的矢量表示一个样本图。
1.2 CART算法
CART算法是决策树的一种,所不同的是,它的分支始终是二分的。用变量y表示分类结果,用X表示p维特征,该算法首先找出p维特征中对分类有效的某个特征x,将样本分成两个本集子样,以树的左右枝表示,并将此特征作为根节点。接下来判断左右子样本集是否只包含纯样本(全部正样本或全部负样本),如果是,则将此样本集定义为叶子;否则,再次在此子样本集中找出有效特征,继续将子样本集空间划分成左右枝,直到被划分的子样本集中只包含纯样本为止。在同一等级的节点中,可以选取相同属性的特征作为节点,这个划分是以递归方式实现的,最终形成一棵二叉树,形状如图3所示。
从根节点到每一个叶子节点,都对应一个规则。分类时,将待测样本的对应特征逐一在此树上从上到下搜索,直到叶子节点,此时,就将该样本的属性划分为该叶节点所表征的类(正样本或负样本)。
在决策树的分支中,常用的分支准则为信息熵法和信息增益法。其中,信息熵是ID3算法中常用的分支方法,而信息增益法主要是C4.5和CART中常用的分支方法。
信息熵本为通信电路携带信息量的大小,在这里反映的是某一个特征阈值对样本的划分准确率。对于训练例集U,假设有m个类别,全局信息熵表示为:
由于噪声的存在,决策树往往出现枝叶过于茂盛或者树干过长的情况,在分类的过程中,这会导致对训练数据过度拟合,使分类的错误率升高,反而不能对验证数据很好地分类。所以,一棵优秀的决策树应该包含剪枝的过程,即用验证数据将树的叶子或节点修剪,防止其对训练数据的过度拟合。剪枝算法有多种,常见的有前向剪枝和后向剪枝两种,CART算法采用的是后向剪枝算法。
2 改进算法
2.1 算法原理
Adaboost算法在每一轮的训练过程中都会判断某一单独特征对训练样本的分类能力,然后加大被错误分类样本的权重,减少被正确分类样本的权重。由于权重在每一轮训练完成之后都在改变,因此,每次选择的特征并不一定是最好特征,只是在当前权值条件下分类最好的特征。为了改善弱分类器对样本的分类能力,选择一棵具有3个节点的二叉树代替原来的弱分类器,即每轮训练都找出3个对分类最优的特征,构成一棵树。弱分类器的分类结果由这3个特征共同决定,比起只用单独特征分类的弱分类器而言,它对样本的分类能力更高。由于每个弱分类器对样本的分类能力提高了,因此,最终的强分类器的分类能力也将提高。为了与Adaboost算法中的弱分类器h区别,改进算法中的弱分类器用H表示。图4描述了本算法形成的分类器模型。
在训练步骤(c)中,考虑了分类错误率和信息熵增益两个因素对分类的影响。算法在每一轮训练中都选择了对分类错误率最小的3个特征,然后再从其中计算信息熵增益最大的特征作为节点。这样的选择保证了弱分类器也能有较小的分类误差,因此最终的强分类器也有较小的分类误差。
3 实验结果与分析
3.1 实验描述
为了说明改进算法的效果,在相同条件下得出了两种算法的实验结果并进行了比较。实验一以人民币图像中的数字0作为样本,样本图像均由人工采集,在不同面额的人民币图像上采集得到正样本图像和负样本图像各500张。实验二以人脸图像作为样本,样本来源于AR(AR Face Database.http://cobweb.ecn.purdue.edu/~aleix/aleix_face_DB.html)人脸数据库,正负样本各1 000张。
实验均选择以图2(a)和图2(b)的Haar-Like模板生成特征。实验过程采用交叉验证的方式完成,所有实验均在同一条件下进行。实验条件:PC机采用AMD AthlonTMII x2220 2.81 GHz处理器和2 GB内存;代码执行平台为MATLAB7.0。
3.2 实验结果
训练样本的数量越大,越能够反映真实样本的分布情况,在训练的过程中,也更能提取出对分类有效的特征。实验首先以数字图像样本为研究对象,以不同数量的样本训练分类器,然后将生成的分类器对200个测试样本分类,得到图5。可以明显看出,随着训练样本数量的增加,分类误差呈现下降的趋势,其下降的速率先快后慢,最后基本稳定在某一数值。实验还发现,当训练样本数量远远大于测试样本时,能够使测试样本的分类误差达到最小。试验中,在选取900个训练样本、100个测试样本的条件下,能够将100个测试样本完全分开,分类准确率达到100%。
由以上实验结果可知,当训练样本数量高于300个时,其分类误差基本保持在某一数值。为此,实验中将全部1 000个样本分成500个训练样本和500个测试样本(训练样本和测试样本中均各含250个正样本和负样本),分别用传统的Adabost算法和改进算法生成强分类器,对测试样本分类。图6显示了两种分类器的分类误差随着训练次数的变化情况。
由图6可以看出,随着训练次数的增加,两种分类器对测试样本的分类误差逐渐减小。在训练次数高于某个数值之后,改进算法的错误率明显低于普通的Adaboost算法,说明改进算法的分类能力较强。由于实验中所选样本的可分性较强,因此,无论是传统的Adaboost算法还是改进算法,其分类误差都比较低(小于1%)。
为了验证算法鲁棒性,实验从AR人脸库中得到正负人脸样本各1 000张,再次比较两种算法的分类情况,如图7所示。从图中可以看出,改进算法对特征不明显的人脸图像分类照样能达到较高的分类精度(99.3%),高于普通的Adaboost算法(94.8%),这说明本算法的鲁棒性较强。
从表1可以看出,改进算法对不同的目标样本分类能力均有所提高,并且提高的程度有所不同。数字样本的Harr-like特征较明显,所以,无论是改进算法还是普通的Adaboost算法,分类误差都较小,而且误差减小率也相对较小。而从两种算法对人脸样本的分类可以看出,改进算法能明显减小分类误差,提高分类器的分类能力。
将生成的分类器应用于目标检测,能够快速检测出目标在图像中的位置。由于改进算法的实现过程保留了传统Adaboost算法中以Haar-Like模板提取特征的过程,因此两种算法耗时相当。图8(b)和图8(d)分别是以上生成的数字分类器和人脸分类器对两种目标的检测结果。
本文以Adaboost算法和CART算法为基础,提出了将这两种算法相结合的改进算法,从理论上详细阐述了算法的原理和步骤。算法的关键在于,在训练样本的每一轮训练中寻找出对分类最有利的3个特征,形成二叉树,用来代替传统Adaboost算法中的弱分类器。树的形状是根据CART算法改进的,提高了单个弱分类器对样本的分类能力,由于强分类器由弱分类构成,因此,强分类器的分类能力也得到提高。最后以人民币图像上的数字0和人脸图像为样本,验证了本算法的可靠性和鲁棒性。较普通的Adaboost算法而言,改进算法对数字样本和人脸样本的分类误差率分别减少20%和86.5%,说明算法对样本的分类能力有所提高。改进算法的每轮训练都要生成有3个节点的二叉树,其训练过程将更加耗时,约等于普通算法的3倍。可以说,改进算法是以更长的训练耗时换取更高的分类精确度。由于改进算法在特征提取过程中保持了传统Adaboost算法的步骤,因此两种算法在目标检测的应用中耗时是相当的。
参考文献
[1] 毛国君. 数据挖掘的概念、系统结构和方法[J]. 计算机工程与设计, 2002,23(8):13-17.
[2] VIOLA P, JONES M. Rapid object detection using a boosted cascade of simple features[C]. Accepted Conference on Computer Vision and Pattern Recognition, 2001(5):511-518.
[3] LIENHART R, MAYDT J. An extended set of Haar-like features for rapid object detection[C]. IEEE ICIP 2002, 2002, 1:900-903.
[4] YOHANNES Y, HODDINOTT J. Classification and regression trees: an introduction[C]. International Food Policy Research Institute 2033 K Street, N.W. Washington, D.C. 20006 U.S.A, 1999.
[5] HORE U W. Comparative implementation of colour analysis based methods for face detection algorithm[C]. Emerging Trends in Engineering and Technology (ICETET), 2010(3):176-179.
[6] LISU L. Research on face detection classifier using an improved adaboost algorithm[C]. International Symposium on Computer Science, 2009(2):199-204.
[7] FREUND Y, SCHAPIRE R E. Experiments with a new boosting algorithm[C]. Machine Learning: Proceedings of the Thirteenth International Conference, San Francisco, 1996(5):148-156.