arXiv:2508.00472cs.LG2025-08被引 3

通过概率采样潜在子空间,生成更贴近真实分布的不平衡表格数据。

A Conditional GAN for Tabular Data Generation with Probabilistic Sampling of Latent Subspaces

  • 基于聚类标签实现数据空间分块,指导生成器在合理子空间采样。
  • 在14个不平衡数据集上提升分类准确率,生成样本保真度高。
  • 适合处理类别不平衡问题的表格数据生成,尤其适用于医疗、金融领域。

表格数据是关系数据库和电子表格中的标准数据形式。然而,与其它数据形式一样,表格数据常面临类别不平衡问题,严重影响各类机器学习任务的性能。一种有效解决方案是使用生成对抗网络(GAN)为少数类合成人工样本。但现有大多数GAN模型未考虑真实数据空间中输入样本的向量子空间,导致生成数据位置任意。此外,类别标签与其他分类变量处理方式相同,使基于类别的条件采样效果不佳。为此,本文提出ctdGAN,一种用于缓解表格数据类别不平衡的条件GAN。首先,ctdGAN执行空间划分,为输入样本分配聚类标签;随后,利用这些标签通过新的概率采样策略和惩罚聚类与类别误判的新损失函数进行样本合成。该方法使生成样本位于与原始数据分布相似的子空间中。我们还引入了其他改进,包括一种简单而有效的聚类级缩放技术,在不改变数据维度的情况下捕捉多个特征模式。在14个不平衡数据集上的评估表明,ctdGAN具有强大的生成高保真样本能力,并显著提升分类准确率。

原文摘要 · Abstract (English)

The tabular form constitutes the standard way of representing data in relational database systems and spreadsheets. But, similarly to other forms, tabular data suffers from class imbalance, a problem that causes serious performance degradation in a wide variety of machine learning tasks. One of the most effective solutions dictates the usage of Generative Adversarial Networks (GANs) in order to synthesize artificial data instances for the under-represented classes. Despite their good performance, most of the proposed GAN models do not take into account the vector subspaces of the input samples in the real data space, leading to data generation in arbitrary locations. In addition, the class labels are handled in the same manner as the other categorical variables, so conditional sampling by class is rendered less effective. To overcome these problems, this study presents ctdGAN, a conditional GAN for alleviating class imbalance in tabular datasets. Initially, ctdGAN executes a space partitioning step to assign cluster labels to the input samples. Subsequently, it utilizes these labels to synthesize samples via a novel probabilistic sampling strategy and a new loss function that penalizes both cluster and class mis-predictions. In this way, ctdGAN generates samples in subspaces that resemble those of the original data distribution. We also introduce several other improvements, including a simple, yet effective cluster-wise scaling technique that captures multiple feature modes without affecting data dimensionality. The evaluation of ctdGAN with 14 imbalanced datasets demonstrated its strong ability in generating high fidelity samples and improving classification accuracy.

表格生成GAN类别不平衡

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。