不同情境下中文文本分类模型的表现及选择

兰秋军　李卫康　刘文星

摘要：针对中文文本分类任务中N-Gram，素贝叶斯、K最近邻和TF-IDF等经典而广泛使用的文本分类模型的选择困惑问题，基于万余篇中文新闻文本语料数据，设计了一系列的对比实验，考察了各模型在不同参数、不同训练数据规模、不同训练文本长度、类别是否偏斜等多种情境下分类性能的表现，总结了各模型的特性，为中文文本分类模型的选择和应用提供了实践依据和参考。

关键词：中文文本；文本分类；数据挖掘；情报分析

中图分类号：TP274；TP302 文献标识码：A

文本挖掘是语言学、统计学以及计算机技术相结合的产物，是对海量文本信息进行自动处理，获取人们感兴趣的、隐含的、有用信息的过程，在信息检索、生物医学、情报获取、舆情分析和市场营销等众多领域备受关注。文本分类作为文本挖掘领域中的核心技术，是各种自然语言处理、应用的基础。其中分类模型的选择对最终结果具有至关重要的影响。然而，因所基于的原理、参数、应用场合各不相同，即使相同的模型其性能表现也往往大相径庭。

新闻文本是一类常见的文本形式，其蕴含的信息量大，是各种情报分析的重要数据源。尽管现有的各个新闻网站以栏目形式对新闻进行了人工划分，然而各网站的分类体系和栏目形式各不相同，因此在具体的新闻挖掘应用项目中，常需将采集的新闻数据重新进行组织和划分。中文文本分类领域中具有代表性的模型是朴素贝叶斯、N-Gram，K最近邻和TF-IDF。这些不同的模型各具有怎样的特性？分别适合哪些场合？在使用时应如何选取合适的参数？人们往往面临困惑。由于各方法在处理细节上有不少差异，很难从理论分析的角度来比较各方法的优劣。因此，基于典型数据，采用实验的方式进行比较是比较通行的做法。本文精心构造了多组实验，从模型参数选取、训练数据规模、训练文本长度、数据是否偏斜等几个情境来考察各模型在不同情境下的性能表现，其结论对中文文本分类模型的选择与参数设置等具有实践指导意义。

1 模型概述

文本分类就是通过计算机程序自动将某个文档归属到事先给定的类别体系中一个或多个类别。现有的文本分类方法大致可归为两类：基于规则的方法和基于机器学习的方法。其中基于规则的方法早在20世纪70年代就已出现，但因规则制定的困难，目前普遍采用的是基于机器学习的方法。而机器学习方法中，基于统计的方法是最具有代表性和使用最为广泛的。其中，N元模型（N-Gram）、朴素贝叶斯（NB），K最近邻（KNN）和TF-IDF又是其中最经典的几个模型。

N-Gram模型基于马尔科夫假设，即下一词的出现概率仅依赖于它前面的N个词，统计N元词串在各类别中出现概率，以此确定文档归属于哪个类别。朴素贝叶斯模型基于贝叶斯定理，假设单词两两独立，获得各文档类别的后验概率，哪个类别概率值大，文档即归属于该类别。KNN的主要思想是先将文档内容转化为特征空间中的特征向量，计算待分类文档与训练文档中每个样本的相似度，找出其中的k个最近邻居，据此判别文档所属类别。TF-IDF则先将文本内容转化为特征向量，然后计算其与类别特征向量间的余弦相似度，以此作为其所属类别的判据。分析上述模型，不难发现，N-Gram模型主要是提取了不同类别文档中字与字之间的顺序依赖关系来构造分类特征，朴素贝叶斯则提取了不同类别文档中词与词之间的概率依赖关系构成分类依据，K近邻直接利用了空间向量模型，以文档相似性特征作为分类依据，而TF-IDF则同时考虑了词在文档本身中的出现频度以及其在不同文档中的出现频度信息。几个模型所抓取的文档类别信息特征明显不同，很难在理论上判别哪个模型更好，更适合哪些情境。因此，从实验的角度来考察是更为切合实际的方案。

2 基础与准备

2.1 算法实现工具

