Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Space

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhang, Hengrui, Zhang, Jiani, Srinivasan, Balasubramaniam, Shen, Zhengyuan, Qin, Xiao, Faloutsos, Christos, Rangwala, Huzefa, Karypis, George
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914792447410176
author Zhang, Hengrui
Zhang, Jiani
Srinivasan, Balasubramaniam
Shen, Zhengyuan
Qin, Xiao
Faloutsos, Christos
Rangwala, Huzefa
Karypis, George
author_facet Zhang, Hengrui
Zhang, Jiani
Srinivasan, Balasubramaniam
Shen, Zhengyuan
Qin, Xiao
Faloutsos, Christos
Rangwala, Huzefa
Karypis, George
contents Recent advances in tabular data generation have greatly enhanced synthetic data quality. However, extending diffusion models to tabular data is challenging due to the intricately varied distributions and a blend of data types of tabular data. This paper introduces Tabsyn, a methodology that synthesizes tabular data by leveraging a diffusion model within a variational autoencoder (VAE) crafted latent space. The key advantages of the proposed Tabsyn include (1) Generality: the ability to handle a broad spectrum of data types by converting them into a single unified space and explicitly capture inter-column relations; (2) Quality: optimizing the distribution of latent embeddings to enhance the subsequent training of diffusion models, which helps generate high-quality synthetic data, (3) Speed: much fewer number of reverse steps and faster synthesis speed than existing diffusion-based methods. Extensive experiments on six datasets with five metrics demonstrate that Tabsyn outperforms existing methods. Specifically, it reduces the error rates by 86% and 67% for column-wise distribution and pair-wise column correlation estimations compared with the most competitive baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2310_09656
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Space
Zhang, Hengrui
Zhang, Jiani
Srinivasan, Balasubramaniam
Shen, Zhengyuan
Qin, Xiao
Faloutsos, Christos
Rangwala, Huzefa
Karypis, George
Machine Learning
Recent advances in tabular data generation have greatly enhanced synthetic data quality. However, extending diffusion models to tabular data is challenging due to the intricately varied distributions and a blend of data types of tabular data. This paper introduces Tabsyn, a methodology that synthesizes tabular data by leveraging a diffusion model within a variational autoencoder (VAE) crafted latent space. The key advantages of the proposed Tabsyn include (1) Generality: the ability to handle a broad spectrum of data types by converting them into a single unified space and explicitly capture inter-column relations; (2) Quality: optimizing the distribution of latent embeddings to enhance the subsequent training of diffusion models, which helps generate high-quality synthetic data, (3) Speed: much fewer number of reverse steps and faster synthesis speed than existing diffusion-based methods. Extensive experiments on six datasets with five metrics demonstrate that Tabsyn outperforms existing methods. Specifically, it reduces the error rates by 86% and 67% for column-wise distribution and pair-wise column correlation estimations compared with the most competitive baselines.
title Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Space
topic Machine Learning
url https://arxiv.org/abs/2310.09656