Constraint-Aware Synthetic Tabular Data Generation via Inter-Column Constraint Discovery with LLM Agents
Organizations: University of California, Davis · independent researcher
Abstract
Generating structurally valid synthetic tabular data remains difficult: outputs with high statistical fidelity and downstream utility can still violate semantically meaningful domain constraints. We study the discovery and enforcement of three complementary inter-column constraint families---equations, linear inequalities, and logical dependencies. Our unified tool-grounded workflow represents all three as machine-executable hypotheses and applies a common interface for full-table validation, deterministic diagnosis, and counterexample-guided revision. A generator-agnostic postprocessor coordinates family-specific repairs on outputs from unchanged tabular generators. Across curated behavioral audits and end-to-end evaluations, the complete workflow improves held-out violation detection over one-shot direct prompting, while postprocessing yields zero measured violations for every retained, applicable constraint, improves downstream utility on most datasets, and largely preserves univariate marginals.
Figures & tables
| Type | Model | Method | Acc. | Prec. | Rec. |
|---|---|---|---|---|---|
| Linear | GPT-5.6 | Baseline | .842 (.016) | .741 (.032) | .684 (.032) |
| Ours | .917 (.096) | .911 (.181) | .833 (.193) | ||
| Claude-5 | Baseline | .815 (.033) | .729 (.089) | .633 (.067) | |
| Ours | .959 (.016) | .975 (.031) | .919 (.032) | ||
| Equational | GPT-5.6 | Baseline | .975 (.037) | .950 (.075) | .950 (.075) |
| Ours | 1.000 (.000) | 1.000 (.000) | 1.000 (.000) |
| Metric | flights | heloc | nba | news | steel | taxi | url | |
|---|---|---|---|---|---|---|---|---|
| Constraint # (LD/Eq/Lin) | 3/7/2–5 | 1/0/11–13 | 0/10–12/3–11 | 7–16/4/0–17 | 1/2/11–25 | 2/2/0 | 2/0/17–29 | |
| LD | CVR (%) | 65.9 0 | 4.0 0 | – | 20.2 0 | 11.7 0 | 12.2 0 | 2.2 0 |
| sCVC (%) | 40.0 0 | 4.0 0 | – | 2.2 0 | 11.7 0 | 6.3 0 | 1.1 0 | |
| Eq. | CVR (%) | 100.0 0 | – | 99.9 0 | 100.0 0 | 98.2 0 | 100.0 0 | – |
| sCVC (%) | 99.3 0 | – | 94.4 0 | 86.1 0 | 63.7 0 | 96.6 0 | – | |
| 0.31 1 | – | 1 | 1 | 0.65 1 | 0.58 1 | – | ||
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Total | Train | Test | Cat. | Num. |
|---|---|---|---|---|---|
| flights | 60,000 | 42,000 | 18,000 | 9 | 14 |
| heloc | 9,084 | 6,358 | 2,726 | 3 | 21 |
| nba | 14,062 | 9,843 | 4,219 | 0 | 26 |
| news | 38,458 | 26,920 | 11,538 | 9 | 45 |
| steel | 34,993 | 24,495 | 10,498 | 3 | 7 |
| taxi | 59,996 | 41,997 | 17,999 | 7 | 15 |
| # | Ground-truth logical dependency |
|---|---|
| 1 | Anxiety Band = Very high (9--10) Caffeine Band = 294 mg or more |
| 2 | Anxiety Band = Very high (9--10) Breathing Rate Band = 20 breaths/min or higher |
| 3 | Anxiety Band = Very high (9--10) Heart Rate Band = 85 bpm or higher |
| 4 | Anxiety Band = Very high (9--10) Alcohol Band = 5 or more drinks |
| 5 | Anxiety Band = Very high (9--10) Diet Quality Band = Low (1--4) |
| 6 | Anxiety Band = Very high (9--10) Therapy Band = 3 or more sessions |
| CTGAN hyperparameter | Setting |
| Training epochs | 300 |
| Batch size | 500 |
| Noise/embedding dimension | 128 |
| Generator hidden layers | |
| Discriminator hidden layers | |
| Generator learning rate |
| TVAE hyperparameter | Setting |
| Training epochs | 300 |
| Batch size | 500 |
| Latent/embedding dimension | 128 |
| Encoder/compression layers | |
| Decoder/decompression layers | |
| Optimizer | Adam |
| Gaussian Copula hyperparameter | Setting |
| Default numerical marginal distribution | Beta |
| TabDDPM hyperparameter | Setting |
| Optimizer updates | 30,000 |
| Maximum training batch size | 4,096 |
| Optimizer | AdamW |
| Initial learning rate | |
| Learning-rate schedule | Linear decay toward zero |
| Denoising MLP hidden layers |
| Column Shapes | ||||
|---|---|---|---|---|
| Dataset | Generator | Raw | Postprocessed | |
| flights | CTGAN | +0.006 | ||
| Gaussian Copula | 0.000 | |||
| TabDDPM | +0.001 | |||
| TVAE | +0.003 | |||
| heloc | CTGAN | -0.006 | ||
| Dataset | Task / metric | Selected evaluation model | Generator | TRTR | Raw TSTR | Postprocessed TSTR | |
|---|---|---|---|---|---|---|---|
| flights | Regression ( ) | Linear regression | CTGAN | +0.016 | |||
| Gaussian Copula | +0.001 | ||||||
| TabDDPM | -0.094 | ||||||
| TVAE | -0.001 | ||||||
| heloc | Classification (ROC-AUC) | Logistic regression | CTGAN | 0.000 | |||
| Gaussian Copula | +0.009 |
| Logical dependency | Equational | Linear | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Dataset | Generator | CVR | sCVC | CVR | sCVC | CVR | sCVC | LFD | |
| flights | CTGAN | ||||||||
| Gaussian Copula | |||||||||
| TabDDPM | |||||||||
| TVAE | |||||||||
| heloc | CTGAN | – | – | – | |||||
| Column Shapes | |||
|---|---|---|---|
| Dataset | KS-guided | Random feasible | KS gain |
| flights | |||
| nba | |||
| news | |||
| taxi | |||
| steel | |||
| Generator | Metric | ||
|---|---|---|---|
| CTGAN | CVR (%) | ||
| LFD | |||
| TVAE | CVR (%) | ||
| LFD | |||
| Gaussian Copula | CVR (%) | ||
| LFD |
Synthetic Tabular Data Generation
Synthetic tabular data generation creates artificial records that preserve useful feature relationships while limiting exposure of sensitive data.
See all 99 papers