LingPipe是基于Java语言的自然语言处理的开源软件包，提供了文本挖掘各阶段的基本功能。由于该软件包的数据处理都基于一个共同框架，采用了相同的基础源代码模块，故本文以其作为算法实现工具，可尽量减少模型本身之外的因素（如文本预处理阶段的分词、特征提取、文本表示等）给模型性能带来的影响。

2.2 实验数据

本文实验数据采集来自新浪、腾讯、凤凰等主流网站。特地挑选了历史、军事、文化、读书、社会几个比较近似，甚至人工也容易分错的文本类别。其中，历史类和军事类的文章比较相近，而文化类和读书类的也常相似。数据采集跨时2个月，去除了所有Html标记和网页中的噪声文本，只包含标题、正文内容以及标点符号。共采集16000篇，去除了部分重复和校验过程中有问题的文档，最终保留14000篇作为本文研究的语料数据。其中含历史1900篇，军事1600篇，文化2500篇，读书4000篇，社会4000篇。每篇文章按类别以txt文件的形式保存。各类别、不同文本长度的文档篇数分布情况如表1所示，所有文档的长度介于10000字节以内，涵盖了网页中的绝大多数新闻文本长度。

2.3 分词与特征项

尽管特征选择和预处理措施都是影响文本分类性能的关键因素，但因各模型算法原理相差太大，无法基于统一的特征项和预处理进行比较。因此，各模型的特征项均以词频为基础，采取各模型常用的特征形式和预处理方式。N-Gram模型本身不需分词，因此未做分词处理，而其它模型则应用中科院分词系统NLPIR_2014进行分词处理。

2.4 分类评价指标

LingPipe提供了一系列指标对模型性能进行评估。本文实验主要采用宏平均、微平均下的F值进行评价，它综合考虑了准确率、召回率两个被广泛认可的分类器评价指标。其详细定义和含义可参见相关文献。此外，实验过程中，还记录了各模型的训练和分类运行时间。这也是反映分类模型性能的一个方面。

3 实验方案与结果

从应用角度来看，分类准确度和处理效率是用户最为关注的两个方面。而影响这两方面的因素无外乎模型本身和待处理的数据，如图1所示。模型本身因素具体包括模型的构造机制和模型参数。其中，模型机制对用户而言是封装的，要提升分类性能，用户只能调整模型参数。而数据方面，文本的词语和语义特点太过复杂和精细，用户难以据此选择模型。然而待处理文本的长度、规模和偏斜程度等是影响分类性能的重要因素，用户可以据此选择最合适的模型。因此，本文主要设计了4组实验考察不同情境下中文文本分类模型的表现。下面具体阐述各组实验的具体方案及结果。

3.1 模型参数与模型性能

所述的4个分类模型中，NB模型和TF-IDF模型没有参数，而N-Gram和KNN模型则分别有一个关键参数N和K。先对这两个模型进行不同参数取值下的实验。所采用的数据集文档篇数情况如表2所示。

3.1.1 关于N-Gram模型参数N的实验

根据N-Gram模型中参数N的含义，字与字之间的概率依赖关系主要由词组造成，汉语超过6个字的词组已经相当稀少。本组实验考察了N取值为1，2，4，6，7的情形，获得结果如表3所示。可见，随着N取值的加大，该分类器的分类性能也越高，但在N超过4以后，分类性能改善幅度已相当小，而训练时间和测试时间却成倍增长，为此，后述实验均取N=4，不再赘述。

3.1.2 关于KNN模型参数K的实验

参数K为经验参数，表示选取的近邻个数，其值的大小对于模型的分类性能有显著的影响。实验中，为确定K最佳值，将K分别取值1，3，5，10，20，获得如表4所示结果。显见，随着K值的增加，分类性能有缓慢下降趋势，表明并非选取的近邻数越多越好。原因在于KNN基于向量空间模型，维数较高，数据比较稀疏，K值越大，反而可能带来更多的误判信息。本实验中，K取值为1时分类性能最优，因此在后述实验中均取该最优值。

3.2 训练集规模与模型性能

