arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.04748cs.CV
收藏

数据集浓缩中的解耦分布匹配

Decomposed Distribution Matching in Dataset Condensation

Sahar Rahimi Malakshan, Mohammad Saeed Ebrahimi Saadabadi, Ali Dabouei, Nasser M. Nasrabadi

AI总结:

本文针对分布匹配式数据集浓缩的性能下降问题,将数据分布解耦为内容与风格,通过特征图统计矩匹配风格并最大化类内KL散度,在多个数据集和持续学习中取得显著提升。

摘要:

数据集浓缩(Dataset Condensation,DC)旨在通过合成一个小型数据集来降低深度神经网络的训练成本,使其与原始大型数据集同样有效。传统上,DC依赖代价高昂的双层优化,这限制了其实用性。近期研究将DC表述为分布匹配问题,从而规避了昂贵的双层优化。然而,这种效率提升牺牲了DC的性能。为研究这种性能下降,我们将数据集分布解耦为内容和风格。我们的观察揭示了两个主要缺陷:1)原始数据与浓缩数据之间的风格差异;2)浓缩数据集的类内多样性有限。我们提出了一种简单而有效的方法,利用特征图的统计矩这一成熟的风格指标,来匹配原始数据与浓缩数据之间的风格信息。此外,我们通过最大化每个合成类别(即内容)内的Kullback-Leibler散度来增强类内多样性。我们在不同规模和分辨率的多个数据集上通过实验证明了方法的有效性,在CIFAR10上取得最高4.1%的提升,在CIFAR100上提升4.2%,在TinyImageNet上提升4.3%,在ImageNet-1K上提升2.0%,在ImageWoof上提升3.3%,在ImageNette上提升2.5%,并在持续学习准确率上提升5.5%。

英文摘要:

Dataset Condensation (DC) aims to reduce deep neural networks training efforts by synthesizing a small dataset such that it will be as effective as the original large dataset. Conventionally, DC relies on a costly bi-level optimization which prohibits its practicality. Recent research formulates DC as a distribution matching problem which circumvents the costly bi-level optimization. However, this efficiency sacrifices the DC performance. To investigate this performance degradation, we decomposed the dataset distribution into content and style. Our observations indicate two major shortcomings of: 1) style discrepancy between original and condensed data, and 2) limited intra-class diversity of condensed dataset. We present a simple yet effective method to match the style information between original and condensed data, employing statistical moments of feature maps as well-established style indicators. Moreover, we enhance the intra-class diversity by maximizing the Kullback-Leibler divergence within each synthetic class, i.e., content. We demonstrate the efficacy of our method through experiments on diverse datasets of varying size and resolution, achieving improvements of up to 4.1% on CIFAR10, 4.2% on CIFAR100, 4.3% on TinyImageNet, 2.0% on ImageNet-1K, 3.3% on ImageWoof, 2.5% on ImageNette, and 5.5% in continual learning accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