Label Dependencies-Aware Set Prediction Networks for Multi-Label Text Classification

Du Xinkai,Han Quanjie,Sun Yalin,Lv Chao,Sun Maosong
DOI: https://doi.org/10.1109/icassp48485.2024.10447036
2023-01-01
Abstract:Multi-label text classification involves extracting all relevant labels from a sentence. Given the unordered nature of these labels, we propose approaching the problem as a set prediction task. To address the correlation between labels, we leverage Graph Convolutional Networks and construct an adjacency matrix based on the statistical relations between labels. Additionally, we enhance recall ability by applying the Bhattacharyya distance to the output distributions of the set prediction networks. We evaluate the effectiveness of our approach on two multi-label datasets and demonstrate its superiority over previous baselines through experimental results.
What problem does this paper attempt to address?