vix.ing · top · new · best · stats

P-TA: Using Proximal Policy Optimization to Enhance Tabular Data Augmentation via Large Language Models

2024/06/17 by Shuo Yang, Yang, Shuo, Chenchen Yuan +7 · 1 citation
Computer Science · Decision Sciences · #Data Quality and Management #FOS: Computer and information sciences #Machine Learning (cs.LG) #Text Readability and Simplification #Topic Modeling

paper · pdf · doi:10.48550/arxiv.2406.11391

openalex publication_date 2024/06/17 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

A multitude of industries depend on accurate and reasonable tabular data augmentation for their business processes. Contemporary methodologies in generating tabular data revolve around utilizing Generative Adversarial Networks (GAN) or fine-tuning Large Language Models (LLM). However, GAN-based approaches are documented to produce samples with common-sense errors attributed to the absence of external knowledge. On the other hand, LLM-based methods exhibit a limited capacity to capture the disparities between synthesized and actual data distribution due to the absence of feedback from a discriminator during training. Furthermore, the decoding of LLM-based generation introduces gradient breakpoints, impeding the backpropagation of loss from a discriminator, thereby complicating the integration of these two approaches. To solve this challenge, we propose using proximal policy optimization (PPO) to apply GANs, guiding LLMs to enhance the probability distribution of tabular features. This approach enables the utilization of LLMs as generators for GANs in synthesizing tabular data. Our experiments demonstrate that PPO leads to an approximately 4% improvement in the accuracy of models trained on synthetically generated data over state-of-the-art across three real-world datasets.

Cited by

Related