通过强化关键特征区域,提升复杂场景下的数据蒸馏效果
Emphasizing Discriminative Features for Dataset Distillation in Complex Scenarios
- 用Grad-CAM激活图增强合成图像中的关键特征区域
- 在ImageNet-1K子集上超越现有最优方法
- 适合研究复杂场景下数据蒸馏的学者参考
数据蒸馏在CIFAR、MNIST和TinyImageNet等简单数据集上表现良好,但在复杂场景中效果下降。本文提出EDF(强调判别特征),利用Grad-CAM激活图增强合成图像中的高激活区域。观察发现:简单数据集中高激活区域覆盖广泛,而复杂场景中其范围显著缩小。与以往方法对所有像素同等处理不同,EDF聚焦于高激活区域。从监督角度,抑制低损失区域的信号,因其包含共性模式。为促进复杂场景研究,我们构建了复杂数据蒸馏基准(Comp-DD),从ImageNet-1K中精心选取16个子集(8个易、8个难)。EDF在复杂场景中持续优于当前最优结果,如ImageNet-1K子集。代码与基准将公开于https://github.com/NUS-HPC-AI-Lab/EDF。
原文摘要 · Abstract (English)
Dataset distillation has demonstrated strong performance on simple datasets like CIFAR, MNIST, and TinyImageNet but struggles to achieve similar results in more complex scenarios. In this paper, we propose EDF (emphasizes the discriminative features), a dataset distillation method that enhances key discriminative regions in synthetic images using Grad-CAM activation maps. Our approach is inspired by a key observation: in simple datasets, high-activation areas typically occupy most of the image, whereas in complex scenarios, the size of these areas is much smaller. Unlike previous methods that treat all pixels equally when synthesizing images, EDF uses Grad-CAM activation maps to enhance high-activation areas. From a supervision perspective, we downplay supervision signals that have lower losses, as they contain common patterns. Additionally, to help the DD community better explore complex scenarios, we build the Complex Dataset Distillation (Comp-DD) benchmark by meticulously selecting sixteen subsets, eight easy and eight hard, from ImageNet-1K. In particular, EDF consistently outperforms SOTA results in complex scenarios, such as ImageNet-1K subsets. Hopefully, more researchers will be inspired and encouraged to improve the practicality and efficacy of DD. Our code and benchmark will be made public at https://github.com/NUS-HPC-AI-Lab/EDF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。