自然语言处理之情感分类

情感分类是指根据文本所表达的含义和情感信息将文本划分成褒扬或者贬义的两种或者几种类型,是对文本作者情感倾向、观点或者态度的划分,有时也称为倾向性分析(opinion analysis)。【1】在电商领域,情感分类能够帮助识别用户的情感倾向,帮助我们优化服务和用户体验,应用场景包括评价情感分类,客服对话中的情感分析,评价排序等。


fig. 1 情感二分类示意图,正向或者负向。【3】

情感分类方法按照机器学习方法的划分,可以分为监督学习,半监督学习,以及非监督学习【1】。我们主要聚焦在监督学习方法,即基于标注文本的机器学习方法。监督方法,包括传统机器学习方法,比如SVM或者决策树方法,优点是方法简单,不足是需要做特征工程,抽取离散文本特征比如n元语义(n-gram);最近比较火的是基于神经网络或者深度学习方法,比如text-cnn,text-rnn,或者更复杂的网络结构。


fig.2 text-CNN示意图【4】

中文文本分类一个挑战就是中文语料的匮乏,为了克服这一问题,我们可以采用半监督学习方法比如协同学习(co-training,利用标注的英文数据和未标注的中文数据进行训练),也可以利用迁移学习方法(transfer-learning)。迁移学习是深度学习领域解决的数据匮乏问题的主要方法,通过借助其他训练任务的数据,来预训练(pre-training)网络或者进行多任务学习(multi-task learning)来改善当前任务的模型效果。迁移学习充分利用不同学习任务的共享信息,同时考虑了当前任务的特殊性。

BERT是google最近推出的自然语言处理中深度特征表示(character representation)的通用模型。借助BERT,我们在多个自然语言处理任务中都取得了突破性的进展。BERT是一个庞大的网络,训练过程是一个技术活,而且耗费资源,但是借助google预训练好的模型,我们进行迁移学习,来改善我们学习任务,包括情感分类,命名体识别(NER)等。

fig. 2  BERT 网络结构示意图,对比GPT和ELMo 【2】

接下来详细介绍我们团队最近在利用bert进行情感分类的实践。步骤如下

1)下载bert源码:

git clone https://github.com/google-research/bert

2)下载google预训练好的中文模型

BERT-Base, Chinese 

3)准备分类情感数据,BERT要求输入数据的格式为(制表符\t分割):

文本id    A分句     B分句     情感标签

具体实例为:

91777    开关机的按键很不合适    按不下去    1

4)配置分类任务的Processor, 可以参考已有的Processor类ColaProcessor

5)修改执行shell run.sh:

python run_classifier.py \

--task_name=s6 \    # 任务名

--do_train=true \     # 是否训练,训练选true,预测选false

--do_eval=true \     #  是否进行验证

--do_predict=true \  #  是否进行预测

--data_dir=${MY_DATASET} \   # 指定输入数据集,train.csv. val.csv, test.csv 分别用于训练,验证和测试

--vocab_file=${BERT_BASE_DIR}/vocab.txt \     # 词典文件

--bert_config_file=${BERT_BASE_DIR}/bert_config.json \    # 模型配置文件

--init_checkpoint=${BERT_BASE_DIR}/bert_model.ckpt \    # bert模型文件

--max_seq_length=128 \  # 文本序列最大长度

--train_batch_size=32 \  # batch_size

--learning_rate=5e-5 \    # 学习率,因为是进行fine-tuning,所以设定的学习率较小

--num_train_epochs=2.0 \    # epoch个数,2-3个epoch足够

--output_dir=./logs/   # 结果数据路径

6) 进行训练

 sh run.sh

最终我们训练出来的结果如下(以fasttext模型的训练结果作为baseline):

method   |    Test F1

---------------------

BERT       |      90.7%    

Fasttext   |      89.6%

-----------------------

我们发现利用bert预训练+fine-tuning的方法在我们的情感分类任务上f1值比baseline模型提升1.1%,确实提升了模型泛化能力。以上只是demo,更细致的结果待续。

综上所述,情感分类是一种重要的文本分类任务,它在电商领域能够帮助我们改善用户体验和服务。基于bert的迁移学习方法,能够帮助我们显著优化情感分类任务的泛化能力,是未来我们线上服务的候选模型之一(大型模型部署环节还没有打通,主要是效率跟不上)。

参考:

【1】宗成庆, 统计自然语言处理,第二版,2017

【2】Devlin et.al.,BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding, arxiv.org:1810.04805, 2018.

【3】https://webhose.io/blog/api/how-to-use-rated-reviews-for-sentiment-classification/

【4】https://zhuanlan.zhihu.com/p/40276005

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,922评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,591评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,546评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,467评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,553评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,580评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,588评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,334评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,780评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,092评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,270评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,925评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,573评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,194评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,437评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,154评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352

推荐阅读更多精彩内容