GPT3Mix: Leveraging Large-scale Language Models for Text Augmentation
About
Large-scale language models such as GPT-3 are excellent few-shot learners, allowing them to be controlled via natural text prompts. Recent studies report that prompt-based direct classification eliminates the need for fine-tuning but lacks data and inference scalability. This paper proposes a novel data augmentation technique that leverages large-scale language models to generate realistic text samples from a mixture of real samples. We also propose utilizing soft-labels predicted by the language models, effectively distilling knowledge from the large-scale language models and creating textual perturbations simultaneously. We perform data augmentation experiments on diverse classification tasks and show that our method hugely outperforms existing text augmentation methods. Ablation studies and a qualitative analysis provide more insights into our approach.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Sequence Classification | MASSIVE | Micro F175.26 | 64 | |
| Sequence Classification | IMDB | Micro F187.69 | 64 | |
| Sequence Classification | Yahoo | Micro F152.93 | 64 | |
| Sequence Classification | ATIS | Micro F185.36 | 64 | |
| Sequence Classification | Huffpost low-resource (test) | Micro F169.46 | 64 | |
| Text Classification | Non-Class -> Class | Accuracy55.81 | 10 | |
| Text Classification | Class -> Class | Accuracy0.5162 | 10 |