Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

EPIC: Effective Prompting for Imbalanced-Class Data Synthesis in Tabular Data Classification via Large Language Models

About

Large language models (LLMs) have demonstrated remarkable in-context learning capabilities across diverse applications. In this work, we explore the effectiveness of LLMs for generating realistic synthetic tabular data, identifying key prompt design elements to optimize performance. We introduce EPIC, a novel approach that leverages balanced, grouped data samples and consistent formatting with unique variable mapping to guide LLMs in generating accurate synthetic data across all classes, even for imbalanced datasets. Evaluations on real-world datasets show that EPIC achieves state-of-the-art machine learning classification performance, significantly improving generation efficiency. These findings highlight the effectiveness of EPIC for synthetic tabular data generation, particularly in addressing class imbalance. Our source code for our work is available at: https://seharanul17.github.io/project-synthetic-tabular-llm/

Jinhee Kim, Taesung Kim, Jaegul Choo• 2024

Related benchmarks

TaskDatasetResultRank
ClassificationDiabetes (test)--
32
Tabular ClassificationTravel (test)
F1 Score0.7018
28
ClassificationElectricity
Mean Test Error Rate0.115
27
ClassificationThyroid
F1 Score94.8
17
ClassificationSICK
F1 Score88.71
15
ClassificationIncome (test)
F1 Score69.16
9
Classificationthyroid (test)
F1 Score94.8
9
ClassificationEye-Movements
Error Rate41
9
ClassificationMagicTelescope
Error Rate0.1975
9
ClassificationCredit
Error Rate27.539
9
Showing 10 of 15 rows

Other info

Code

Follow for update