Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Peng, Letian, Gu, Yi, Dong, Chengyu, Wang, Zihan, Shang, Jingbo
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929387969970176
author Peng, Letian
Gu, Yi
Dong, Chengyu
Wang, Zihan
Shang, Jingbo
author_facet Peng, Letian
Gu, Yi
Dong, Chengyu
Wang, Zihan
Shang, Jingbo
contents For extremely weak-supervised text classification, pioneer research generates pseudo labels by mining texts similar to the class names from the raw corpus, which may end up with very limited or even no samples for the minority classes. Recent works have started to generate the relevant texts by prompting LLMs using the class names or definitions; however, there is a high risk that LLMs cannot generate in-distribution (i.e., similar to the corpus where the text classifier will be applied) data, leading to ungeneralizable classifiers. In this paper, we combine the advantages of these two approaches and propose to bridge the gap via a novel framework, \emph{text grafting}, which aims to obtain clean and near-distribution weak supervision for minority classes. Specifically, we first use LLM-based logits to mine masked templates from the raw corpus, which have a high potential for data synthesis into the target minority class. Then, the templates are filled by state-of-the-art LLMs to synthesize near-distribution texts falling into minority classes. Text grafting shows significant improvement over direct mining or synthesis on minority classes. We also use analysis and case studies to comprehend the property of text grafting.
format Preprint
id arxiv_https___arxiv_org_abs_2406_11115
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification
Peng, Letian
Gu, Yi
Dong, Chengyu
Wang, Zihan
Shang, Jingbo
Computation and Language
For extremely weak-supervised text classification, pioneer research generates pseudo labels by mining texts similar to the class names from the raw corpus, which may end up with very limited or even no samples for the minority classes. Recent works have started to generate the relevant texts by prompting LLMs using the class names or definitions; however, there is a high risk that LLMs cannot generate in-distribution (i.e., similar to the corpus where the text classifier will be applied) data, leading to ungeneralizable classifiers. In this paper, we combine the advantages of these two approaches and propose to bridge the gap via a novel framework, \emph{text grafting}, which aims to obtain clean and near-distribution weak supervision for minority classes. Specifically, we first use LLM-based logits to mine masked templates from the raw corpus, which have a high potential for data synthesis into the target minority class. Then, the templates are filled by state-of-the-art LLMs to synthesize near-distribution texts falling into minority classes. Text grafting shows significant improvement over direct mining or synthesis on minority classes. We also use analysis and case studies to comprehend the property of text grafting.
title Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification
topic Computation and Language
url https://arxiv.org/abs/2406.11115