Exploring Small Language Models with Prompt-Learning Paradigm for Efficient Domain-Specific Text Classification

📅 2023-09-26
🏛️ arXiv.org
📈 Citations: 8
✨ Influential: 1
📄 PDF
🤖 AI Summary
Intent recognition in retail customer service is hindered by severe scarcity of labeled data. Method: This paper investigates synergistic optimization of small language models (SLMs, <1B parameters) with prompt learning. We systematically evaluate SLMs for few-shot and zero-shot text classification—first such study—and propose active few-shot sampling and multi-prompt ensemble strategies. We further identify prompt engineering as the decisive factor governing SLM zero-shot performance. Results: T5-base achieves 75% accuracy using only 15% of the labeled data. After prompt optimization, FLAN-T5-large’s zero-shot accuracy improves from <18% to over 31%, substantially narrowing the gap with GPT-3.5-turbo (55.16%). Our approach establishes an efficient, lightweight paradigm for intent recognition in low-resource domains.
📝 Abstract
Domain-specific text classification faces the challenge of scarce labeled data due to the high cost of manual labeling. Prompt-learning, known for its efficiency in few-shot scenarios, is proposed as an alternative to traditional fine-tuning methods. And besides, although large language models (LLMs) have gained prominence, small language models (SLMs, with under 1B parameters) offer significant customizability, adaptability, and cost-effectiveness for domain-specific tasks, given industry constraints. In this study, we investigate the potential of SLMs combined with prompt-learning paradigm for domain-specific text classification, specifically within customer-agent interactions in retail. Our evaluations show that, in few-shot settings when prompt-based model fine-tuning is possible, T5-base, a typical SLM with 220M parameters, achieve approximately 75% accuracy with limited labeled data (up to 15% of full data), which shows great potentials of SLMs with prompt-learning. Based on this, We further validate the effectiveness of active few-shot sampling and the ensemble strategy in the prompt-learning pipeline that contribute to a remarkable performance gain. Besides, in zero-shot settings with a fixed model, we underscore a pivotal observation that, although the GPT-3.5-turbo equipped with around 154B parameters garners an accuracy of 55.16%, the power of well designed prompts becomes evident when the FLAN-T5-large, a model with a mere 0.5% of GPT-3.5-turbo's parameters, achieves an accuracy exceeding 31% with the optimized prompt, a leap from its sub-18% performance with an unoptimized one. Our findings underscore the promise of prompt-learning in classification tasks with SLMs, emphasizing the benefits of active few-shot sampling, and ensemble strategies in few-shot settings, and the importance of prompt engineering in zero-shot settings.
Problem

Research questions and friction points this paper is trying to address.

Addresses data scarcity in customer intent recognition
Reduces dependency on extensive labeled datasets
Enhances small models' performance with minimal data
Innovation

Methods, ideas, or system contributions that make the work stand out.

Prompt-based learning reduces data dependency
Active sampling and ensemble learning enhance performance
Small models achieve competitive results with minimal data
🔎 Similar Papers
No similar papers found.
💼 Related Jobs
No related jobs found.
University of Helsinki | Ingka Group | IKEA
H
Hengyu Luo
University of Helsinki, Finland
P
Peng Liu
Ingka Group, IKEA, Malmö, Sweden
S
Stefan Esping
Ingka Group, IKEA, Malmö, Sweden