杭研技术秀 | 短文本内容检测反垃圾技术

【安全知识】2019-12-19

分享到

本文针对短文本数据，作者采用了一种基于二元文法双词共现的概率统计建模方法，在色情-辱骂分类问题上取得了比传统LDA主题模型更优的效果，并且准确率和召回率指标能够随着主题个数的增加而稳定提高。

一、技术背景介绍

数据挖掘工程师都知道，短文本数据在我们的生活中无处不在。只要我们上网，我们就会遇到形式多样的短文本，比如文本广告、图片标题、微博、优酷视频、豆瓣评论、短信、百度知道里的提问和回答。数据挖掘中的内容分析任务就是要从这些海量的短文本数据中提取出精准的主题，为后续的内容识别分类、用户建模画像、意图识别、热点话题爆发检测等任务提供良好的数据基础。

尽管文本分析领域已经有成熟的数据挖掘方法，但是针对短文本的主题挖掘还是存在很多技术瓶颈。传统的文本主题挖掘方法，比如PLSA（概率潜在语义分析）、LDA（隐含狄利克雷分布），都假设了一篇文章的内容有多种主题混合而成，并且每个主题都有各自不同的权重，占主导地位的主题就体现了一篇文章的主旨。但是，这些传统的文本建模技术是针对文档层面的词语共现模式进行设计的，在运用到短文本数据的时候就遇到了严重的稀疏性问题。在长文本数据中，我们有足够长的文字篇幅让这些传统技术学习到词语的共现关系，主题代表性强的词能够在文本中经常出现。但是在短文中，词语的wordcount并不能反映哪个词更重要；而且在短文本数据中，由于缺乏足够的上下文信息，这些传统技术对多义词的鉴别能力更是大幅下降。

二、短文本模型原理

针对业务中遇到短文本垃圾评论数据，我们采用针对二元文法概率建模的BTM模型（A Biterm Topic Model for Short Texts，www2013）对数据进行主题建模和检测，在对”色情“-“辱骂”的短文本数据的分类问题中，相比传统的LDA主题模型，得到了更好的分类效果。

传统的概率统计方法，本质上是在document层次上对词语的共现模型进行设计的，得到的主题，可以用这个主题相关的词语来表示。而在短文本问题中，在document的层次上反应的上下文信息较少，BTM模型直接对Bi-gram二元文法的词语共现进行建模。BTM模型中，将短文本中的每对共现词，看成是一定概率分布下随机生成的过程，可以用概率图模型直观地表示，如图2。在BTM中，除了假设每一条文本数据的内容是由多种主题构成的，针对二元文法中共现的词语pair，同一条数据中共同出现的两个词语也被设计成从同一个主题下的分布生成。

不同于传统的LDA概率模型，BTM短文本概率统计模型的数据生成过程设计如下：

值得一提的是BTM模型直接对二元文法中的词语进行概率分布下的随机生成，而不是对文档document进行建模，因此文档的主题分布特征只能通过“贝叶斯定理”计算得到，

BTM模型的训练学习过程如下：

从以上模型设计可以看到BTM的学习训练过程极为高效，模型参数学习相对于传统LDA文本模型更为简单。

三、短文本内容检测性能测试分析

以下是业务中BTM模型在短文本评论数据上对“色情”-“辱骂”主题的分类检测流程：

图5所示的是业务中处理的短文本数据经过预处理过滤、分词得到的样例数据。

图6所示的是短文本数据词频分布统计，分布情况基本和推荐系统的用户、商品的大数据分布类似，集中体现出长尾的现象。

以下是对BTM和传统LDA在短文本数据内容检测识别任务上的性能比较。

图7是对两种文本模型在“色情”VS“辱骂”分类任务上的准确率（ACC）性能比较，我们测试比较了BTM和LDA模型在不同topic主题个数下的性能变化曲线。

图8是对两种文本模型在“色情”VS“辱骂”分类任务上的AUC性能指标的比较，我们同样测试比较了BTM和LDA模型在不同topic主题个数下的性能变化曲线。

图9是BTM和LDA文本模型在短文本数据分类（“色情”VS“辱骂”）上的AUC性能曲线比较。

从结果中，可以看到BTM在短文本主题学习上比传统的LDA概率文本模型有明显的优势：不仅在ACC和AUC具体指标上BTM有更好的效果，传统LDA文本模型学习到的概率主题特征在topicNum>300之后，主题特征的同质化现象严重（短文本数据集的词稀疏性、主题稀疏性的影响较重），分类器训练和预测的时候，预测分数趋向集中，CV确定分类阈值变得困难，而且偏离0.5的中心值；实际经验情况，概率模型的先验参数需要尽可能调低，最优阈值分数确定在0.585左右。

为了更直观地观察BTM学习到的“语义主题”，我们打印出每个主题下的最具代表性的词语进行可视化描述。如图10所示，红色方框圈出来的“语义主题”具有明显的“色情”倾向性；蓝色方框圈出来的“语义主题”具有明显的“辱骂”倾向性。在图11-1和图11-2中我们选取了其中4个“语义主题”进行词云图的可视化，可以明显看到各个“主题”的语义倾向性。