pith. sign in

arxiv: 2510.09783 · v2 · pith:SXYCEOSMnew · submitted 2025-10-10 · 💻 cs.LG · cs.AI· stat.ML

Large Language Models for Imbalanced Classification: Diversity makes the difference

classification 💻 cs.LG cs.AIstat.ML
keywords samplesdiversityminorityclassificationmethodapproachesgenerategenerated
0
0 comments X
read the original abstract

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.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 1 Pith paper

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Retrieval Augmented Classification for Confidential Documents

    cs.CR 2026-04 unverdicted novelty 4.0

    RAC delivers stable 96% accuracy and up to 94% F1 on unbalanced confidential document classification while keeping sensitive information out of model parameters, matching or exceeding fine-tuned models in robustness.