arXiv:2506.24125cs.CVcs.AI2025-06NeurIPS被引 11

用数据残差匹配提升数据蒸馏效率与精度

FADRM: Fast and Accurate Data Residual Matching for Dataset Distillation

  • 引入数据级残差连接,平衡新知识与原始信息保留
  • 压缩比0.8%下,单模型准确率达47.7%,多模型达50.0%
  • 训练时间与显存降低50%,适合高效数据蒸馏场景

残差连接在模型架构中被广泛研究和应用,但在更挑战性的以数据为中心的方法中尚未探索。本文首次提出数据残差匹配(Data Residual Matching),通过数据级跳跃连接促进数据生成并缓解数据信息消失问题。该方法在像素空间优化获得的新知识与原始数据模态中的核心局部信息之间保持平衡,专用于数据蒸馏任务。结合优化层面的改进,本方法显著提升计算效率,在训练时间与峰值GPU内存使用上均减少50%。所提Fast and Accurate Data Residual Matching for Dataset Distillation(FADRM)达到新最优性能:在ResNet-18作为学生模型、ImageNet-1K压缩比为0.8%时,单模型测试准确率为47.7%,多模型为50.0%,分别优于RDED +5.7%,超过先进多模型方法EDC和CV-DD +1.4%和+4.0%。代码已开源。

原文摘要 · Abstract (English)

Residual connection has been extensively studied and widely applied at the model architecture level. However, its potential in the more challenging data-centric approaches remains unexplored. In this work, we introduce the concept of Data Residual Matching for the first time, leveraging data-level skip connections to facilitate data generation and mitigate data information vanishing. This approach maintains a balance between newly acquired knowledge through pixel space optimization and existing core local information identification within raw data modalities, specifically for the dataset distillation task. Furthermore, by incorporating optimization-level refinements, our method significantly improves computational efficiency, achieving superior performance while reducing training time and peak GPU memory usage by 50%. Consequently, the proposed method Fast and Accurate Data Residual Matching for Dataset Distillation (FADRM) establishes a new state-of-the-art, demonstrating substantial improvements over existing methods across multiple dataset benchmarks in both efficiency and effectiveness. For instance, with ResNet-18 as the student model and a 0.8% compression ratio on ImageNet-1K, the method achieves 47.7% test accuracy in single-model dataset distillation and 50.0% in multi-model dataset distillation, surpassing RDED by +5.7% and outperforming state-of-the-art multi-model approaches, EDC and CV-DD, by +1.4% and +4.0%. Code is available at: https://github.com/Jiacheng8/FADRM.

数据蒸馏残差匹配高效训练图像分类

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