Detecting diseases from social media has diverse types of applications such as disease spread detection and health monitoring, etc. Traditionally, language models (LMs) have been adopted for this purpose and they have achieved significant performance on this task. However, there is still ongoing research on improving the discriminating representations of LMs. To this end, in this paper, we propose a novel fine-tuning method that unifies Contrastive Learning (CL) and language modeling. Specifically, we propose a self-augmentation method that augments hidden representations of the model with its own representations. The proposed method consists of two branches, where the first branch is a traditional LM learning the features for the given data whereas the second branch is augmented with first branch representations to learn features encouraging generalization. CL further improves representations by pulling original and augmented versions closer and pushing other samples away. Our approach improves performance over the traditional fine-tuning approach on 3 NLP datasets of binary, multi-label, and multi-class classification tasks consisting of social media posts related to various diseases. Notably, it improves the F1-score up to \(2.48\%\) on baseline and \(2.1\%\) on state-of-the-art methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Improving Text Representation for Disease Detection from Social Media via Self-augmentation and Contrastive Learning

  • Pervaiz Iqbal Khan,
  • Andreas Dengel,
  • Sheraz Ahmed

摘要

Detecting diseases from social media has diverse types of applications such as disease spread detection and health monitoring, etc. Traditionally, language models (LMs) have been adopted for this purpose and they have achieved significant performance on this task. However, there is still ongoing research on improving the discriminating representations of LMs. To this end, in this paper, we propose a novel fine-tuning method that unifies Contrastive Learning (CL) and language modeling. Specifically, we propose a self-augmentation method that augments hidden representations of the model with its own representations. The proposed method consists of two branches, where the first branch is a traditional LM learning the features for the given data whereas the second branch is augmented with first branch representations to learn features encouraging generalization. CL further improves representations by pulling original and augmented versions closer and pushing other samples away. Our approach improves performance over the traditional fine-tuning approach on 3 NLP datasets of binary, multi-label, and multi-class classification tasks consisting of social media posts related to various diseases. Notably, it improves the F1-score up to \(2.48\%\) on baseline and \(2.1\%\) on state-of-the-art methods.