ReTabSyn: Realistic Tabular Data Synthesis via Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lin, Xiaofeng, Kim, Seungbae, Li, Zhuoya, DeSoto, Zachary, Fleming, Charles, Cheng, Guang
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914386372722688
author Lin, Xiaofeng
Kim, Seungbae
Li, Zhuoya
DeSoto, Zachary
Fleming, Charles
Cheng, Guang
author_facet Lin, Xiaofeng
Kim, Seungbae
Li, Zhuoya
DeSoto, Zachary
Fleming, Charles
Cheng, Guang
contents Deep generative models can help with data scarcity and privacy by producing synthetic training data, but they struggle in low-data, imbalanced tabular settings to fully learn the complex data distribution. We argue that striving for the full joint distribution could be overkill; for greater data efficiency, models should prioritize learning the conditional distribution $P(y\mid \bm{X})$, as suggested by recent theoretical analysis. Therefore, we overcome this limitation with \textbf{ReTabSyn}, a \textbf{Re}inforced \textbf{Tab}ular \textbf{Syn}thesis pipeline that provides direct feedback on feature correlation preservation during synthesizer training. This objective encourages the generator to prioritize the most useful predictive signals when training data is limited, thereby strengthening downstream model utility. We empirically fine-tune a language model-based generator using this approach, and across benchmarks with small sample sizes, class imbalance, and distribution shift, ReTabSyn consistently outperforms state-of-the-art baselines. Moreover, our approach can be readily extended to control various aspects of synthetic tabular data, such as applying expert-specified constraints on generated observations.
format Preprint
id arxiv_https___arxiv_org_abs_2603_10823
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ReTabSyn: Realistic Tabular Data Synthesis via Reinforcement Learning
Lin, Xiaofeng
Kim, Seungbae
Li, Zhuoya
DeSoto, Zachary
Fleming, Charles
Cheng, Guang
Machine Learning
Deep generative models can help with data scarcity and privacy by producing synthetic training data, but they struggle in low-data, imbalanced tabular settings to fully learn the complex data distribution. We argue that striving for the full joint distribution could be overkill; for greater data efficiency, models should prioritize learning the conditional distribution $P(y\mid \bm{X})$, as suggested by recent theoretical analysis. Therefore, we overcome this limitation with \textbf{ReTabSyn}, a \textbf{Re}inforced \textbf{Tab}ular \textbf{Syn}thesis pipeline that provides direct feedback on feature correlation preservation during synthesizer training. This objective encourages the generator to prioritize the most useful predictive signals when training data is limited, thereby strengthening downstream model utility. We empirically fine-tune a language model-based generator using this approach, and across benchmarks with small sample sizes, class imbalance, and distribution shift, ReTabSyn consistently outperforms state-of-the-art baselines. Moreover, our approach can be readily extended to control various aspects of synthetic tabular data, such as applying expert-specified constraints on generated observations.
title ReTabSyn: Realistic Tabular Data Synthesis via Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2603.10823