TabDLM: Free-Form Tabular Data Generation via Joint Numerical-Language Diffusion
Organizations: Washington University in St. Louis · Peking University · Ant Group
Abstract
Synthetic tabular data generation has attracted growing attention due to its importance for data augmentation, foundation models, and privacy. However, real-world tabular datasets increasingly contain free-form text fields (e.g., reviews or clinical notes) alongside structured numerical and categorical attributes. Generating such heterogeneous tables with joint modeling of different modalities remains challenging. Existing approaches broadly fall into two categories: diffusion-based methods and LLM-based methods. Diffusion models can capture complex dependencies over numerical and categorical features in continuous or discrete spaces, but extending them to open-ended text is nontrivial and often leads to degraded text quality. In contrast, LLM-based generators naturally produce fluent text, yet their discrete tokenization can distort precise or wide-range numerical values, hindering accurate modeling of both numbers and language. In this work, we propose TabDLM, a unified framework for free-form tabular data generation via a joint numerical-language diffusion model built on masked diffusion language models (MDLMs). TabDLM models textual and categorical features through masked diffusion, while modeling numerical features with a continuous diffusion process through learned specialized numeric tokens embedding; bidirectional attention then captures cross-modality interactions within a single model. Extensive experiments on diverse benchmarks demonstrate the effectiveness of TabDLM compared to strong diffusion- and LLM-based baselines.
Figures & tables
| Method | MathExpr | ProfileBio | |||||
|---|---|---|---|---|---|---|---|
| Shape | Trend | Op-MR | Exp-MR | Shape | Trend | Bio-MR | |
| Qwen2.5-7B | |||||||
| Qwen2.5-14B | |||||||
| Qwen2.5-7B | |||||||
| LLaDA-8B | |||||||
| TabDLM | 3.69 | 6.07 | 99.99 | 99.80 | 4.84 | 7.81 | 98.60 |
| Method | Amazon | Arxiv | ||||||
|---|---|---|---|---|---|---|---|---|
| Shape (%) | Trend (%) | MLE | Shape (%) | Trend (%) | MLE | |||
| Real | ||||||||
| Qwen2.5-7B | ||||||||
| Qwen2.5-14B | ||||||||
| Qwen2.5-7B | ||||||||
| LLaDA-8B | ||||||||
| Method | Adult | Default | Shoppers | Magic | Beijing | Average |
|---|---|---|---|---|---|---|
| CTGAN | 16.84 / 20.23 | 16.83 / 26.95 | 21.15 / 13.08 | 9.81 / 7.00 | 21.39 / 22.95 | 17.20 / 18.04 |
| TVAE | 14.22 / 14.15 | 10.17 / 19.50 | 24.51 / 18.67 | 8.25 / 5.82 | 19.16 / 18.01 | 15.26 / 15.23 |
| GOGGLE | 16.97 / 45.29 | 17.02 / 21.94 | 22.33 / 23.90 | 1.90 / 9.47 | 16.93 / 45.94 | 15.03 / 29.31 |
| STaSy | 11.29 / 14.51 | 5.77 / 5.96 | 9.37 / 8.49 | 6.29 / 6.61 | 6.71 / 8.00 | 7.89 / 8.71 |
| CoDi | 21.38 / 22.49 | 15.77 / 68.41 | 31.84 / 17.78 | 11.56 / 6.53 | 16.94 / 7.07 | 19.50 / 24.46 |
| TabDDPM | 1.75 / 3.01 | 1.57 / 4.89 | 2.72 / 6.61 | 1.01 / 1.70 | 1.30 / 2.71 | 1.67 / 3.78 |
| Method | Shape | Trend | MLE | C2ST | -Precision |
|---|---|---|---|---|---|
| TabDLM - noFloatAE | |||||
| TabDLM - onlyContDiff | |||||
| TabDLM - onlyMDLM | |||||
| TabDLM |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Column | Example Value |
|---|---|
| operation_x1 | sin |
| operation_x2 | log |
| operation_between | mul |
| latex_expression | \sin(2.75) \times \log(6.40) |
| Column | Example Value |
|---|---|
| age | |
| salary | |
| sex | female |
| birth_state | California |
| college | Harvard University |
| degree | master |
| Component | Rule / Template |
|---|---|
| Age descriptor | [21, 30]: in the early career stage [31, 40]: in the career-building stage [41, 50]: in the established career stage [51, 60]: in the advanced career stage [61, 70]: in the late career stage |
| Salary descriptor | [75, 100]: a comfortable income [101, 150]: a strong professional income [151, 200]: a high-level income |
| Biography template | This {sex} individual is {age_desc}. {He/She} was born in {birth_state} and completed higher education at {college}, earning a {degree} degree. {He/She} works as a {occupation}. {He/She} earns {salary_desc}. |
| Column | Example Value |
|---|---|
| price | |
| review_time | (days since earliest review date) |
| rating | |
| verified | true |
| category | Science Fiction & Fantasy > Fantasy |
| brand | Visit Amazon’s J. R. R. Tolkien Page |
| Column | Example Value |
|---|---|
| submission_time | (days since earliest review date) |
| submission_year | |
| title_length | |
| abstract_length | |
| category | Category_3 |
| title | IRC-safe Graph Autoencoder for unsupervised anomaly detection |
| Dataset | #Rows | #Num | #Cat | #Max Cat | #Train | #Validation | #Test | Task |
|---|---|---|---|---|---|---|---|---|
| Adult | 48,842 | 6 | 9 | 42 | 28,943 | 3,618 | 16,281 | Classification |
| Default | 30,000 | 14 | 11 | 11 | 24,000 | 3,000 | 3,000 | Classification |
| Shoppers | 12,330 | 10 | 8 | 20 | 9,864 | 1,233 | 1,233 | Classification |
| Magic | 19,019 | 10 | 1 | 2 | 15,215 | 1,902 | 1,902 | Classification |
| Beijing | 43,824 | 7 | 5 | 31 | 35,058 | 4,383 | 4,383 | Regression |
| Method | Adult | Default | Shoppers | Magic | Beijing | Average |
|---|---|---|---|---|---|---|
| CTGAN | ||||||
| TVAE | ||||||
| GOGGLE | ||||||
| STaSy | ||||||
| CoDi | ||||||
| TabDDPM |
| Method | Adult | Default | Shoppers | Magic | Beijing | Average |
|---|---|---|---|---|---|---|
| CTGAN | ||||||
| TVAE | ||||||
| GOGGLE | ||||||
| STaSy | ||||||
| CoDi | ||||||
| TabDDPM |
| Methods | Adult | Default | Shoppers | Magic | Beijing | Average Gap |
|---|---|---|---|---|---|---|
| AUC | AUC | AUC | AUC | RMSE | ||
| Real | ||||||
| CTGAN | ||||||
| TVAE | ||||||
| GOGGLE | ||||||
| STaSy |
| Method | Adult | Default | Shoppers | Magic | Beijing | Average |
|---|---|---|---|---|---|---|
| CTGAN | ||||||
| TVAE | ||||||
| GOGGLE | ||||||
| STaSy | ||||||
| CoDi | ||||||
| TabDDPM |
| Method | Adult | Default | Shoppers | Magic | Beijing | Average |
|---|---|---|---|---|---|---|
| CTGAN | ||||||
| TVAE | ||||||
| GOGGLE | ||||||
| STaSy | ||||||
| CoDi | ||||||
| TabDDPM | 0.9998 |
| Method | Adult | Default | Shoppers | Beijing | Average |
|---|---|---|---|---|---|
| STaSy | |||||
| CoDi | |||||
| TabDDPM | |||||
| TabSyn | |||||
| TabDiff | |||||
| TabDLM |
Synthetic Tabular Data Generation
Synthetic tabular data generation creates artificial records that preserve useful feature relationships while limiting exposure of sensitive data.
See all 98 papers