CLAILGFeb 20, 2024

Comparing Specialised Small and General Large Language Models on Text Classification: 100 Labelled Samples to Achieve Break-Even Performance

arXiv:2402.12819v327 citationsh-index: 32EMNLP
Originality Incremental advance
AI Analysis

This work addresses the data efficiency problem for NLP practitioners by quantifying the trade-offs between model specialization and generalization in low-resource settings, though it is incremental in comparing existing methods.

The study determined that specialized small language models require an average of 100 labeled samples to match or exceed the performance of general large language models on text classification tasks, with this number increasing by 100-200% when accounting for performance variance.

When solving NLP tasks with limited labelled data, researchers typically either use a general large language model without further update, or use a small number of labelled samples to tune a specialised smaller model. In this work, we answer an important question -- how many labelled samples are required for the specialised small models to outperform general large models, while taking the performance variance into consideration. By observing the behaviour of fine-tuning, instruction-tuning, prompting and in-context learning on 8 language models, we identify such performance break-even points across 8 representative text classification tasks of varying characteristics. We show that the specialised models often need only few samples (on average $100$) to be on par or better than the general ones. At the same time, the number of required labels strongly depends on the dataset or task characteristics, with fine-tuning on binary datasets requiring significantly more samples. When performance variance is taken into consideration, the number of required labels increases on average by $100 - 200\%$. Finally, larger models do not consistently lead to better performance and lower variance, with 4-bit quantisation having negligible impact.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes