返回
Solving Data Imbalance in Text Classification With Constructing Contrastive Samples
DOI:10.1109/ACCESS.2023.3306805.png)
摘要
En 中文
Contrastive learning (CL) has been successfully applied in Natural Language Processing (NLP) as a powerful representation learning method and has shown promising results in various downstream tasks. Recent research has highlighted the importance of constructing effective contrastive samples through data augmentation. However, current data augmentation methods primarily rely on random word deletion, substitution, and cropping, which may introduce noisy samples and hinder representation learning. In this article, we propose a novel approach to address data imbalance in text classification by constructing contrastive samples. Our method involves the use of a Label-indicative Component to generate high-quality positive samples for the minority class, along with the introduction of a Hard Negative Mixing strategy to synthesize challenging negative samples at the feature level. By applying supervised contrastive learning to these samples, we are able to obtain superior text representations, which significantly benefit text classification tasks with imbalanced data. Our approach effectively mitigates distributional biases and promotes noise-resistant representation learning. To validate the effectiveness of our method, we conducted experiments on benchmark datasets (THUCNews, AG's News, 20NG) as well as the imbalanced FDCNews dataset. The code for our method is publicly available at the following GitHub repository: https://github.com/hanggun/CLDMTC.
Keyword:
Text categorization
Data augmentation
Task analysis
Data models
Representation learning
Training
Semantics
Data imbalance
contrastive learning
data augmentation
hard negative samples
text classification
期刊
IF:
3.6
论文数:
9.8W
被引数:
29.4W
机构
引用论文
Quantum semi-supervised generative adversarial network for enhanced data classification
SCIENTIFIC REPORTS
IF3.9
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent使用新型MIL-53(Fe) 作为高效吸附剂通过吸附从水溶液中去除砷
RSC Advances
IF0

