高级检索

基于KTSM的多任务情绪原因对抽取方法

Multi-task Emotion-cause Pair Extraction Based on KTSM

  • 摘要: 情绪原因对抽取的目的是抽取情绪子句和原因子句进行配对。现有的方法没有充分考虑到两个辅助任务(情绪抽取、原因抽取)和主任务情绪原因对抽取之间的关系,而现有的两阶段模型存在误差传播问题。此外,现有的模型也没有很好地解决情绪和原因的样本位置不平衡问题。为了解决以上问题,该文提出了一个新的端到端的基于知识图谱、Transformer的共享交互多任务模型(KTSM)。首先,通过多层次共享模块充分地模拟不同任务间的交互,挖掘主任务(情绪原因对提取)和两个辅助任务(情绪提取和原因提取)间的共享信息,从而提高主任务的性能。其次,为了解决标签位置不平衡问题,根据知识图路径长短筛选出合适的标签并构建特定任务的特征,使模型能够专注于提取具有相应情绪-原因关系的对。在ECPE基准数据集上的实验结果表明,模型取得了良好的性能,特别是在位置不平衡样本上表现良好。

     

    Abstract: The purpose of sentiment-cause pair extraction is to extract sentiment-sentence and cause-sentence pairs. The existing two-stage model suffers from error propagation problems. In addition, the previous model did not well address the problem of unbalanced sample positions of emotion and reason. To solve the above problems, this paper proposes a new end-to-end multitask model of shared interaction based on Knowledge Graph and Transformer (KTSM). Firstly, it models the interactions between different tasks through multi-level shared modules to mine the shared information between the main task emotion-cause pair extraction and the two auxiliary tasks emotion extraction and cause extraction. Second, the appropriate labels are filtered and task-specific features are constructed based on the knowledge graph path length, enabling the model to focus on extracting pairs with corresponding emotion-cause relationships. On the ECPE benchmark dataset, experimental results show that the model in this paper achieves good performance, especially on imbalanced samples.

     

/

返回文章
返回