arXiv:2410.07579cs.CV2024-10ECCV被引 17

用泰勒展开加速数据蒸馏,训练速度提升近一半且效果更好

Teddy: Efficient Large-Scale Dataset Distillation via Taylor-Approximated Matching

  • 用一阶泰勒近似替代多步梯度,大幅降低内存开销
  • 只需训练一个基础模型,生成弱模型池提升效率,精度最高超前人12.8%
  • 适合需要高效处理大规模数据集(如ImageNet)的研究者

数据蒸馏旨在将大规模数据集压缩为更小的合成数据集,使在该数据集上训练的模型能在真实数据上有效泛化。现有方法依赖双层优化:每轮迭代需训练新模型,并通过展开计算图传播梯度,导致内存和时间复杂度高,难以扩展至ImageNet等大数据集。本文提出Teddy框架,采用泰勒展开理论推导出一种内存高效的近似方法,将原需多步梯度的表达转换为一阶形式;同时发现,仅需训练一个基础模型,生成预缓存的弱模型池,即可在不重复训练新模型的前提下,显著提升时间和性能表现。大量实验表明,Teddy在Tiny-ImageNet和原始尺寸ImageNet-1K上均达到当前最佳效率与性能,相比先前方法精度最高提升12.8%,运行时间减少46.6%。代码将开源于https://github.com/Lexie-YU/Teddy。

原文摘要 · Abstract (English)

Dataset distillation or condensation refers to compressing a large-scale dataset into a much smaller one, enabling models trained on this synthetic dataset to generalize effectively on real data. Tackling this challenge, as defined, relies on a bi-level optimization algorithm: a novel model is trained in each iteration within a nested loop, with gradients propagated through an unrolled computation graph. However, this approach incurs high memory and time complexity, posing difficulties in scaling up to large datasets such as ImageNet. Addressing these concerns, this paper introduces Teddy, a Taylor-approximated dataset distillation framework designed to handle large-scale dataset and enhance efficiency. On the one hand, backed up by theoretical analysis, we propose a memory-efficient approximation derived from Taylor expansion, which transforms the original form dependent on multi-step gradients to a first-order one. On the other hand, rather than repeatedly training a novel model in each iteration, we unveil that employing a pre-cached pool of weak models, which can be generated from a single base model, enhances both time efficiency and performance concurrently, particularly when dealing with large-scale datasets. Extensive experiments demonstrate that the proposed Teddy attains state-of-the-art efficiency and performance on the Tiny-ImageNet and original-sized ImageNet-1K dataset, notably surpassing prior methods by up to 12.8%, while reducing 46.6% runtime. Our code will be available at https://github.com/Lexie-YU/Teddy.

数据蒸馏高效训练图像分类泰勒展开

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