Large Language Models for Imbalanced Classification: Diversity makes the difference
Not on the map. Why?
Abstract (arXiv)
Oversampling is one of the most widely used approaches for addressing imbalanced classification. The core idea is to generate additional minority samples to rebalance the dataset. Most existing methods, such as SMOTE, require converting categorical variables into numerical vectors, which often leads to information loss. Recently, large language model (LLM)-based methods have been introduced to overcome this limitation. However, current LLM-based approaches typically generate minority samples with limited diversity, reducing robustness and generalizability in downstream classification tasks. To address this gap, we propose a novel LLM-based oversampling method designed to enhance diversity. First, we introduce a sampling strategy that conditions synthetic sample generation on both minority labels and features. Second, we develop a new permutation strategy for fine-tuning pre-trained LLMs. Third, we fine-tune the LLM not only on minority samples but also on interpolated samples to further enrich variability. Extensive experiments on 10 tabular datasets demonstrate that our method significantly outperforms eight SOTA baselines. The generated synthetic samples are both realistic and diverse. Moreover, we provide theoretical analysis through an entropy-based perspective, proving that our method encourages diversity in the generated samples.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 23 of its 33 references that are in the map. First posted on arXiv 10 Oct 2025.
Builds on
23 of 33 references.
References · 23
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2019SciBERT: A Pretrained Language Model for Scientific Textcited by 1,404
- 2019Class-Balanced Loss Based on Effective Number of Samplescited by 1,239
- 2019An Introduction to Variational Autoencoderscited by 997
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 987
- 2019Modeling Tabular data using Conditional GANcited by 695
- 2021Tabular Data: Deep Learning is Not All You Needcited by 530
- 2021Deep Neural Networks and Tabular Data: A Surveycited by 489
- 2020TAPAS: Weakly Supervised Table Parsing via Pre-trainingcited by 432
- 2019TabFact: A Large-scale Dataset for Table-based Fact Verificationcited by 425
- 2020Reliable Fidelity and Diversity Metrics for Generative Modelscited by 425
- 2022Language Models are Realistic Tabular Data Generatorscited by 290
11 more
- 2022TabLLM: Few-shot Classification of Tabular Data with Large Language Modelscited by 273
- 2022Why do tree-based models still outperform deep learning on tabular data?cited by 198
- 2018Data Synthesis based on Generative Adversarial Networkscited by 184
- 2023Table Meets LLM: Can Large Language Models Understand Structured Table Data? A Benchmark and Empirical Studycited by 116
- 2023Synthcity: facilitating innovative use cases of synthetic data in different data modalitiescited by 76
- 2023Generative Table Pre-training Empowers Models for Tabular Predictioncited by 55
- 2023Curated LLM: Synergy of LLMs and Data Curation for tabular augmentation in low-data regimescited by 52
- 2018Table-to-Text: Describing Table Region with Natural Languagecited by 41
- 2020RPT: Relational Pre-trained Transformer Is Almost All You Need towards Democratizing Data Preparationcited by 27
- 2021OCT-GAN: Neural ODE-based Conditional Tabular GANscited by 27
- 2024Generating Realistic Tabular Data with Large Language Modelscited by 9
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Retrieval Augmented Classification for Confidential Documentscited by 0
- 2026Class-Structure Preservation Beats Diversity: A Comprehensive Benchmark of Text Augmentation Methods for Imbalanced Text Classificationcited by 0
- 2026SMOTE-VAR: An Uncertainty-Aware Oversampling Method for Predicting Depression Remission in University Studentscited by 0
- 2026Predicting Symptoms of Amotivation and Anhedonia among University Students with a Novel Oversampling Methodcited by 0
- 2026Predicting Symptoms of Amotivation and Anhedonia among University Students with a Novel Oversampling Method31 Aug 2026
- 2026SMOTE-VAR: An Uncertainty-Aware Oversampling Method for Predicting Depression Remission in University Students31 Aug 2026
- 2026Class-Structure Preservation Beats Diversity: A Comprehensive Benchmark of Text Augmentation Methods for Imbalanced Text Classification3 Jun 2026
- 2026Retrieval Augmented Classification for Confidential Documents9 Apr 2026