CN113312477A - 一种基于图注意力的半结构文本分类方案 - Google Patents
一种基于图注意力的半结构文本分类方案 Download PDFInfo
- Publication number
- CN113312477A CN113312477A CN202110415787.8A CN202110415787A CN113312477A CN 113312477 A CN113312477 A CN 113312477A CN 202110415787 A CN202110415787 A CN 202110415787A CN 113312477 A CN113312477 A CN 113312477A
- Authority
- CN
- China
- Prior art keywords
- matrix
- text
- vector
- graph
- classification
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 239000011159 matrix material Substances 0.000 claims abstract description 82
- 239000013598 vector Substances 0.000 claims abstract description 57
- 238000012549 training Methods 0.000 claims abstract description 23
- 238000000034 method Methods 0.000 claims abstract description 14
- 238000004364 calculation method Methods 0.000 claims abstract description 8
- 238000004140 cleaning Methods 0.000 claims abstract description 6
- 238000007781 pre-processing Methods 0.000 claims abstract description 6
- 230000008569 process Effects 0.000 claims description 7
- 238000012512 characterization method Methods 0.000 claims description 3
- 238000006243 chemical reaction Methods 0.000 claims description 3
- 230000006835 compression Effects 0.000 claims description 3
- 238000007906 compression Methods 0.000 claims description 3
- 230000009467 reduction Effects 0.000 claims description 3
- 230000011218 segmentation Effects 0.000 claims description 2
- 238000000605 extraction Methods 0.000 abstract description 5
- 230000006872 improvement Effects 0.000 abstract description 4
- 230000000694 effects Effects 0.000 description 4
- 238000012545 processing Methods 0.000 description 4
- 238000007635 classification algorithm Methods 0.000 description 3
- 238000010586 diagram Methods 0.000 description 3
- 230000007246 mechanism Effects 0.000 description 3
- 230000009286 beneficial effect Effects 0.000 description 2
- 238000013145 classification model Methods 0.000 description 2
- 230000006870 function Effects 0.000 description 2
- 238000003058 natural language processing Methods 0.000 description 2
- 238000011176 pooling Methods 0.000 description 2
- 238000006467 substitution reaction Methods 0.000 description 2
- 238000004458 analytical method Methods 0.000 description 1
- 238000004422 calculation algorithm Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 230000008451 emotion Effects 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 238000012423 maintenance Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
- G06F16/30—Information retrieval; Database structures therefor; File system structures therefor of unstructured textual data
- G06F16/35—Clustering; Classification
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Data Mining & Analysis (AREA)
- Databases & Information Systems (AREA)
- Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Machine Translation (AREA)
- Information Retrieval, Db Structures And Fs Structures Therefor (AREA)
- Document Processing Apparatus (AREA)
Abstract
一种基于图注意力的半结构文本分类方案,它包含以下技术方案步骤:步骤一,文本预处理、数据清洗,形成图矩阵;步骤二,形成编码层,采用albert预训练模型获取embedding矩阵;步骤三,对embedding矩阵进行attention操作,并基于图矩阵进行加权计算;步骤四,对特征向量矩阵压缩后进行文本分类。针对金融领域文本分类问题,尤其是在半结构化的文本的情况下,寻常的分类器无法结合结构化信息进行识别,给出一种提取结构信息并且融入到分类***中,对最终的分类器准确率提升是行之有效的,同时针对模型结构,尤其是取末尾几层transformer、跨句间结构权重比同句内更高、multi‑headattention等步骤,大大提高了信息表征抽取的能力,从而增加最后softmax分类的准确率,具有较大的市场推广价值。
Description
技术领域
本发明涉及文本分类技术领域,具体涉及一种基于图注意力的半结构文本分类方案。
背景技术
在自然语言处理中,文本的分类是一个比较成熟也比较常见的文本处理任务。常见的有文本的标签分类,情感分类等任务,在训练样本充分的情况下,当前的这些任务都能取得比较好的效果,在一些私有领域(如金融领域),存在一类比较特殊的文本数据信息,它主要由大量的数字构成,并且根据一定的排列顺序来区分不同的数据类型,如图1、图2所示,图1是二级现券成交信息,图2是一级投标信息,判断它们不仅需要依赖文字信息,还需要考虑它们的结构信息,在各种文本处理的任务中,需要对这类数据进行分类。专业人员根据文字信息,以及它们的排列结构信息来判断一段文本的类别。因此如何构建既考虑文本信息,又考虑文本排列的结构信息,这对现有的分类算法来说,是一个难点问题。
关于文本分类的任务,有一系列的算法模型,如Fasttext,textcnn和textrnn等,以及随着预训练模型的发展,衍生的一系列基于bert的分类模型,融合attention机制的textcnn+attention,textrnn+attention等等,这些模型能够比较好地解决通用域的一些文本分类问题,针对金融领域的文本语料,很大一部分是如上描述的半结构文本数据,当前的处理技术主要有两种,一是基于“关键词+规则引擎”的技术,根据关键词判别或其余的说辞方式区分所属的不同业务。二是采用以上提到的NLP领域常见的文本分类算法模型,如Bert+的分类模型,需要考虑计算性能的则采用FastText、TextCNN等模型,针对带有结构信息,或者需要依赖结构信息来进行判断的分类任务;两种方法都有一定的局限性,首先基于关键词+规则引擎的方案,需要行业专家提供经验并整合起来,一般在比较理想化的语料中会处理得比较好,在应对实际线上业务数据就会捉襟见肘,要么是规则冲突要么是考虑不周,而且规则堆积之后维护成本极高,且运行效率也会越来越低;其次,基于深度学习的文本分类算法,目前的方案都没有将“结构信息”融合进模型的学习中,虽然基于Bert+的一些预训练模型,能够学习到字符的位置信息,但是仍然不能有效的学习到结构特征,在区分闲聊等语料和业务语料(一级投标、二级现券成交等)数据的时候效果很不错,但是针对相似度比较高的半结构化文本(一级投标和二级现券成交等)的分类时,效果会明显下降很多。
现有方案均没有很好得利用数据内部的结构化信息,因此,若可以将此部分的信息融入到判断方案里头,理应会得到不错的提升。
发明内容
本发明的目的在于针对现有技术的缺陷和不足,提供一种基于图注意力的半结构文本分类方案,以解决上述背景技术中提出的问题,本发明针对金融领域文本分类问题,尤其是在半结构化的文本的情况下,寻常的分类器无法结合结构化信息进行识别,给出一种提取结构信息并且融入到分类***中,对最终的分类器准确率提升是行之有效的,同时针对模型结构,尤其是取末尾几层transformer、跨句间结构权重比同句内更高、multi-headattention等步骤,大大提高了信息表征抽取的能力,从而增加最后softmax分类的准确率,具有较大的市场推广价值。
为实现上述目的,本发明包含以下技术方案步骤:步骤一,文本预处理、数据清洗,形成图矩阵M;步骤二,形成编码层,采用albert预训练模型获取embedding矩阵;步骤三,对embedding矩阵进行attention操作,并基于图矩阵进行加权计算;步骤四,对特征向量矩阵压缩后进行文本分类。
所述的步骤一文本通常含有大量的噪音数据,首先对文本进行预处理清洗,如全半角转换、英文大小写统一、多个空白字符合并、去停用词等,文本切割,根据文本换行符,以及空格键、逗号分号、Tab键等分割符的判断,将文本数据切割成多个词或短句的形式,判断短句、词在文本中的位置,所处的行、列的位置信息,判断它们的连接关系,采用简单逻辑规则,在水平方向上,一个词与左右的词直接关联,在垂直方向上,如果水平方向处于同一位置,则关联,如此形成如下图4所示的图矩阵M,图矩阵的表达上,本方案采用不同的值表达不同的关联,单行内相邻字符之间的关联用数值1标识,要素之间的关联用2标识,其余未标出的区域均默认设置为0。
所述的步骤二将文本通过事先在专业语料训练过的预训练模型(此处我们以albert为例进行阐述,也可用其他的预训练模型如bert、xlnet、roberta等替代),记每个字符对应倒数第i层transformer的输出向量为我们将最后4层transformer的输出进行concatenate拼接,那么每个字符对应的拼接后向量即为这些向量汇集得到embedding矩阵。
对查询向量和键向量计算内积并缩放得到权重λij,对所有权重进行聚合得到初始的权重矩阵Φ,具体地:
针对以上的操作也可以跟transformer一样,进行multi-head化和norm的操作再输出。
所述的步骤四拿到特征向量矩阵后,通过max或者mean等pooling层,将矩阵降维压缩成向量;再接入全连接层,最后进入softmax层输出分类结果,训练上,在lossfunction中采取增加惩罚项的方式,防止loss因为模长增大而降低,训练的时候根据业务需求,主要采用recall指标作为metric的方式。
本发明的工作原理:首先通过预处理,对输入的文本进行数据清洗,拆分文本的内部结构,根据要素之间的联系提取关系图矩阵,并且根据关系的不同分配权重得到邻接矩阵M,其次是采用成熟的预训练模型albert,对文本进行embeddding获得文本的向量矩阵E,在此基础上我们加入了一个attention层,对每个字向量分别通过与查询矩阵WQ、键矩阵WK、值矩阵WV相乘得到查询向量键向量值向量通过计算出连接权重;不同于其它的graph attention的机制,本方案将邻接矩阵和attention后获得的权重矩阵直接相乘,得到加权后的权重,最后权重与相乘后通过池化层和全连接层,通过softmax进行分类。
采用上述技术方案后,本发明有益效果为:针对金融领域文本分类问题,尤其是在半结构化的文本的情况下,寻常的分类器无法结合结构化信息进行识别,给出一种提取结构信息并且融入到分类***中,对最终的分类器准确率提升是行之有效的,同时针对模型结构,尤其是取末尾几层transformer、跨句间结构权重比同句内更高、multi-headattention等步骤,大大提高了信息表征抽取的能力,从而增加最后softmax分类的准确率,具有较大的市场推广价值。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。
图1是传统金融领域二级现券成交信息局部示意图;
图2是传统金融领域一级投标信息局部示意图;
图3是本发明的技术流程结构示意图;
图4是本发明的图矩阵M结构示意图;
具体实施方式
参看图1~图4所示,本具体实施方式包含以下技术方案步骤:步骤一,文本预处理、数据清洗,形成图矩阵M;步骤二,形成编码层,采用albert预训练模型获取embedding矩阵;步骤三,对embedding矩阵进行attention操作,并基于图矩阵进行加权计算;步骤四,对特征向量矩阵压缩后进行文本分类。
进一步的,所述的步骤一文本通常含有大量的噪音数据,首先对文本进行预处理清洗,如全半角转换、英文大小写统一、多个空白字符合并、去停用词等,文本切割,根据文本换行符,以及空格键、逗号分号、Tab键等分割符的判断,将文本数据切割成多个词或短句的形式,判断短句、词在文本中的位置,所处的行、列的位置信息,判断它们的连接关系,采用简单逻辑规则,在水平方向上,一个词与左右的词直接关联,在垂直方向上,如果水平方向处于同一位置,则关联,如此形成如下图4所示的图矩阵M,图矩阵的表达上,本方案采用不同的值表达不同的关联,单行内相邻字符之间的关联用数值1标识,要素之间的关联用2标识,其余未标出的区域均默认设置为0。
进一步的,所述的步骤二将文本通过事先在专业语料训练过的预训练模型(此处我们以albert为例进行阐述,也可用其他的预训练模型如bert、xlnet、roberta等替代),记每个字符对应倒数第i层transformer的输出向量为我们将最后4层transformer的输出进行concatenate拼接,那么每个字符对应的拼接后向量即为这些向量汇集得到embedding矩阵。
对查询向量和键向量计算内积并缩放得到权重λij,对所有权重进行聚合得到初始的权重矩阵Φ,具体地:
针对以上的操作也可以跟transformer一样,进行multi-head化和norm的操作再输出。
进一步的,所述的步骤四拿到特征向量矩阵后,通过max或者mean等pooling层,将矩阵降维压缩成向量;再接入全连接层,最后进入softmax层输出分类结果,训练上,在lossfunction中采取增加惩罚项的方式,防止loss因为模长增大而降低,训练的时候根据业务需求,主要采用recall指标作为metric的方式。
进一步的,本发明主要针对金融领域文本数据的分类问题,尤其是在处理半结构化文本的时候效果更佳,方案中获取embedding矩阵涉及到的预训练模型,可采用已有的语言模型、预训练模型或者是将来出现的相关技术替换。
本发明的工作原理:首先通过预处理,对输入的文本进行数据清洗,拆分文本的内部结构,根据要素之间的联系提取关系图矩阵,并且根据关系的不同分配权重得到邻接矩阵M,其次是采用成熟的预训练模型albert,对文本进行embeddding获得文本的向量矩阵E,在此基础上我们加入了一个attention层,对每个字向量分别通过与查询矩阵WQ、键矩阵WK、值矩阵WV相乘得到查询向量键向量值向量通过计算出连接权重;不同于其它的graph attention的机制,本方案将邻接矩阵和attention后获得的权重矩阵直接相乘,得到加权后的权重,最后权重与相乘后通过池化层和全连接层,通过softmax进行分类。
采用上述技术方案后,本发明有益效果为:针对金融领域文本分类问题,尤其是在半结构化的文本的情况下,寻常的分类器无法结合结构化信息进行识别,给出一种提取结构信息并且融入到分类***中,对最终的分类器准确率提升是行之有效的,同时针对模型结构,尤其是取末尾几层transformer、跨句间结构权重比同句内更高、multi-headattention等步骤,大大提高了信息表征抽取的能力,从而增加最后softmax分类的准确率,具有较大的市场推广价值。
以上所述,仅用以说明本发明的技术方案而非限制,本领域普通技术人员对本发明的技术方案所做的其它修改或者等同替换,只要不脱离本发明技术方案的精神和范围,均应涵盖在本发明的权利要求范围当中。
Claims (5)
1.一种基于图注意力的半结构文本分类方案,其特征在于:它包含以下技术方案步骤:步骤一,文本预处理、数据清洗,形成图矩阵M;步骤二,形成编码层,采用albert预训练模型获取embedding矩阵;步骤三,对embedding矩阵进行attention操作,并基于图矩阵进行加权计算;步骤四,对特征向量矩阵压缩后进行文本分类。
2.根据权利要求1所述的一种基于图注意力的半结构文本分类方案,其特征在于:所述的步骤一文本通常含有大量的噪音数据,首先对文本进行预处理清洗,如全半角转换、英文大小写统一、多个空白字符合并、去停用词等,文本切割,根据文本换行符,以及空格键、逗号分号、Tab键等分割符的判断,将文本数据切割成多个词或短句的形式,判断短句、词在文本中的位置,所处的行、列的位置信息,判断它们的连接关系,采用简单逻辑规则,在水平方向上,一个词与左右的词直接关联,在垂直方向上,如果水平方向处于同一位置,则关联,如此形成如下图4所示的图矩阵M,图矩阵的表达上,本方案采用不同的值表达不同的关联,单行内相邻字符之间的关联用数值1标识,要素之间的关联用2标识,其余未标出的区域均默认设置为0。
5.根据权利要求1所述的一种基于图注意力的半结构文本分类方案,其特征在于:所述的步骤四拿到特征向量矩阵后,通过max或者mean等pooling层,将矩阵降维压缩成向量;再接入全连接层,最后进入softmax层输出分类结果,训练上,在loss function中采取增加惩罚项的方式,防止loss因为模长增大而降低,训练的时候根据业务需求,主要采用recall指标作为metric的方式。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110415787.8A CN113312477A (zh) | 2021-04-19 | 2021-04-19 | 一种基于图注意力的半结构文本分类方案 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110415787.8A CN113312477A (zh) | 2021-04-19 | 2021-04-19 | 一种基于图注意力的半结构文本分类方案 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113312477A true CN113312477A (zh) | 2021-08-27 |
Family
ID=77372295
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110415787.8A Pending CN113312477A (zh) | 2021-04-19 | 2021-04-19 | 一种基于图注意力的半结构文本分类方案 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113312477A (zh) |
Cited By (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114925668A (zh) * | 2022-05-20 | 2022-08-19 | 电子科技大学 | 自动生成新闻的***、方法及存储介质 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200285944A1 (en) * | 2019-03-08 | 2020-09-10 | Adobe Inc. | Graph convolutional networks with motif-based attention |
CN111651992A (zh) * | 2020-04-24 | 2020-09-11 | 平安科技(深圳)有限公司 | 命名实体标注方法、装置、计算机设备和存储介质 |
CN112434720A (zh) * | 2020-10-22 | 2021-03-02 | 暨南大学 | 一种基于图注意力网络的中文短文本分类方法 |
-
2021
- 2021-04-19 CN CN202110415787.8A patent/CN113312477A/zh active Pending
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200285944A1 (en) * | 2019-03-08 | 2020-09-10 | Adobe Inc. | Graph convolutional networks with motif-based attention |
CN111651992A (zh) * | 2020-04-24 | 2020-09-11 | 平安科技(深圳)有限公司 | 命名实体标注方法、装置、计算机设备和存储介质 |
CN112434720A (zh) * | 2020-10-22 | 2021-03-02 | 暨南大学 | 一种基于图注意力网络的中文短文本分类方法 |
Non-Patent Citations (2)
Title |
---|
PETAR VELICKOVIC等: "GRAPH ATTENTION NETWORKS", ARXIV:1710.10903V3, 4 February 2018 (2018-02-04), pages 1 - 12 * |
许晶航等: "基于图注意力网络的因果关系抽取", 计算机研究与发展, vol. 57, no. 1, 31 January 2020 (2020-01-31), pages 159 - 174 * |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114925668A (zh) * | 2022-05-20 | 2022-08-19 | 电子科技大学 | 自动生成新闻的***、方法及存储介质 |
CN114925668B (zh) * | 2022-05-20 | 2023-04-07 | 电子科技大学 | 自动生成新闻的***、方法及存储介质 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Jung | Semantic vector learning for natural language understanding | |
CN111125334B (zh) | 一种基于预训练的搜索问答*** | |
CN112231447B (zh) | 一种中文文档事件抽取的方法和*** | |
CN111325029B (zh) | 一种基于深度学习集成模型的文本相似度计算方法 | |
CN112270188B (zh) | 一种提问式的分析路径推荐方法、***及存储介质 | |
Curtotti et al. | Corpus based classification of text in Australian contracts | |
CN113743097A (zh) | 基于跨度共享和语法依存关系增强的情感三元组抽取方法 | |
CN114818703B (zh) | 基于BERT语言模型和TextCNN模型的多意图识别方法及*** | |
CN112800184B (zh) | 基于Target-Aspect-Opinion联合抽取的短文本评论情感分析方法 | |
CN113672718A (zh) | 基于特征匹配和领域自适应的对话意图识别方法及*** | |
CN114936277A (zh) | 相似问题匹配方法和户相似问题匹配*** | |
CN113065349A (zh) | 基于条件随机场的命名实体识别方法 | |
CN112988970A (zh) | 一种服务于智能问答***的文本匹配算法 | |
CN115187066A (zh) | 风险识别方法、装置、电子设备及存储介质 | |
CN113312477A (zh) | 一种基于图注意力的半结构文本分类方案 | |
Cao et al. | Attention where it matters: Rethinking visual document understanding with selective region concentration | |
CN113342949A (zh) | 智库专家与待研究课题的匹配方法及*** | |
CN110287396A (zh) | 文本匹配方法及装置 | |
CN113177164B (zh) | 基于大数据的多平台协同新媒体内容监控管理*** | |
CN115952794A (zh) | 融合双语敏感词典和异构图的汉泰跨语言敏感信息识别方法 | |
CN112883183B (zh) | 构建多分类模型的方法、智能客服方法和相关装置及*** | |
CN114579729A (zh) | 一种融合多算法模型的faq问答匹配方法和*** | |
CN115204140A (zh) | 一种基于注意力机制与知识图谱的法律条文预测方法 | |
CN114036953A (zh) | 一种基于先验知识的金融领域实体及意图识别方法 | |
CN113761131A (zh) | 一种将文本结构化为表格的深度学习模型装置 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |