图卷积神经网络在关系抽取中的应用

之前使用关联规则建立起关系三元组之间的关联关系和关联强度,进而将关系样本连接起来,组成一张无向网,进而抽取关系三元组中的前项词向量和实体类型、后项词向量和实体类型(对于前项、后项是词组的情况,将词组中的词向量按照element
wise连加,合并成一个词向量),将前项、后项的词向量和实体类型(one-hot方式编码)连接一下,作为无向网中节点的特征,最终使用图卷积神经网络进行节点分类。
图卷积神经网络直接使用图卷积神经网络作者提供的源代码:
https://github.com/squirrel1982/gcn
跑完后发现分类结果并不好,我的关系类型一共16种(包括noEdge),测试集共1万个样本,模型结果输出的混淆矩阵如下:
与之相对,BiGRU-2Attention的结果如下: