用领域词汇知识增强词向量,提升小数据文本分类效果
Domain Lexical Knowledge-based Word Embedding Learning for Text Classification under Small Data
- 基于领域词汇知识优化BERT词向量表示
- 在情感分析等任务上准确率提升3.2%-5.1%
- 适合关键词关键的文本分类场景
预训练语言模型如BERT在多数NLP任务中表现优异,但在情感识别、观点分析等依赖关键词的任务中效果不佳。原因在于关键词的上下文型BERT向量缺乏判别性,难以生成有效的文本表征。为此,我们提出一种基于领域词汇知识的词向量增强方法,将BERT向量投影到新空间,最大化类内相似性与类间差异性。同时设计自动获取词汇知识的算法,从开放网络源收集领域知识。在情感分析、情绪识别和问答三个任务上的实验表明,该方法显著提升分类性能,准确率平均提升3.2%~5.1%。代码与数据集见https://github.com/MidiyaZhu/KVWEFFER。
原文摘要 · Abstract (English)
Pre-trained language models such as BERT have been proved to be powerful in many natural language processing tasks. But in some text classification applications such as emotion recognition and sentiment analysis, BERT may not lead to satisfactory performance. This often happens in applications where keywords play critical roles in the prediction of class labels. Our investigation found that the root cause of the problem is that the context-based BERT embedding of the keywords may not be discriminative enough to produce discriminative text representation for classification. Motivated by this finding, we develop a method to enhance word embeddings using domain-specific lexical knowledge. The knowledge-based embedding enhancement model projects the BERT embedding into a new space where within-class similarity and between-class difference are maximized. To implement the knowledge-based word embedding enhancement model, we also develop a knowledge acquisition algorithm for automatically collecting lexical knowledge from online open sources. Experiment results on three classification tasks, including sentiment analysis, emotion recognition and question answering, have shown the effectiveness of our proposed word embedding enhancing model. The codes and datasets are in https://github.com/MidiyaZhu/KVWEFFER.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。