arXiv:2506.19465cs.CVcs.AI2025-06被引 1

用新型合成数据+逆向风格迁移,提升视觉模型预训练效果

Stylized Structural Patterns for Improved Neural Network Pre-training

  • 提出神经分形生成新合成数据,再通过逆向风格迁移增强真实感
  • 相比现有合成数据,生成图像与真实图像分布差距降低,FID降11%
  • 适合缺乏真实数据或受限于隐私/版权的模型开发场景

当前计算机视觉深度学习模型依赖大量真实图像数据,但其收集困难且存在隐私与法律风险,限制了商业化应用。已有研究尝试使用合成数据替代,但性能普遍不足。本文提出两阶段方法:首先通过改进的神经分形结构生成一类新型合成数据;其次引入逆向风格迁移技术,将少量无版权的真实图像视觉特征迁移至合成数据集,提升其表现力。我们利用核初始距离(KID)分析合成数据与真实图像间的域差距,结果表明该方法显著缩小了分布差异。实验显示,在不同任务中均体现实际效益:在合成数据上预训练EDM2扩散模型,图像生成的FID降低11%;自编码器重建误差下降20%,说明数据表征能力提升;基于该数据训练的ViT-S分类模型在ImageNet-100上准确率提升超10%。本工作为缺乏大规模真实数据时构建实用模型提供了新路径。

原文摘要 · Abstract (English)

Modern deep learning models in computer vision require large datasets of real images, which are difficult to curate and pose privacy and legal concerns, limiting their commercial use. Recent works suggest synthetic data as an alternative, yet models trained with it often underperform. This paper proposes a two-step approach to bridge this gap. First, we propose an improved neural fractal formulation through which we introduce a new class of synthetic data. Second, we propose reverse stylization, a technique that transfers visual features from a small, license-free set of real images onto synthetic datasets, enhancing their effectiveness. We analyze the domain gap between our synthetic datasets and real images using Kernel Inception Distance (KID) and show that our method achieves a significantly lower distributional gap compared to existing synthetic datasets. Furthermore, our experiments across different tasks demonstrate the practical impact of this reduced gap. We show that pretraining the EDM2 diffusion model on our synthetic dataset leads to an 11% reduction in FID during image generation, compared to models trained on existing synthetic datasets, and a 20% decrease in autoencoder reconstruction error, indicating improved performance in data representation. Furthermore, a ViT-S model trained for classification on this synthetic data achieves over a 10% improvement in ImageNet-100 accuracy. Our work opens up exciting possibilities for training practical models when sufficiently large real training sets are not available.

合成数据图像生成预训练风格迁移

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