基于门控双层异构图注意力网络的半监督短文本分类

针对现有的基于异构图神经网络的短文本分类方法未充分利用节点之间的有效信息，以及存在的过拟合问题，文中提出基于门控双层异构图注意力网络的半监督短文本分类方法(Semi-Supervised Short Text Classification with Gated Double-Layer Heterogeneous Graph Attention Network, GDHG).GDHG包含节点注意力机制和门控异构图注意力网络两层.首先，使用节点注意力机制，训练不同类型的节点注意力系数，再将系数输入门控异构图注意力网络，训练得到门控双层注意力.然后，将门控双层注意力与节点的不同状态相乘，得到聚合的节点特征.最后，使用softmax函数对文本进行分类.GDHG利用节点注意力机制和门控异构图注意力网络的信息遗忘机制对节点信息进行聚集，得到有效的相邻节点信息，进而挖掘不同邻居节点的隐藏信息，提高聚合远程节点信息的能力.在Twitter、MR、Snippets、AGNews四个短文本数据集上的实验验证GDHG性能较优.