arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.09656cs.LG
收藏

基于潜在空间中基于分数的扩散的混合类型表格数据合成

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

University of Illinois at Chicago(伊利诺伊大学芝加哥分校);Amazon Web Services(亚马逊网络服务);George Mason University(乔治梅森大学)

Hengrui Zhang, Jiani Zhang, Balasubramaniam Srinivasan, Zhengyuan Shen, Xiao Qin, Christos Faloutsos, Huzefa Rangwala, George Karypis

AI总结:

针对表格数据分布复杂、类型混杂导致扩散模型应用困难的问题,提出在VAE潜在空间中运用扩散模型的Tabsyn方法,兼具通用性、高质量与高速度,实验显示其在多项指标上显著优于现有基线。

摘要:

表格数据生成领域的近期进展已大幅提升了合成数据的质量。然而,由于表格数据的分布复杂多样且数据类型混杂,将扩散模型拓展应用于表格数据仍颇具挑战。本文提出Tabsyn方法,该方法在变分自编码器(VAE,Variational Autoencoder)构建的潜在空间中利用扩散模型来合成表格数据。所提出的Tabsyn的核心优势包括:(1)通用性:可通过将各类数据转换到统一空间并显式捕获列间关联,处理广泛的数据类型;(2)质量:优化潜在嵌入的分布以改进后续扩散模型的训练,助力生成高质量合成数据;(3)速度:与现有基于扩散的方法相比,反向步数少得多,合成速度更快。在6个数据集上采用5项指标开展的大量实验表明,Tabsyn的性能优于现有方法。具体而言,与最具竞争力的基线方法相比,它将列分布估计和列对相关性估计的错误率分别降低了86%和67%。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