Multi-label disaster text classification via supervised contrastive learning for social media data

Publisher:
PERGAMON-ELSEVIER SCIENCE LTD
Publication Type:
Journal Article
Citation:
Computers and Electrical Engineering, 2022, 104
Issue Date:
2022-12-01
Full metadata record
Social media is a crucial way to release information in a timely manner during disasters, which provides help to people who suffer from disasters. In this disaster information, each message may contain multiple labels. The single-label classification method cannot be adapted to a multi-label classification problem. In addition, there is a lack of representative baseline datasets due to the differences between disaster data sources. We propose a supervised contrastive learning based multi-label classification framework as a general framework for data processing and model training. Specifically, it learns features of disaster data and identifies different disaster type information, then trains a multi-label classification model of disaster texts in different dimensions. The results of empirical studies on three disaster text classification datasets show that this method can effectively improve the accuracy of the model and the representation ability of semantic information.
Please use this identifier to cite or link to this item: