用自适应方法提升数据增强效果,降低噪声且计算成本低。
SAFLEX: Self-Adaptive Augmentation via Feature Label Extrapolation
- 通过特征标签外推学习增强样本权重与软标签。
- 在少样本和分布外任务中显著提升性能,噪声减少。
- 兼容多种增强策略与框架,适合新数据类型适配。
数据增强是深度学习中提升模型性能的关键技术,尤其在标注数据稀缺时尤为重要。传统方法依赖人工设计,适用性受限;现代可学习增强方法虽更灵活,但计算开销大且难融入主流流程。本文提出 SAFLEX(基于特征标签外推的自适应增强),通过高效双层优化算法,学习任意上游增强流水线生成的样本权重与软标签。SAFLEX 在极低计算成本下有效降低上游增强引入的噪声与标签错误。作为通用模块,SAFLEX 在自然图像、医学图像及表格数据上均表现优异,尤其在少样本学习与分布外泛化任务中优势明显。它可无缝集成于 RandAug、CutMix 等常见增强方法,以及 Stable Diffusion 等大模型生成增强,并兼容 CLIP 微调等框架。结果表明,现有增强流程可通过 SAFLEX 便捷适配新数据与任务,推动训练框架向更自适应、鲁棒方向发展。
原文摘要 · Abstract (English)
Data augmentation, a cornerstone technique in deep learning, is crucial in enhancing model performance, especially with scarce labeled data. While traditional techniques are effective, their reliance on hand-crafted methods limits their applicability across diverse data types and tasks. Although modern learnable augmentation methods offer increased adaptability, they are computationally expensive and challenging to incorporate within prevalent augmentation workflows. In this work, we present a novel, efficient method for data augmentation, effectively bridging the gap between existing augmentation strategies and emerging datasets and learning tasks. We introduce SAFLEX (Self-Adaptive Augmentation via Feature Label EXtrapolation), which learns the sample weights and soft labels of augmented samples provided by any given upstream augmentation pipeline, using a specifically designed efficient bilevel optimization algorithm. Remarkably, SAFLEX effectively reduces the noise and label errors of the upstream augmentation pipeline with a marginal computational cost. As a versatile module, SAFLEX excels across diverse datasets, including natural and medical images and tabular data, showcasing its prowess in few-shot learning and out-of-distribution generalization. SAFLEX seamlessly integrates with common augmentation strategies like RandAug, CutMix, and those from large pre-trained generative models like stable diffusion and is also compatible with frameworks such as CLIP's fine-tuning. Our findings highlight the potential to adapt existing augmentation pipelines for new data types and tasks, signaling a move towards more adaptable and resilient training frameworks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。