为了考察各模型在不同训练集规模情境下的性能表现，分8次小实验，每次从实验语料中抽取1000，2000，3000，4000，5000，6000，7000，8000篇文档构成训练集，抽取1000篇文档构成测试集。抽样过程中，为了消除各类别新闻分布不一致、文本长度不一致所带来的影响，进行了适度控制，即确保8次实验中，各类别新闻的比例保持一致，各文本长度所占比例也保持一致。在此控制下，随机抽取样本，每次实验进行3次，以其平均值作为最后结果。实验结果如表5所示，图2展示了不同训练集规模情境下，模型性能（微平均下F值）的情况；图3和图4则分别展示了分类训练时间和测试时间与训练集规模的关系。

由表5以及图2～图4可知，在各类别数据相近的情况下，随着训练集规模的增大，各模型的分类性能也均得到提升并渐趋于平稳，与文献结论相同。KNN改善效果最为明显。就运行时间而言，各分类模型随着训练集规模的增大，训练时间明显增加，而测试时间仅KNN分类模型显著增加，其它则变化微小。通过逐渐加大训练集规模，实验还发现，在测试集不变的情况下，训练集达到一定规模后（例如7000篇），即使再显著增大训练集规模，分类性能的改善也非常微弱。

3.3 文本长度与性能

为了考察各模型在不同文本长度情境下性能的表现，将训练数据的每个类别都按文档大小进行划分，抽取5个子集，分别为1000字节以下，1000-2000字节，2000-3000字节，3000-5000字节，5000字节以上。在保证5个子集的类别分布和规模分布一致的前提下，随机抽取样本构成训练数据，实验数据如表6所示。该实验共进行3次，取3次实验结果的平均值作为最终结果，如表7所示。图5展示了不同文本长度情境下，模型性能（微平均下F值）的情况；图6和图7则展示了分类训练时间和测试时间与文本长度的关系。

对文本长度情境而言，从表7以及图5～图7可以看出，随着文本长度的增加，除KNN模型外，其它3个分类模型的准确性能在初期快速提升，其后趋缓而渐趋平稳，但KNN快速提升后却逐渐下降。实验还发现，在文本长度短时（小于1000字节），TF-IDF模型要好于其它3个模型。从时间性能来看，各模型都随文本长度增加而近似呈线性增长趋势。其中NB和TF-IDF相对较逊一筹。

3.4 类别偏斜与模型性能

为了考察各模型在类别是否偏斜情境下性能的表现，我们设计了类别均衡和非均衡两组实验，实验数据集如表8所示，训练集和测试集规模相同，非均衡训练集中读书和社会两个类别远高于其它类别数据，而均衡训练集中各类别文本数相同。每组实验共进行3次，随机抽取样本构成训练数据。取3次实验结果的平均值作为最终结果，如表9所示。

对类别偏斜情境而言，观察可知，各模型整体性能及各类别分类性能在均衡训练集情境下的表现均优于在非均衡训练集下的性能表现。在非均衡训练集情境下，对各小类而言，包含文本数较多的读书和社会两个类别的分类性能要优于其它类别，与文献对训练集类别分布对文本分类影响的研究结果类似。4个分类模型中，对于类别均衡数据而言，TF-IDF表现最佳，对于类别非均衡数据而言，NB表现最佳。

4 结论

本文所考察的几个模型是当前文本分类领域应用最为广泛、最为经典的。在实践当中，各模型的表现各异，而在理论上又很难分析和评价其优劣。为此本文构建了多组实验来考察不同情境下各模型的表现，形成的结论及模型选择建议如下：

1）几个模型在运行效率方面没有明显的区别，训练和测试时间都与数据集的规模和文本长度呈线性关系。实验结果看，NB模型和TF-IDF虽然稍逊，但实际应用中，此差别并不明显，可以忽略。

2）不管在何种情境下，KNN的表现都最差，因此不推荐采用此模型。

3）从训练集规模来看，几种模型都是随着规模增大而分类精确性稳步增加，训练集的大小不构成模型选择的关键依据。

4）N-Gram、TF-IDF、NB三个模型的分类精度受文本长度的影响差别不大。都随文本长度的增加而精度得到提升，且都在长度低于1kb（约400汉字）时提升较明显，而此后提升速度放缓。KNN则未能因文本长度的增加而提升其精确度。

5）对类别分布是否偏斜的情况，NB模型表现出良好的稳定性，比N-Gram和TF-IDF都要好。因此，在类别偏斜严重的情况下，推荐采用NB模型。