arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.15657cs.LG

面向不平衡分类的合成表格数据生成:重叠类别的惊人有效性

Synthetic Tabular Data Generation for Imbalanced Classification: The Surprising Effectiveness of an Overlap Class

Indian Institute of Technology, Bombay(印度理工学院孟买分校)

Annie D'souza, Swetha M, Sunita Sarawagi

AI总结:

针对表格不平衡分类中深度生成模型少数类样本质量偏低的问题,本文提出将二分类标签重构为含重叠类别的三分类标签,并在分类器训练前移除该类,实验证明可同时提升多种扩散与GAN模型及分类器性能。

摘要:

在表格数据上构建分类器时处理类别分布不平衡,长期以来一直是一个受到关注的问题。一种流行方法是使用合成生成的数据增强训练数据集。经典增强技术仅限于对现有少数类样本进行线性插值,而近期容量更高的深度生成模型正展现出更大前景。然而,在构建深度生成模型时处理类别分布不平衡同样是一个具有挑战性的问题,其研究程度尚不及不平衡分类器模型训练。我们表明,最先进的深度生成模型所生成的少数类样本质量明显低于多数类样本。我们提出一种新技术,通过为少数类与多数类分布重叠的区域引入一个类别,将二分类标签转换为三分类标签。我们表明,仅仅对训练集进行这种预处理,就能够显著提升跨多种最先进的扩散模型和基于 GAN 的模型所生成数据的质量。在使用合成数据训练分类器时,我们从训练数据中移除重叠类别,并论证准确率提升背后的原因。我们在四个真实数据集、五种不同分类器和五种生成模型上进行了大量实验,结果表明我们的方法不仅提升了最先进模型的合成器性能,也提升了分类器性能。

英文摘要:

Handling imbalance in class distribution when building a classifier over tabular data has been a problem of long-standing interest. One popular approach is augmenting the training dataset with synthetically generated data. While classical augmentation techniques were limited to linear interpolation of existing minority class examples, recently higher capacity deep generative models are providing greater promise. However, handling of imbalance in class distribution when building a deep generative model is also a challenging problem, that has not been studied as extensively as imbalanced classifier model training. We show that state-of-the-art deep generative models yield significantly lower-quality minority examples than majority examples. %In this paper, we start with the observation that imbalanced data training of generative models trained imbalanced dataset which under-represent the minority class. We propose a novel technique of converting the binary class labels to ternary class labels by introducing a class for the region where minority and majority distributions overlap. We show that just this pre-processing of the training set, significantly improves the quality of data generated spanning several state-of-the-art diffusion and GAN-based models. While training the classifier using synthetic data, we remove the overlap class from the training data and justify the reasons behind the enhanced accuracy. We perform extensive experiments on four real-life datasets, five different classifiers, and five generative models demonstrating that our method enhances not only the synthesizer performance of state-of-the-art models but also the classifier performance.

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

↑