arXiv:2504.21152cs.LGcs.AI2025-04被引 6

用GAN精炼生成数据,提升少数样本回归预测效果

SMOGAN: Synthetic Minority Oversampling with GAN Refinement for Imbalanced Regression

  • 两阶段生成:先合成再用GAN过滤优化
  • 在23个数据集上均优于传统方法
  • 适合处理特征-目标关系复杂的不平衡回归任务

不平衡回归指目标变量分布偏斜的预测任务,此类偏斜使神经网络等模型过度关注密集区域,导致对稀疏区域(少数样本)预测性能差。现有方法多借鉴分类领域的技术,如线性插值或加高斯噪声生成合成数据,但难以捕捉复杂非线性分布,生成样本与真实关系不符。为此,本文提出SMOGAN,一种两阶段不平衡回归过采样框架:第一阶段由现有过采样器在稀疏目标区域生成初始合成样本;第二阶段引入DistGAN——一种基于分布感知的GAN,通过对抗损失与最大均值差异(MMD)目标联合优化,将合成样本调整至真实特征-目标联合分布。在23个不平衡数据集上的实验表明,SMOGAN始终优于不带DistGAN过滤层的基准方法。

原文摘要 · Abstract (English)

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.

不平衡回归生成模型数据增强GAN

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