In multi-label text classification (MLTC), each given document is associated
with a set of correlated labels. To capture label correlations, previous
classifier-chain and sequence-to-sequence models transform MLTC to a sequence
prediction task. However, they tend to suffer from label order dependency,
label combination over-fitting and error propagation problems. To address these
problems, we introduce a novel approach with multi-task learning to enhance
label correlation feedback. We first utilize a joint embedding (JE) mechanism
to obtain the text and label representation simultaneously. In MLTC task, a
document-label cross attention (CA) mechanism is adopted to generate a more
discriminative document representation. Furthermore, we propose two auxiliary
label co-occurrence prediction tasks to enhance label correlation learning: 1)
Pairwise Label Co-occurrence Prediction (PLCP), and 2) Conditional Label
Co-occurrence Prediction (CLCP). Experimental results on AAPD and RCV1-V2
datasets show that our method outperforms competitive baselines by a large
margin. We analyze low-frequency label performance, label dependency, label
combination diversity and coverage speed to show the effectiveness of our
proposed method on label correlation learning.

通过引入多任务学习和标签相关性反馈机制来增强标签相关性学习，其中采用文档 - 标签交叉注意力机制来生成一个更具有区分度的文档表示，以及两个辅助标签共现预测任务来增强标签相关性学习，实验结果表明，我们的方法在 AAPD 和 RCV1-V2 数据集上优于竞争性基线方法。