arXiv:2410.13674cs.CVcs.AI2024-10ICCV被引 4

用图像引导扩散模型生成渐变数据,提升长尾和低质数据训练效果

Diffusion Curriculum: Synthetic-to-Real Data Curriculum via Image-Guided Diffusion

  • 通过调节图像引导强度,生成从合成到真实的连续数据谱
  • 在iWildCam上提升OOD与ID准确率2.7%和2.1%,在ImageNet-LT上尾类准确率从4.4%提至23.64%
  • 适合处理数据稀缺或质量差的场景,如长尾分类与低质图像学习

低质量或稀缺数据给深度神经网络训练带来显著挑战。传统数据增强难以生成差异显著的新数据,而扩散模型可通过文本提示生成高质量、多样化的合成数据。然而,仅依赖文本引导无法控制合成图像与原始数据的相似度,导致分布外数据损害模型性能。为此,本文研究图像引导策略,实现合成与真实图像间的连续插值:强图像引导生成接近训练数据但难学的图像;弱图像引导生成易学但分布差距大的图像。基于此构建新型“扩散课程”(DisCL),动态调整各训练阶段的图像引导强度,聚焦模型难点样本,并评估最优引导水平以增强困难数据的学习。将DisCL应用于长尾分类和低质数据学习任务,在iWildCam上提升OOD与ID宏准确率2.7%和2.1%;在ImageNet-LT上,尾类准确率由4.4%提升至23.64%,全类准确率提升4.02%。

原文摘要 · Abstract (English)

Low-quality or scarce data has posed significant challenges for training deep neural networks in practice. While classical data augmentation cannot contribute very different new data, diffusion models opens up a new door to build self-evolving AI by generating high-quality and diverse synthetic data through text-guided prompts. However, text-only guidance cannot control synthetic images' proximity to the original images, resulting in out-of-distribution data detrimental to the model performance. To overcome the limitation, we study image guidance to achieve a spectrum of interpolations between synthetic and real images. With stronger image guidance, the generated images are similar to the training data but hard to learn. While with weaker image guidance, the synthetic images will be easier for model but contribute to a larger distribution gap with the original data. The generated full spectrum of data enables us to build a novel "Diffusion Curriculum (DisCL)". DisCL adjusts the image guidance level of image synthesis for each training stage: It identifies and focuses on hard samples for the model and assesses the most effective guidance level of synthetic images to improve hard data learning. We apply DisCL to two challenging tasks: long-tail (LT) classification and learning from low-quality data. It focuses on lower-guidance images of high-quality to learn prototypical features as a warm-up of learning higher-guidance images that might be weak on diversity or quality. Extensive experiments showcase a gain of 2.7% and 2.1% in OOD and ID macro-accuracy when applying DisCL to iWildCam dataset. On ImageNet-LT, DisCL improves the base model's tail-class accuracy from 4.4% to 23.64% and leads to a 4.02% improvement in all-class accuracy.

扩散模型数据增强长尾学习图像生成

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