基于标签感知的门控循环采集方法

    公开(公告)号:CN114328927A

    公开(公告)日:2022-04-12

    申请号:CN202111655179.0

    申请日:2021-12-30

    IPC分类号: G06F16/35 G06F40/289 G06N3/04

    摘要: 本发明公开了一种基于标签感知的门控循环采集方法,包括如下步骤:S1、混合注意力层:混合注意层计算前一时间步分类预测向量与当前时间步文本词嵌入之间的注意权重;S2、文本特征提取层:使用卷积神经网络来提取文本特征;S3、循环采集层:门控循环单元根据前一时间步的分类结果和当前时间步的文本特征进行分类预测。本发明的分类结果具有更高的F1值(F1score)和更低的汉明损失(Hamming‑loss)。这说明模型的多次重读机制能够提升多标签分类的性能。本发明针对现有多标签分类算法的不足,将多标签分类任务建模为一个循环决策过程,考虑到了标签间的相关性,提升了模型的分类性能。