基于多任务有监督对比学习的微博转发预测方法

MICROBLOG RETWEET PREDICTION METHOD BASED ON MULTI-TASK SUPERVISED CONTRASTIVE LEARNING

  • 摘要: 针对在微博转发预测研究中,由于数据集不平衡导致分类器偏差显著的问题,提出一种基于多任务有监督对比学习的微博转发预测方法。该方法包含转发预测和立场检测两个任务,两个任务提取各自相关的特征,并均由用于特征表示学习的对比学习分支,以及用于分类器学习的交叉煽驱动分支组成,学习从特征表示学习逐步过渡到分类器学习。实验结果表明,该方法有效地实现了对"转发"类别样本的正确预测,提高了微博转发预测的性能。

     

    Abstract: Aiming at the problem that the classifiers have significant bias due to the imbalance of data set for microblog retweet prediction, this paper proposes a microblog retweet prediction method based on multi- task supervised contrastive learning. This method consisted of two tasks: retweet prediction and stance detection. Both tasks extracted relevant features and consisted of two branches: a contrastive learning branch for feature representation learning and a cross- entropy driven branch for classifier learning, where the learning was progressively transited from feature representation learning to classifier learning. The experimental results show that the proposed method can effectively achieve the correct prediction for the samples of "retweet" category, and improve the performance of microblog retweet prediction.

     

/

返回文章
返回