A Note on Statistically Accurate Tabular Data Generation Using Large Language Models
Andrey Sidorenko
TL;DR
The paper tackles the challenge of generating realistic tabular data with large language models while preserving complex feature dependencies, especially for categorical variables. It introduces a probability-driven prompting approach that first infers conditional distributions and then samples from these distributions to create synthetic rows, reducing reliance on token-by-token generation. In a California demographic case study, the method demonstrates improved statistical fidelity over table-wide and cell-by-cell prompting, while remaining computationally scalable with a fixed set of distributional prompts. The approach offers a practical, adaptable solution for privacy-preserving, high-volume tabular data synthesis.
Abstract
Large language models (LLMs) have shown promise in synthetic tabular data generation, yet existing methods struggle to preserve complex feature dependencies, particularly among categorical variables. This work introduces a probability-driven prompting approach that leverages LLMs to estimate conditional distributions, enabling more accurate and scalable data synthesis. The results highlight the potential of prompting probability distributions to enhance the statistical fidelity of LLM-generated tabular data.
