Abstract:Multi-label text classification tasks face challenges such as sample diversity, complexity, and the need for effective utilization of label correlations. In this paper, we propose a model that integrates multi-granularity fusion of text sequence features and label semantic correlation information. Our model leverages graph convolutional networks to extract label semantic correlation, which enhances classification performance for samples with similar labels and addresses label omission issues. Additionally, text convolutional neural networks are employed to extract multi-granularity sense group features from text sequences, calculate their similarity with semantic correlation label distributions, and dynamically adjust the similarity between text context and label information. This approach tackles the limitations of feature extraction in short texts and label confusion. We replace the original multi-hot label encoding in model training with a label distribution that fuses text multi-granularity sense group features and label correlation information, using a more precise encoding method for soft alignment based on label probability distributions. This enhances the model's resilience to noisy data, avoiding the issue of assigning high-confidence probabilities to incorrect categories due to hard-coded supervision. Our model's performance improvement on noisy datasets significantly surpasses that achieved by label smoothing. Extensive experiments on three legal text datasets and two generalized multi-label datasets demonstrate the model's excellent performance. Our approach is applicable in various real-world scenarios, such as legal judgment prediction, news categorization, and recommendation systems, where accurate multi-label classification is crucial. Ablation and experiments on noisy datasets validate the model's effectiveness and robustness.

Multi-label classification of legal text based on label embedding and capsule network

MFLSCI: Multi-granularity fusion and label semantic correlation information for multi-label legal text classification

MLGN:A Multi-Label Guided Network for Improving Text Classification

Hierarchical Taxonomy-Aware and Attentional Graph Capsule RCNNs for Large-Scale Multi-Label Text Classification

Enhancing Deep Learning-Based Multi-label Text Classification with Capsule Network

A Smart Contract Classification Method Based on Label Embedding and Collaborative Attention Mechanism

Two-Stage Label Embedding Via Neural Factorization Machine for Multi-Label Classification

An Exploration of Encoder-Decoder Approaches to Multi-Label Classification for Legal and Biomedical Text

Multi-label Text Classification Based on the Label Correlation Mixture Model.

Gated Neural Network with Regularized Loss for Multi-label Text Classification.

Multi-label text classification based on semantic-sensitive graph convolutional network

Hierarchical multi-instance multi-label learning for Chinese patent text classification

SGM: Sequence Generation Model for Multi-label Classification.

Co-attention network with label embedding for text classification

Semantic-Unit-Based Dilated Convolution for Multi-Label Text Classification.

Multi-Label Classification with Label Graph Superimposing

Multi-label Text Classification Model Based on Multi-level Constraint Augmentation and Label Association Attention

KeNet:Knowledge-enhanced Doc-Label Attention Network for Multi-label text classification

A Hybrid Model Based on Convolutional Neural Network and Long Short-Term Memory for Multi-label Text Classification

A Label-Specific Attention-Based Network with Regularized Loss for Multi-label Classification

Research of multi-label text classification based on label attention and correlation networks