基于词-主题-文本异质网络的短文本分类方法

SHORT TEXT CLASSIFICATION METHOD BASED ON WORD-TOPIC-DOCUMENT HETEROGENEOUS NETWORK

  • 摘要: 针对现有分类方法未考虑长距离词的语义相关性和文本间潜在主题共享的问题,提出一种基于词-主题-文本异质网络(WTDHN)的短文本分类方法。通过Word2vec训练词的上下文语义向量;构建词相关性矩阵以充足的词共现信息增强短文本各级别语义学;构建以词、主题和文本为节点的异质网络,并采用图卷积学习节点之间的高阶邻域信息,丰富短文本语义。相较于基准分类模型,该方法在五个公开短文本数据集上的分类准确率平均提高1.56%。

     

    Abstract: The existing short text classification methods ignore the semantic relevance between long-distance words and potential topic sharing between documents. To solve this issue, a novel short text classification method based on word-topic-document heterogeneous network (WTDHN) is proposed. The proposed method yielded the contextual semantic vectors of words through Word2vec. A word correlation matrix was constructed to enhance the learning of the potential topic distribution and the word distribution by sufficient word co-occurrence information. A heterogeneous network was constructed, with word, topic and document nodes included. The high-order neighborhood information between word, topic and document nodes was learned through the graph convolution operation, improving the semantic expression of short texts. The results on five public short text datasets show that the proposed method improves classification accuracy by 1.56% on average than the benchmark models.

     

/

返回文章
返回