基于朴素贝叶斯的文本分类算法
原文地址:http://cn.soulmachine.me/blog/20100528/作者: 灵魂机器新浪博客:www.weibo.com/soulmachine摘要:常用的文本分类方法有支持向量机、K-近邻算法和朴素贝叶斯。其中朴素贝叶斯具有容易实现,运行速度快的特点,被广泛使用。本文详细介绍了朴素贝叶斯的基本原理,讨论了两种常见模型:多项式模型(MM)和伯努利模型(B
原文地址:http://cn.soulmachine.me/blog/20100528/
作者: 灵魂机器
新浪博客:www.weibo.com/soulmachine
摘要:常用的文本分类方法有支持向量机、K-近邻算法和朴素贝叶斯。其中朴素贝叶斯具有容易实现,运行速度快的特点,被广泛使用。本文详细介绍了朴素贝叶斯的基本原理,讨论了两种常见模型:多项式模型(MM)和伯努利模型(BM),实现了可运行的代码,并进行了一些数据测试。
关键字:朴素贝叶斯;文本分类
Text Classification Algorithm Based on Naive Bayes
Author: soulmachine
Email:soulmachine@gmail.com
Blog:http://cn.soulmachine.me
Abstract:Usually there are three methods for text classification: SVM、KNN and Naïve Bayes. Naïve Bayes is easy to implement and fast, so it is widely used. This article introduced the theory of Naïve Bayes and discussed two popular models: multinomial model(MM) and Bernoulli model(BM) in details, implemented runnable code and performed some data tests.
Keywords: naïve bayes; text classification
1 贝叶斯原理
1.1 贝叶斯公式
设A、B是两个事件,且P(A)>0,称 P(Y|X)=P(XY)P(X) 为事件A发生的条件下事件B发生的条件概率。
乘法公式 P(XYZ)=P(Z|XY)P(Y|X)P(X)
全概率公式 P(X)=P(X|Y1)+P(X|Y2)+…+P(X|Yn)
贝叶斯公式 P(Yi|X)=P(XYi)P(X)=P(X|Yi)P(Yi)P(X)=P(X|Yi)P(Yi)∑j=1nP(X|Yj)
在此处,贝叶斯公式,我们要用到的是 P(Yi|X)=P(X|Yi)P(Yi)P(X)
以上公式,请读者参考《概率论与数理统计(第五版)》的1.4节“条件概率”(这里将原书中的A换成了X,B换成了Y),获得更深的理解。
1.2 贝叶斯定理在分类中的应用
在分类(classification)问题中,常常需要把一个事物分到某个类别。一个事物具有很多属性,把它的众多属性看做一个向量,即 x=(x1,x2,x3,…,xn) ,用x这个向量来代表这个事物。类别也是有很多种,用集合 Y=y1,y2,…ym 表示。如果x属于 y1 类别,就可以给x打上 y1 标签,意思是说x属于 y1 类别。这就是所谓的分类(Classification)。
x的集合记为X,称为属性集。一般X和Y的关系是不确定的,你只能在某种程度上说x有多大可能性属于类 y1 ,比如说x有80%的可能性属于类 y1 ,这时可以把X和Y看做是随机变量, P(Y|X) 称为Y的后验概率(posterior probability),与之相对的,P(Y)称为Y的先验概率(prior probability)1。
在训练阶段,我们要根据从训练数据中收集的信息,对X和Y的每一种组合学习后验概率 P(Y|X) 。分类时,来了一个实例x,在刚才训练得到的一堆后验概率中找出所有的 P(Y|x) , 其中最大的那个y,即为x所属分类。根据贝叶斯公式,后验概率为 P(Y|X)=P(X|Y)P(Y)P(X)
在比较不同Y值的后验概率时,分母P(X)总是常数,因此可以忽略。先验概率P(Y)可以通过计算训练集中属于每一个类的训练样本所占的比例容易地估计。
我们来举个简单的例子,让读者对上述思路有个形象的认识2。
考虑一个医疗诊断问题,有两种可能的假设:(1)病人有癌症。(2)病人无癌症。样本数据来自某化验测试,它也有两种可能的结果:阳性和阴性。假设我们已经有先验知识:在所有人口中只有0.008的人患病。此外,化验测试对有病的患者有98%的可能返回阳性结果,对无病患者有97%的可能返回阴性结果。
上面的数据可以用以下概率式子表示:
P(cancer)=0.008,P(无cancer)=0.992
P(阳性|cancer)=0.98,P(阴性|cancer)=0.02
P(阳性|无cancer)=0.03,P(阴性|无cancer)=0.97
假设现在有一个新病人,化验测试返回阳性,是否将病人断定为有癌症呢?
在这里,Y={cancer,无cancer},共两个类别,这个新病人是一个样本,他有一个属性阳性,可以令x=(阳性)。我们可以来计算各个类别的后验概率:
P(cancer | 阳性) = P(阳性 | cancer)p(cancer)=0.98* 0.008 = 0.0078
P(无cancer | 阳性) =P(阳性 | 无cancer)* p(无cancer)=0.03* 0.992 = 0.0298
因此,应该判断为无癌症。
在这个例子中,类条件概率,P(cancer | 阳性)和P(无cancer | 阳性)直接告诉了我们。 |
一般地,对类条件概率 P(X|Y) 的估计,有朴素贝叶斯分类器和贝叶斯信念网络两种方法,这里介绍朴素贝叶斯分类器。
1.3 朴素贝叶斯分类器
1、条件独立性
给定类标号y,朴素贝叶斯分类器在估计类条件概率时假设属性之间条件独立。条件独立假设可以形式化的表达如下:
比如,对于一篇文章,
Good good study,Day day up.
可以用一个文本特征向量来表示,x=(Good, good, study, Day, day , up)
。一般各个词语之间肯定不是相互独立的,有一定的上下文联系。但在朴素贝叶斯文本分类时,我们假设个单词之间没有联系,可以用一个文本特征向量来表示这篇文章,这就是“朴素”的来历。
2、朴素贝叶斯如何工作
有了条件独立假设,就不必计算X和Y的每一种组合的类条件概率,只需对给定的Y,计算每个 xi 的条件概率。后一种方法更实用,因为它不需要很大的训练集就能获得较好的概率估计。
3、估计分类属性的条件概率
P(xi|Y=y) 怎么计算呢?它一般根据类别y下包含属性 xi 的实例的比例来估计。以文本分类为例,xi表示一个单词, P(xi|Y=y)= 包含该类别下包含单词的xi的文章总数/ 该类别下的文章总数。
4、贝叶斯分类器举例 假设给定了如下训练样本数据,我们学习的目标是根据给定的天气状况判断你对PlayTennis这个请求的回答是Yes还是No。
Day | Outlook | Temperature | Humidity | Wind | PlayTennis |
D1 | Sunny | Hot | High | Weak | No |
D2 | Sunny | Hot | High | Strong | No |
D3 | Overcast | Hot | High | Weak | Yes |
D4 | Rain | Mild | High | Weak | Yes |
D5 | Rain | Cool | Normal | Weak | Yes |
D6 | Rain | Cool | Normal | Strong | No |
D7 | Overcast | Cool | Normal | Strong | Yes |
D8 | Sunny | Mild | High | Weak | No |
D9 | Sunny | Cool | Normal | Weak | Yes |
D10 | Rain | Mild | Normal | Weak | Yes |
D11 | Sunny | Mild | Normal | Strong | Yes |
D12 | Overcast | Mild | High | Strong | Yes |
D13 | Overcast | Hot | Normal | Weak | Yes |
D14 | Rain | Mild | High | Strong | No |
可以看到这里样本数据集提供了14个训练样本,我们将使用此表的数据,并结合朴素贝叶斯分类器来分类下面的新实例:
x = (Outlook = Sunny,Temprature = Cool,Humidity = High,Wind = Strong)
在这个例子中,属性向量X=(Outlook, Temperature, Humidity, Wind),类集合Y={Yes, No}。我们需要利用训练数据计算后验概率 P(Yes|x) 和 P(No|x) ,如果 P(Yes|x)>P(No|x) ,那么新实例分类为Yes,否则为No。
为了计算后验概率,我们需要计算先验概率P(Yes)和P(No)和类条件概率 P(xi|Y) 。
因为有9个样本属于Yes,5个样本属于No,所以 P(Yes)=914 , P(No)=514 。类条件概率计算如下:
P(Outlook=Sunny|Yes)=29 P(Outlook=Sunny|No)=35
P(Temprature=Cool|Yes)=39 P(Temprature=Cool|No)=15
P(Humidity=High|Yes)=39 P(Humidity=High|No)=45 P(Wind=Strong|Yes)=39 P(Wind=Strong|No)=35
后验概率计算如下:
5、条件概率的m估计
假设有来了一个新样本 x1=(Outlook=Cloudy,Temprature=Cool,Humidity=High,Wind=Strong) ,要求对其分类。我们来开始计算,
P(Outlook=Cloudy|Yes)=09=0P(Outlook=Cloudy|No)=05=0
计算到这里,大家就会意识到,这里出现了一个新的属性值,在训练样本中所没有的。如果有一个属性的类条件概率为0,则整个类的后验概率就等于0,我们可以直接得到后验概率 P(Yes|x1)=P(No|x1)=0 ,这时二者相等,无法分类。
当训练样本不能覆盖那么多的属性值时,都会出现上述的窘境。简单的使用样本比例来估计类条件概率的方法太脆弱了,尤其是当训练样本少而属性数目又很大时。
解决方法是使用m估计(m-estimate)方法来估计条件概率:
2 朴素贝叶斯文本分类算法
现在开始进入本文的主旨部分:如何将贝叶斯分类器应用到文本分类上来。
2.1文本分类问题
在文本分类中,假设我们有一个文档d∈X,X是文档向量空间(document space),和一个固定的类集合C={c1,c2,…,cj},类别又称为标签。显然,文档向量空间是一个高维度空间。我们把一堆打了标签的文档集合<d,c>作为训练样本,<d,c>∈X×C。例如:
<d,c>={Beijing joins the World Trade Organization, China}
对于这个只有一句话的文档,我们把它归类到 China,即打上china标签。
我们期望用某种训练算法,训练出一个函数γ,能够将文档映射到某一个类别: γ:X→C
这种类型的学习方法叫做有监督学习,因为事先有一个监督者(我们事先给出了一堆打好标签的文档)像个老师一样监督着整个学习过程。
朴素贝叶斯分类器是一种有监督学习,常见有两种模型,多项式模型(multinomial model)和伯努利模型(Bernoulli model)3。
2.2 多项式模型
2.2.1 基本原理
在多项式模型中, 设某文档 d=(t1,t2,…,tk) ,tk是该文档中出现过的单词,允许重复,则
先验概率 P(c)= 类c下单词总数/整个训练样本的单词总数
类条件概率 P(tk|c)= (类c下单词tk在各个文档中出现过的次数之和+1)/(类c下单词总数+|V|)
V是训练样本的单词表(即抽取单词,单词出现多次,只算一个),|V|
则表示训练样本包含多少种单词。在这里,m=|V|, p=1/|V|
。
P(tk|c)= 可以看作是单词tk在证明d属于类c上提供了多大的证据,而P(c)则可以认为是类别c在整体上占多大比例(有多大可能性)。
2.2.2 伪代码4
2.2.3 举例
给定一组分类好了的文本训练数据,如下:
docId | doc | 类别In c=China? |
1 | Chinese Beijing Chinese | yes |
2 | Chinese Chinese Shanghai | yes |
3 | Chinese Macao | yes |
4 | Tokyo Japan Chinese | no |
给定一个新样本 > Chinese Chinese Chinese Tokyo Japan
对其进行分类。该文本用属性向量表示为d=(Chinese, Chinese, Chinese, Tokyo, Japan)
,类别集合为Y={yes, no}
。
类yes下总共有8个单词,类no下总共有3个单词,训练样本单词总数为11,因此 P(yes)=811,P(no)=311 。类条件概率计算如下:
P(Chinese|yes)=5+18+6=614=37
P(Japan|yes)=P(Tokyo|yes)=0+18+6=114
P(Chinese|no)=1+13+6=29
P(Japan|no)=P(Tokyo|no)=1+13+6=29
分母中的8,是指yes类别下textc的长度,也即训练样本的单词总数,6是指训练样本有Chinese,Beijing,Shanghai, Macao, Tokyo, Japan 共6个单词,3是指no类下共有3个单词。
有了以上类条件概率,开始计算后验概率,
P(yes|d)=(37)3×114×114×811=108184877≈0.00058417
P(no|d)=(29)3×29×29×311=32216513≈0.00014780
因此,这个文档属于类别china。
2.3 伯努利模型
2.3.1 基本原理
P(c)= 类c下文件总数/整个训练样本的文件总数
P(tk|c)= (类c下包含单词tk的文件数+1)/(类c下单词总数+2)
在这里, m=2,p=12 。
后验概率的计算,也有点变化,见下面的伪代码。
2.3.2 伪代码
2.3.3 举例
还是使用前面例子中的数据,不过模型换成了使用伯努利模型。
类yes下总共有3个文件,类no下有1个文件,训练样本文件总数为11,因此 P(yes)=34,P(Chinese|yes)=3+13+2=45
P(Japan|yes)=P(Tokyo|yes)=0+13+2=15
P(Beijing|yes)=P(Macao|yes)=P(Shanghai|yes)=1+13+2=25
P(Chinese|no)=1+11+2=23
P(Japan|no)=P(Tokyo|no)=1+11+2=23
P(Beijing|no)=P(Macao|no)=P(Shanghai|no)=0+11+2=13
有了以上类条件概率,开始计算后验概率,
因此,这个文档不属于类别china。
2.4 两个模型的区别
二者的计算粒度不一样,多项式模型以单词为粒度,伯努利模型以文件为粒度,因此二者的先验概率和类条件概率的计算方法都不同。
计算后验概率时,对于一个文档d,多项式模型中,只有在d中出现过的单词,才会参与后验概率计算,伯努利模型中,没有在d中出现,但是在全局单词表中出现的单词,也会参与计算,不过是作为“反方”参与的。
3 代码详解
本文附带了一个eclipse工程,有完整的源代码,以及一个微型文本训练库。
ChineseSpliter用于中文分词,StopWordsHandler用于判断一个单词是否是停止词,ClassifyResult用于保存结果,IntermediateData用于预处理文本语料库,TrainnedModel用于保存训练后得到的数据,NaiveBayesClassifier是基础类,包含了贝叶斯分类器的主要代码,MultiNomialNB是多项式模型,类似的,BernoulliNB是伯努利模型,二者都继承自NaiveBayesClassifier,都只重写了父类的计算先验概率,类条件概率和后验概率这3个函数。
3.1 中文分词
中文分词不是本文的重点,这里我们直接使用第三方工具,本源码使用的是极易中文分词组件,你还可以使用MMSEG,中科院的ICTCLAS等等。
3.2 停止词处理
停止词(Stop Word)是指那些无意义的字或词,如“的”、“在”等。去掉文档中的停止词也是必须的一项工作,这里简单的定义了一些常见的停止词,并根据这些常用停止词在分词时进行判断。
3.3 预处理数据
我们这里使用搜狗的文本分类语料库作为训练样本,把SogouC.reduced.20061102.tar.gz解压到D盘,目录结构为
IntermediateData.java主要用于处理文本数据,将所需要的信息计算好,存放到数据库文件中。
中间数据文件主要保存了如下信息,
我们使用命令
将文本训练库的信息计算好,保存到中间文件中。以后的阶段,我们都不再需要文本语料库了,只需要reduced.db。
3.3 训练
基本的框架代码都在NaiveBayesClassifier中,MultiNomialNB和BernoulliNB都只是重新实现(override)了这三个函数。
训练函数如下:
我们使用命令:
开始训练,得到的模型文件保存在reduced.mdl中。
3.4 分类
有了模型文件,就可以用它来进行分类了。
可以使用命令
对文本文件temp.txt进行分类。
还可以将当初训练出这个模型文件的文本库,进行分类,看看正确率有多少,即“吃自己的狗食”,命令行如下
分类函数如下:
测试正确率的函数getCorrectRate(),核心代码就是对每个文本文件调用classify(),将得到的类别和原始的类别比较,经过统计后就可以得到百分比。
更多细节请读者阅读源代码。
参考文献
-
Pang-Ning Tan, Michael Steinbach, Vipin Kumar, 《数据挖掘导论》,北京:人民邮电出版社,2007,第140~145页。↩
-
石志伟, 吴功宜, “基于朴素贝叶斯分类器的文本分类算法”, 第一届全国信息检索与内容安全学术会议,2004↩
-
洞庭散人,“基于朴素贝叶斯分类器的文本分类算法(上)”,“基于朴素贝叶斯分类器的文本分类算法(下)”,2008↩
-
Christopher D. Manning, Prabhakar Raghavan, Hinrich Schütze, Introduction to Information Retrieval, Cambridge University Press, 2008, chapter 13, Text classification and Naive Bayes.↩
更多推荐
所有评论(0)