HCL-MTC: Hierarchical Contrastive Learning for Multi-label Text Classification

Abstract:Multi-label text classification is a big chal-001 lenging subtask in text classification, where 002 labels generally form a tree structure. Ex-003 isting solutions learn the label tree structure 004 in a shallow manner and ignore the distinc-005 tive information between labels. To address 006 this problem, we propose a Hierarchical Con-007 trastive Learning for Multi-label Text Clas-008 sification (HCL-MTC), which constructs the 009 graph based on the contrastive knowledge be-010 tween labels. Specifically, we formulate the 011 MTC as a multi-task learning by introducing a 012 sampling hierarchical contrastive loss, which 013 learns both the correlative and distinctive label 014 information and is beneficial in learning deep 015 label hierarchy. The experimental results show 016 that the proposed model can achieve consid-017 erable improvements on both public datasets 018 (i.e., RCV1-v2 and WoS). 019
Computer Science
What problem does this paper attempt to address?