Label-template based Few-Shot Text Classification with Contrastive Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hou, Guanghua, Cao, Shuhui, Ouyang, Deqiang, Wang, Ning
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916521788309504
author Hou, Guanghua
Cao, Shuhui
Ouyang, Deqiang
Wang, Ning
author_facet Hou, Guanghua
Cao, Shuhui
Ouyang, Deqiang
Wang, Ning
contents As an algorithmic framework for learning to learn, meta-learning provides a promising solution for few-shot text classification. However, most existing research fail to give enough attention to class labels. Traditional basic framework building meta-learner based on prototype networks heavily relies on inter-class variance, and it is easily influenced by noise. To address these limitations, we proposes a simple and effective few-shot text classification framework. In particular, the corresponding label templates are embed into input sentences to fully utilize the potential value of class labels, guiding the pre-trained model to generate more discriminative text representations through the semantic information conveyed by labels. With the continuous influence of label semantics, supervised contrastive learning is utilized to model the interaction information between support samples and query samples. Furthermore, the averaging mechanism is replaced with an attention mechanism to highlight vital semantic information. To verify the proposed scheme, four typical datasets are employed to assess the performance of different methods. Experimental results demonstrate that our method achieves substantial performance enhancements and outperforms existing state-of-the-art models on few-shot text classification tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2412_10110
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Label-template based Few-Shot Text Classification with Contrastive Learning
Hou, Guanghua
Cao, Shuhui
Ouyang, Deqiang
Wang, Ning
Computation and Language
Artificial Intelligence
As an algorithmic framework for learning to learn, meta-learning provides a promising solution for few-shot text classification. However, most existing research fail to give enough attention to class labels. Traditional basic framework building meta-learner based on prototype networks heavily relies on inter-class variance, and it is easily influenced by noise. To address these limitations, we proposes a simple and effective few-shot text classification framework. In particular, the corresponding label templates are embed into input sentences to fully utilize the potential value of class labels, guiding the pre-trained model to generate more discriminative text representations through the semantic information conveyed by labels. With the continuous influence of label semantics, supervised contrastive learning is utilized to model the interaction information between support samples and query samples. Furthermore, the averaging mechanism is replaced with an attention mechanism to highlight vital semantic information. To verify the proposed scheme, four typical datasets are employed to assess the performance of different methods. Experimental results demonstrate that our method achieves substantial performance enhancements and outperforms existing state-of-the-art models on few-shot text classification tasks.
title Label-template based Few-Shot Text Classification with Contrastive Learning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2412.10110