Fine-tuned generative LLM oversampling can improve performance over traditional techniques on multiclass imbalanced text classification
Nicolas Antonio Cloutier, Nathalie Japkowicz
Abstract
A common challenge in classification and data mining is the class imbalance problem, where one class in a dataset has significantly more samples than another. The presence of this problem typically worsens the performance of classifiers trained on these datasets by giving them a strong preference for better-represented classes above others. Resampling is the process of adding (oversampling) or taking away (undersampling) samples from a dataset, typically with the purpose of introducing balance to the dataset before a classifier is trained. With recent advances in generative large language models (LLMs), certain authors have proposed the use of these models to generate new samples as a form of oversampling for imbalanced text data. This method has shown initial success, but has not been systematically compared to more traditional methods of resampling across many domains. We find that this use of generative LLMs generally outperforms other methods on imbalanced multiclass classification, but not on binary classification.