Abstract:In the multi-label text classification task, a text usually corresponds to multiple label categories, and the labels have correlation and hierarchical structure. However, when the label hierarchy is unknown, the number of various labels is not balanced, which makes it difficult for the model to classify low-frequency labels. At the same time, due to the existence of similar labels, the model will be difficult to distinguish similar labels. In this paper, we propose a multi-label text classification model based on multi-level constraint augmentation and label association attention. Compared with traditional methods, our method has two contributions: (1) In order to alleviate the problem of unbalanced number of different label categories and ensure the rationality of sample generation, we propose a data augmentation method based on multi-level constraints. In the process of sample generation, this method uses historical generation information, sample original text information and sample topic to constrain the generated text. (2) In order to make the model recognize the associated labels accurately, we propose an interaction mechanism based on label association attention and filter gate. This method combines text information and label weight information. At the same time, our classification model considers the important weights of text sentences and effectively utilizes the co-occurrence relationship between labels. Experimental results on three benchmark datasets show that our model outperforms state-of-the-art methods on all main evaluation metrics, especially on low-frequency label prediction with sparse samples.

Multi-Label Classification Algorithm Based on Association Rule Mining

Mining Item-Item and Between-Set Correlated Association Rules

Fuzzy Clustering-Based Quantitative Association Rules Mining in Multidimensional Data Set

Effective Mining Algorithm of Association Rules Based on Benefit Measure

Improved Multi-label Classification with Frequent Label-set Mining and Association

Association Rules Mining Based on the Discriminative Concept Lattice

Mining of Multi-Relational Association Rules

Multi-label classification by exploiting label correlations

Visual Analysis of User-Driven Association Rule Mining

A New Approach to Classification Based on Association Rule Mining

AN IMPROVED ALGORITHM BASED ON CATEGORY CLUSTERING FOR MINING ASSOCIATION RULES

MrCAR: A Multi-relational Classification Algorithm Based on Association Rules

Exploiting Label Relationship in Multi-Label Learning

Efficient Method for Mining Multiple-Level and Generalized Association Rules

A survey of multi-label classification based on supervised and semi-supervised learning

Joint Multi-Label Classification and Label Correlations with Missing Labels and Feature Selection

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

Joint Label-Specific Features and Correlation Information for Multi-Label Learning.

Multi-Label Learning by Exploiting Label Correlations Locally

Learning Label Correlations for Multi-Label Online Passive Aggressive Classification Algorithm

Multi-Label Learning Based on Clustering Optimization