arXiv:2506.06853cs.LGstat.ML2025-06ICML被引 1

用曲率增强数据增广,提升回归模型泛化能力

Curvature Enhanced Data Augmentation for Regression

  • 基于数据流形的二阶近似生成新样本
  • 在多个数据集上显著优于现有方法
  • 适合追求高泛化性能的回归任务研究者

参数量庞大的深度学习模型虽易过拟合,却常表现出良好泛化性,这得益于有效的正则化技术,其中数据增广应用最广。尽管在分类任务中成效显著,数据增广在回归任务中的研究仍较少。近期有方法基于数据流形的一阶近似生成合成数据。本文在此基础上提出理论框架与实用工具,用于近似和采样通用数据流形,并引入曲率增强流形采样(CEMS)方法。该方法利用数据流形的二阶表示,实现高效采样与新样本重建。在多个数据集上的广泛评估表明,CEMS在分布内与分布外场景下均表现优异,且计算开销极低。代码已公开于 https://github.com/azencot-group/CEMS。

原文摘要 · Abstract (English)

Deep learning models with a large number of parameters, often referred to as over-parameterized models, have achieved exceptional performance across various tasks. Despite concerns about overfitting, these models frequently generalize well to unseen data, thanks to effective regularization techniques, with data augmentation being among the most widely used. While data augmentation has shown great success in classification tasks using label-preserving transformations, its application in regression problems has received less attention. Recently, a novel \emph{manifold learning} approach for generating synthetic data was proposed, utilizing a first-order approximation of the data manifold. Building on this foundation, we present a theoretical framework and practical tools for approximating and sampling general data manifolds. Furthermore, we introduce the Curvature-Enhanced Manifold Sampling (CEMS) method for regression tasks. CEMS leverages a second-order representation of the data manifold to enable efficient sampling and reconstruction of new data points. Extensive evaluations across multiple datasets and comparisons with state-of-the-art methods demonstrate that CEMS delivers superior performance in both in-distribution and out-of-distribution scenarios, while introducing only minimal computational overhead. Code is available at https://github.com/azencot-group/CEMS.

数据增广回归任务流形学习

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