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

SMOGAN:基于GAN优化的不平衡回归合成少数类过采样

SMOGAN: Synthetic Minority Oversampling with GAN Refinement for Imbalanced Regression

Shayan Alahyari, Mike Domaratzki

AI总结:

针对不平衡回归中数据分布复杂非线性导致现有合成数据不准确的问题,提出两步过采样框架SMOGAN,利用分布感知GAN(DistGAN)作为过滤层优化初始合成样本,使其对齐真实联合分布,在23个数据集上显著优于默认方法。

摘要:

不平衡回归是指目标变量偏斜的预测任务。这种偏斜性阻碍了机器学习模型,尤其是神经网络,它们集中于密集区域,因此在代表性不足的(少数类)样本上表现不佳。尽管该问题很重要,但针对不平衡回归仅提出了少数方法。许多现有的不平衡回归解决方案借鉴了类别不平衡领域的技术,如线性插值和添加高斯噪声,以在稀疏区域创建合成数据。然而,在许多情况下,数据的潜在分布是复杂且非线性的。因此,这些方法生成的合成样本无法准确表示真实的特征-目标关系。为了克服这些局限性,我们提出了SMOGAN,一个用于不平衡回归的两步过采样框架。在第一阶段,现有的过采样器在稀疏目标区域生成初始合成样本。在第二阶段,我们引入了DistGAN,一种分布感知的GAN,作为SMOGAN的过滤层,并通过增强有最大均值差异目标的对抗损失来优化这些样本,使其与真实的联合特征-目标分布对齐。在23个不平衡数据集上的大量实验表明,SMOGAN始终优于没有DistGAN过滤层的默认过采样方法。

英文摘要:

Imbalanced regression refers to prediction tasks where the target variable is skewed. This skewness hinders machine learning models, especially neural networks, which concentrate on dense regions and therefore perform poorly on underrepresented (minority) samples. Despite the importance of this problem, only a few methods have been proposed for imbalanced regression. Many of the available solutions for imbalanced regression adapt techniques from the class imbalance domain, such as linear interpolation and the addition of Gaussian noise, to create synthetic data in sparse regions. However, in many cases, the underlying distribution of the data is complex and non-linear. Consequently, these approaches generate synthetic samples that do not accurately represent the true feature-target relationship. To overcome these limitations, we propose SMOGAN, a two-step oversampling framework for imbalanced regression. In Stage 1, an existing oversampler generates initial synthetic samples in sparse target regions. In Stage 2, we introduce DistGAN, a distribution-aware GAN that serves as SMOGAN's filtering layer and refines these samples via adversarial loss augmented with a Maximum Mean Discrepancy objective, aligning them with the true joint feature-target distribution. Extensive experiments on 23 imbalanced datasets show that SMOGAN consistently outperforms the default oversampling method without the DistGAN filtering layer.

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

↑